Update app.py

app.py

@@ -8,22 +8,27 @@ import os
 CLASSIFIER = "Jzuluaga/accent-id-commonaccent_xlsr-en-english"
 
 def download_video(url):
-    """Handles YouTube and direct video links"""
-    if "youtube.com" in url:
-        yt = YouTube(url)
-        stream = yt.streams.filter(progressive=True, file_extension='mp4').first()
-        video_path = stream.download()
-        return video_path
-    else:  # Direct download (assumes URL is direct mp4 link)
-        # Download file locally
-        import requests
-        local_filename = "temp_video.mp4"
-        with requests.get(url, stream=True) as r:
-            r.raise_for_status()
-            with open(local_filename, 'wb') as f:
-                for chunk in r.iter_content(chunk_size=8192):
-                    f.write(chunk)
-        return local_filename
+    """Handles YouTube and direct video links with error handling"""
+    try:
+        if "youtube.com" in url or "youtu.be" in url:
+            yt = YouTube(url)
+            stream = yt.streams.filter(progressive=True, file_extension='mp4').first()
+            if not stream:
+                raise ValueError("No suitable video stream found.")
+            return stream.download()
+        else:
+            # For direct MP4 links, download file
+            import requests
+            local_filename = "temp_video.mp4"
+            with requests.get(url, stream=True) as r:
+                r.raise_for_status()
+                with open(local_filename, 'wb') as f:
+                    for chunk in r.iter_content(chunk_size=8192):
+                        f.write(chunk)
+            return local_filename
+    except Exception as e:
+        raise RuntimeError(f"Failed to download video: {e}")
+
 
 def extract_audio(video_path):
     clip = VideoFileClip(video_path)