update handler

app.py

@@ -8,6 +8,7 @@ import os
 from moviepy.editor import VideoFileClip
 from pydub import AudioSegment
 import uuid
+import shutil
 
 MODEL_ID = "dima806/english_accents_classification"
 feature_extractor = AutoFeatureExtractor.from_pretrained(MODEL_ID)
@@ -31,12 +32,18 @@ def download_video(url):
         'outtmpl': video_path,
         'quiet': True,
         'no_warnings': True,
-        'noplaylist': True,
+        'http_headers': {
+            'User-Agent': 'Mozilla/5.0',
+        }
     }
     with yt_dlp.YoutubeDL(ydl_opts) as ydl:
         ydl.download([url])
 
-    return video_path
+    if not os.path.exists(video_path):
+        shutil.rmtree(temp_dir)
+        raise Exception("Failed to download video")
+
+    return video_path, temp_dir
 
 def extract_audio(video_file):
     audio_path = video_file.replace(".mp4", ".wav")
@@ -62,21 +69,26 @@ def classify_accent(audio_path):
     return PRETTY.get(label, label), round(confidence * 100, 2)
 
 def analyze_accent(url, file):
+    temp_dir = None
     try:
         if file is not None:
             file_path = file.name
-            audio_path = file_path if file_path.endswith(".wav") else extract_audio(file_path)
-        elif url:
-            video_path = download_video(url)
+            audio_path = file_path if file_path.lower().endswith(".wav") else extract_audio(file_path)
+        elif url and url.strip():
+            video_path, temp_dir = download_video(url)
             audio_path = extract_audio(video_path)
             os.remove(video_path)
         else:
-            return "Error", 0.0
+            return "Error: Please provide a URL or upload a file.", 0.0
 
         accent, confidence = classify_accent(audio_path)
         os.remove(audio_path)
+        if temp_dir:
+            shutil.rmtree(temp_dir)
         return accent, confidence
     except Exception as e:
+        if temp_dir:
+            shutil.rmtree(temp_dir)
         return f"Error: {str(e)}", 0.0
 
 iface = gr.Interface(
@@ -89,7 +101,11 @@ iface = gr.Interface(
         gr.Textbox(label="Predicted Accent"),
         gr.Number(label="Confidence (%)")
     ],
-    title="English Accent Analyzer"
+    title="English Accent Analyzer",
+    description=(
+        "Enter a YouTube URL or upload a video/audio file to analyze the English accent.\n\n"
+        "💡 If your video URL doesn't work, try uploading a video file instead."
+    )
 )
 
 iface.launch()

requirements.txt

@@ -89,3 +89,5 @@ urllib3==2.5.0
 Werkzeug==3.1.3
 youtube-dl==2021.12.17
 yt-dlp==2025.6.9
+pytube @ git+https://github.com/pytube/pytube.git@a32fff39058a6f7e5e59ecd06a7467b71197ce35
+