Update app.py

app.py

@@ -1,20 +1,18 @@
 # To run this code you need to install the following dependencies:
 # pip install google-genai gradio
 
-import base64
 import os
 import gradio as gr
 import json
 from google import genai
 from google.genai import types
-api_key=os.environ.get("GEMINI_API_KEY")
 
-def generate_video_title(user_input, api_key):
+def generate_video_title(youtube_url, api_key):
     """
-    Generate video title using Gemini API
+    Analyze YouTube video and generate title using Gemini API
     
     Args:
-        user_input (str): The input text describing the video
+        youtube_url (str): YouTube video URL
         api_key (str): The Gemini API key
     
     Returns:
@@ -24,8 +22,8 @@ def generate_video_title(user_input, api_key):
         if not api_key:
             return "Error: Please provide a valid Gemini API key"
         
-        if not user_input.strip():
-            return "Error: Please provide input text describing your video"
+        if not youtube_url.strip():
+            return "Error: Please provide a YouTube video URL"
         
         # Initialize the client with the provided API key
         client = genai.Client(api_key=api_key)
@@ -35,7 +33,7 @@ def generate_video_title(user_input, api_key):
             types.Content(
                 role="user",
                 parts=[
-                    types.Part.from_text(text=user_input),
+                    types.Part.from_text(text=youtube_url),
                 ],
             ),
         ]
@@ -82,9 +80,9 @@ def generate_video_title(user_input, api_key):
 def create_gradio_interface():
     """Create and return the Gradio interface"""
     
-    with gr.Blocks(title="Video Title Generator", theme=gr.themes.Soft()) as app:
-        gr.HTML("<h1 style='text-align: center; color: #2563eb;'>🎬 Video Title Generator</h1>")
-        gr.HTML("<p style='text-align: center; color: #6b7280;'>Generate compelling video titles using AI</p>")
+    with gr.Blocks(title="YouTube Video Title Generator", theme=gr.themes.Soft()) as app:
+        gr.HTML("<h1 style='text-align: center; color: #2563eb;'>🎬 YouTube Video Title Generator</h1>")
+        gr.HTML("<p style='text-align: center; color: #6b7280;'>Analyze YouTube videos and generate titles with AI</p>")
         
         with gr.Row():
             with gr.Column(scale=2):
@@ -95,15 +93,15 @@ def create_gradio_interface():
                     info="Your API key is not stored and only used for this session"
                 )
                 
-                user_input = gr.Textbox(
-                    label="Video Description",
-                    placeholder="Describe your video content here...",
-                    lines=5,
-                    info="Provide details about your video content to generate an appropriate title"
+                youtube_url_input = gr.Textbox(
+                    label="YouTube Video URL",
+                    placeholder="https://www.youtube.com/watch?v=...",
+                    lines=2,
+                    info="Paste the YouTube video URL here"
                 )
                 
                 with gr.Row():
-                    generate_btn = gr.Button("Generate Title", variant="primary", size="lg")
+                    generate_btn = gr.Button("Analyze Video & Generate Title", variant="primary", size="lg")
                     clear_btn = gr.Button("Clear", variant="secondary")
             
             with gr.Column(scale=1):
@@ -117,26 +115,24 @@ def create_gradio_interface():
         # Event handlers
         generate_btn.click(
             fn=generate_video_title,
-            inputs=[user_input, api_key_input],
+            inputs=[youtube_url_input, api_key_input],
             outputs=output
         )
         
         clear_btn.click(
             fn=lambda: ("", ""),
-            outputs=[user_input, output]
+            outputs=[youtube_url_input, output]
         )
         
         # Example inputs
-        gr.HTML("<h3 style='margin-top: 20px;'>💡 Example Inputs:</h3>")
+        gr.HTML("<h3 style='margin-top: 20px;'>💡 Example YouTube URLs:</h3>")
         examples = gr.Examples(
             examples=[
-                ["A cooking tutorial showing how to make homemade pasta from scratch with step-by-step instructions"],
-                ["A travel vlog exploring the hidden gems of Tokyo, including local food markets and traditional temples"],
-                ["A tech review of the latest smartphone, covering camera quality, battery life, and performance"],
-                ["A fitness workout routine for beginners focusing on bodyweight exercises at home"],
-                ["A DIY home improvement project showing how to build a wooden bookshelf"]
+                ["https://www.youtube.com/watch?v=dQw4w9WgXcQ"],
+                ["https://youtu.be/dQw4w9WgXcQ"],
+                ["https://www.youtube.com/watch?v=example123"]
             ],
-            inputs=user_input,
+            inputs=youtube_url_input,
             label="Click on any example to try it out:"
         )
         
@@ -146,15 +142,19 @@ def create_gradio_interface():
             ### How to use this app:
             
             1. **Enter your Gemini API Key**: Get one from [Google AI Studio](https://aistudio.google.com/app/apikey)
-            2. **Describe your video**: Provide details about your video content
-            3. **Click "Generate Title"**: The AI will create a compelling title in German
-            4. **Copy the result**: Use the generated title for your video
+            2. **Paste YouTube URL**: Copy and paste the YouTube video URL
+            3. **Click "Analyze Video & Generate Title"**: Gemini will analyze the video and generate a title
+            4. **Copy the result**: Use the generated title
             
-            ### Tips for better results:
-            - Be specific about your video content
-            - Include the video type (tutorial, review, vlog, etc.)
-            - Mention key topics or themes
-            - Describe your target audience if relevant
+            ### Supported URL formats:
+            - `https://www.youtube.com/watch?v=VIDEO_ID`
+            - `https://youtu.be/VIDEO_ID`
+            - YouTube playlist URLs (will analyze the first video)
+            
+            ### Note:
+            - The AI analyzes the YouTube video content to generate relevant titles
+            - Generated titles are in German as per the system instruction
+            - Make sure the YouTube video is publicly accessible
             """)
     
     return app