Update app.py

app.py

@@ -226,13 +226,23 @@ def generate_valid_filename(query):
 import whisper
 import time
 from pytube import YouTube
+from pytube.exceptions import PytubeError
 
 
 def download_video(url):
-    video = YouTube(url)
-    stream = video.streams.filter(file_extension='mp4')
-    stream.download()
-    return stream.default_filename
+    try:
+        video = YouTube(url)
+        stream = video.streams.filter(file_extension='mp4').first()
+        if stream:
+            return stream.download()
+        else:
+            raise ValueError("No suitable stream found.")
+    except PytubeError as e:
+        print(f"PytubeError: {e}")
+        raise
+    except Exception as e:
+        print(f"Unexpected error: {e}")
+        raise
 
 
 def video_to_text(filename):