Create main.py

main.py

@@ -0,0 +1,275 @@
+import gradio as gr
+import tempfile
+import os
+import yt_dlp
+import re
+from pathlib import Path
+import google.generativeai as genai
+from google.generativeai import upload_file, get_file
+import time
+
+# Configure Google AI
+GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
+if GOOGLE_API_KEY:
+    genai.configure(api_key=GOOGLE_API_KEY)
+else:
+    raise ValueError("GOOGLE_API_KEY environment variable not set")
+
+def is_valid_url(url):
+    """Check if URL is from supported platforms"""
+    patterns = [
+        r'(https?://)?(www\.)?(youtube\.com|youtu\.be)',
+        r'(https?://)?(www\.)?(instagram\.com|instagr\.am)',
+        r'(https?://)?(www\.)?(tiktok\.com)',
+        r'(https?://)?(www\.)?(twitter\.com|x\.com)',
+    ]
+    
+    for pattern in patterns:
+        if re.search(pattern, url, re.IGNORECASE):
+            return True
+    return False
+
+def download_video(url, progress=gr.Progress()):
+    """Download video from URL using yt-dlp"""
+    if not is_valid_url(url):
+        raise gr.Error("Please enter a valid YouTube, Instagram, TikTok, or X video URL")
+    
+    progress(0.1, desc="Starting download...")
+    
+    # Create temp file
+    temp_video = tempfile.NamedTemporaryFile(delete=False, suffix='.mp4')
+    temp_video.close()
+
+    # yt-dlp options optimized for HF Spaces
+    ydl_opts = {
+        'outtmpl': temp_video.name,
+        'format': 'best[filesize<100M]/worst[filesize<100M]/best',
+        'quiet': True,
+        'no_warnings': True,
+        'nooverwrites': False,
+        'user_agent': 'Mozilla/5.0 (compatible; GradioApp/1.0)',
+        'retries': 2,
+        'fragment_retries': 2,
+        'extractor_retries': 2,
+        'socket_timeout': 30,
+        'nocheckcertificate': True,
+    }
+
+    progress(0.3, desc="Downloading video...")
+    
+    try:
+        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+            ydl.download([url])
+
+        # Verify download
+        if os.path.exists(temp_video.name) and os.path.getsize(temp_video.name) > 0:
+            file_size_mb = os.path.getsize(temp_video.name) / (1024 * 1024)
+            
+            if file_size_mb > 100:
+                os.unlink(temp_video.name)
+                raise gr.Error(f"Video too large ({file_size_mb:.1f}MB). Please use a smaller video.")
+            
+            progress(0.7, desc="Processing video...")
+            return temp_video.name
+        else:
+            raise gr.Error("Downloaded file is empty or doesn't exist")
+
+    except Exception as e:
+        # Clean up on error
+        if os.path.exists(temp_video.name):
+            try:
+                os.unlink(temp_video.name)
+            except:
+                pass
+        
+        error_msg = str(e)
+        if "403" in error_msg or "Forbidden" in error_msg:
+            raise gr.Error("Video download was blocked. Try a different video or use a shorter one.")
+        elif "404" in error_msg:
+            raise gr.Error("Video not found. Please check the URL.")
+        else:
+            raise gr.Error(f"Download failed: {error_msg}")
+
+def analyze_video_with_ai(video_path, query, progress=gr.Progress()):
+    """Analyze video using Google Gemini AI"""
+    if not video_path or not os.path.exists(video_path):
+        raise gr.Error("No video file found")
+    
+    progress(0.1, desc="Uploading video to AI...")
+    
+    try:
+        # Upload video to Google AI
+        processed_video = upload_file(video_path)
+        
+        progress(0.5, desc="Processing video...")
+        
+        # Wait for processing
+        while processed_video.state.name == "PROCESSING":
+            time.sleep(2)
+            processed_video = get_file(processed_video.name)
+        
+        if processed_video.state.name == "FAILED":
+            raise gr.Error("Video processing failed")
+        
+        progress(0.8, desc="Generating response...")
+        
+        # Generate AI response
+        model = genai.GenerativeModel('gemini-2.0-flash-exp')
+        
+        prompt = f"""
+        Analyze this video and respond to the user's question: {query}
+        
+        Provide a comprehensive, insightful response that includes:
+        1. Direct analysis of the video content
+        2. Key insights and observations
+        3. Specific details from what you can see/hear
+        4. Actionable takeaways if relevant
+        
+        Be conversational and engaging while being thorough and accurate.
+        """
+        
+        response = model.generate_content([processed_video, prompt])
+        
+        progress(1.0, desc="Complete!")
+        
+        return response.text
+        
+    except Exception as e:
+        raise gr.Error(f"AI analysis failed: {str(e)}")
+    
+    finally:
+        # Clean up video file
+        try:
+            if video_path and os.path.exists(video_path):
+                os.unlink(video_path)
+        except:
+            pass
+
+def process_video_and_chat(url, query):
+    """Main function to download video and get AI response"""
+    if not url.strip():
+        raise gr.Error("Please enter a video URL")
+    
+    if not query.strip():
+        raise gr.Error("Please enter a question about the video")
+    
+    # Download video
+    progress = gr.Progress()
+    progress(0.0, desc="Starting...")
+    
+    video_path = download_video(url, progress)
+    
+    # Analyze with AI
+    response = analyze_video_with_ai(video_path, query, progress)
+    
+    return response
+
+# Create Gradio interface
+def create_interface():
+    with gr.Blocks(
+        title="The Plug - AI Video Analyzer",
+        theme=gr.themes.Soft(),
+        css="""
+        .gradio-container {
+            max-width: 800px !important;
+            margin: auto !important;
+        }
+        .header {
+            text-align: center;
+            margin-bottom: 2rem;
+        }
+        .footer {
+            text-align: center;
+            margin-top: 2rem;
+            color: #666;
+        }
+        """
+    ) as demo:
+        
+        # Header
+        gr.HTML("""
+        <div class="header">
+            <h1>🎥 The Plug - AI Video Analyzer</h1>
+            <p>Analyze videos from YouTube, Instagram, TikTok, and X with AI</p>
+        </div>
+        """)
+        
+        with gr.Row():
+            with gr.Column():
+                # Video URL input
+                video_url = gr.Textbox(
+                    label="Video URL",
+                    placeholder="https://youtube.com/watch?v=... or Instagram/TikTok/X link",
+                    lines=1,
+                    info="Paste a video URL from YouTube, Instagram, TikTok, or X"
+                )
+                
+                # Query input
+                query = gr.Textbox(
+                    label="Your Question",
+                    placeholder="What is the main topic? Summarize the key points...",
+                    lines=3,
+                    info="Ask anything about the video content"
+                )
+                
+                # Submit button
+                submit_btn = gr.Button("🚀 Analyze Video", variant="primary", size="lg")
+        
+        with gr.Row():
+            with gr.Column():
+                # Response output
+                response = gr.Textbox(
+                    label="AI Analysis",
+                    lines=15,
+                    max_lines=25,
+                    interactive=False,
+                    show_copy_button=True
+                )
+        
+        # Example queries
+        gr.Examples(
+            examples=[
+                ["https://www.youtube.com/watch?v=dQw4w9WgXcQ", "What is this video about?"],
+                ["", "Summarize the key points mentioned"],
+                ["", "What are the main takeaways?"],
+                ["", "Who is the target audience?"],
+            ],
+            inputs=[video_url, query],
+            label="Example Questions"
+        )
+        
+        # Footer
+        gr.HTML("""
+        <div class="footer">
+            <p>Built with ❤️ using Gradio and Google Gemini AI</p>
+            <p>Supports YouTube, Instagram, TikTok, and X video URLs</p>
+        </div>
+        """)
+        
+        # Connect the function
+        submit_btn.click(
+            fn=process_video_and_chat,
+            inputs=[video_url, query],
+            outputs=response,
+            show_progress=True
+        )
+        
+        # Also allow Enter key to submit
+        query.submit(
+            fn=process_video_and_chat,
+            inputs=[video_url, query],
+            outputs=response,
+            show_progress=True
+        )
+    
+    return demo
+
+# Launch the app
+if __name__ == "__main__":
+    demo = create_interface()
+    demo.launch(
+        share=True,
+        server_name="0.0.0.0",
+        server_port=7860,
+        show_error=True
+    )