Update app.py

app.py

@@ -4,11 +4,11 @@ import tempfile
 import requests
 import gradio as gr
 import cv2
-from pytube import YouTube
 from google import genai
 from google.genai import types
 from google.genai.types import Part
 from tenacity import retry, stop_after_attempt, wait_random_exponential
+import yt_dlp  # Use yt-dlp for robust YouTube downloading
 
 # Retrieve API key from environment variables.
 GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
@@ -51,21 +51,23 @@ def hhmmss_to_seconds(time_str: str) -> float:
 
 def download_video(video_url: str) -> str:
     """
-    Download the video from a URL (either YouTube or direct link) and return the local file path.
+    Download the video from a URL. If it's a YouTube URL, use yt-dlp;
+    otherwise, use requests for direct links.
+    Returns the local file path.
     """
     local_file = None
     if "youtube.com" in video_url or "youtu.be" in video_url:
-        yt = YouTube(video_url)
-        stream = yt.streams.filter(file_extension="mp4", progressive=True).first()
-        if stream is None:
-            raise ValueError("No suitable mp4 stream found on YouTube.")
-        temp_file = tempfile.NamedTemporaryFile(suffix=".mp4", delete=False)
-        stream.stream_to_buffer(temp_file)
-        temp_file.flush()
-        local_file = temp_file.name
-        temp_file.close()
+        ydl_opts = {
+            'format': 'mp4',
+            'outtmpl': '%(id)s.%(ext)s',
+            'noplaylist': True,
+            'quiet': True,
+        }
+        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+            info = ydl.extract_info(video_url, download=True)
+            local_file = ydl.prepare_filename(info)
     else:
-        # Assume it's a direct link to a video file, download using requests.
+        # Assume it's a direct link to a video file.
         response = requests.get(video_url, stream=True)
         if response.status_code == 200:
             temp_file = tempfile.NamedTemporaryFile(suffix=".mp4", delete=False)