Update app.py

app.py

@@ -1,9 +1,36 @@
 import gradio as gr
 from detect import SimpleOfflineAccentClassifier
+import yt_dlp
+import os
 
-def analyze_audio(audio_file):
-    if audio_file is None:
-        return "Please upload an audio file."
+def download_youtube_audio(url):
+    try:
+        ydl_opts = {
+            'format': 'bestaudio/best',
+            'postprocessors': [{
+                'key': 'FFmpegExtractAudio',
+                'preferredcodec': 'wav',
+            }],
+            'outtmpl': 'temp_audio',
+            'quiet': True,
+            'no_warnings': True
+        }
+        
+        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+            ydl.download([url])
+        
+        return 'temp_audio.wav'
+    except Exception as e:
+        return None
+
+def analyze_audio(audio_file, youtube_url):
+    if youtube_url:
+        audio_file = download_youtube_audio(youtube_url)
+        if not audio_file:
+            return "Failed to download YouTube audio."
+    
+    if not audio_file:
+        return "Please upload an audio file or provide a YouTube URL."
     
     try:
         classifier = SimpleOfflineAccentClassifier()
@@ -42,6 +69,11 @@ with gr.Blocks() as interface:
                 type="filepath"
             )
             
+            youtube_url = gr.Textbox(
+                label="Or enter YouTube URL",
+                placeholder="https://www.youtube.com/watch?v=..."
+            )
+            
             classify_btn = gr.Button("Analyze Accent")
         
         with gr.Column():
@@ -52,7 +84,7 @@ with gr.Blocks() as interface:
     
     classify_btn.click(
         fn=analyze_audio,
-        inputs=audio_input,
+        inputs=[audio_input, youtube_url],
         outputs=output_text
     )