Update app.py

app.py

@@ -2,11 +2,10 @@ import gradio as gr
 from moviepy.editor import VideoFileClip
 from speechbrain.pretrained import EncoderClassifier
 import torchaudio
-from pytubefix import YouTube
-from pytubefix.cli import on_progress
 import requests
 import os
 import torch
+import yt_dlp
 
 CLASSIFIER = "Jzuluaga/accent-id-commonaccent_xlsr-en-english"
 
@@ -15,15 +14,20 @@ def get_default_device():
     return torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 def download_video(url):
-    """Download video from YouTube or direct MP4 URL using pytubefix."""
+    """Download video from YouTube or direct MP4 URL using yt_dlp or requests."""
     try:
         if "youtube.com" in url or "youtu.be" in url:
-            yt = YouTube(url, on_progress_callback=on_progress)
-            stream = yt.streams.filter(progressive=True, file_extension='mp4').first()
-            if not stream:
-                raise ValueError("No suitable video stream found.")
-            video_path = stream.download()
-            return video_path
+            output_path = "temp_video.%(ext)s"
+            ydl_opts = {
+                'format': 'best[ext=mp4]/best',
+                'outtmpl': output_path,
+                'quiet': True,
+                'noplaylist': True,
+            }
+            with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+                info_dict = ydl.extract_info(url, download=True)
+                downloaded_path = output_path.replace("%(ext)s", info_dict['ext'])
+            return downloaded_path
         else:
             # Direct MP4 file download
             local_filename = "temp_video.mp4"
@@ -70,20 +74,19 @@ def process_video(url):
     except Exception as e:
         return f"Error: {e}", ""
     finally:
-        # Clean up temporary files
         for f in [video_path, audio_path]:
             if f and os.path.exists(f):
                 os.remove(f)
 
 iface = gr.Interface(
     fn=process_video,
-    inputs=gr.Textbox(label="Enter Public Video URL (YouTube, Loom, direct MP4)"),
+    inputs=gr.Textbox(label="Enter Public Video URL (YouTube or direct MP4 link)"),
     outputs=[
         gr.Textbox(label="Detected Accent"),
         gr.Textbox(label="Confidence Score")
     ],
     title="English Accent Classifier",
-    description="Paste a public video URL to detect the English accent and confidence score."
+    description="Paste a public video URL (YouTube or MP4) to detect the English accent and confidence score."
 )
 
 if __name__ == "__main__":