Update app.py

app.py

@@ -15,7 +15,8 @@ def download_video(url):
             stream = yt.streams.filter(progressive=True, file_extension='mp4').first()
             if not stream:
                 raise ValueError("No suitable video stream found.")
-            return stream.download()
+            video_path = stream.download()  # Store the download path
+            return video_path
         else:
             # For direct MP4 links, download file
             import requests
@@ -25,11 +26,13 @@ def download_video(url):
                 with open(local_filename, 'wb') as f:
                     for chunk in r.iter_content(chunk_size=8192):
                         f.write(chunk)
-            return local_filename
+            video_path = local_filename # Store the download path
+            return video_path
     except Exception as e:
         raise RuntimeError(f"Failed to download video: {e}")
 
 
+
 def extract_audio(video_path):
     clip = VideoFileClip(video_path)
     audio_path = "temp_audio.wav"
@@ -51,16 +54,20 @@ def classify_accent(audio_path):
     return predicted_accent, f"{confidence:.2f}%"
 
 def process_video(url):
+    video_path = None
+    audio_path = None
     try:
         video_path = download_video(url)
         audio_path = extract_audio(video_path)
         accent, confidence = classify_accent(audio_path)
+        return accent, confidence
+    except Exception as e:
+        return f"Error: {e}", ""
     finally:
-        # Cleanup temp files if they exist
         for f in [video_path, audio_path]:
-            if os.path.exists(f):
+            if f and os.path.exists(f):
                 os.remove(f)
-    return accent, confidence
+
 
 # Gradio interface
 iface = gr.Interface(