Fix import issue

app.py

@@ -1,7 +1,5 @@
 import gradio as gr
 from video_accent_analyzer import VideoAccentAnalyzer
-import ffmpeg
-import os
 
 analyzer = VideoAccentAnalyzer()
 
@@ -12,20 +10,18 @@ def analyze_video(url=None, video_file=None):
     elif video_file:
         result = analyzer.analyze_local_video(video_file)
     else:
-        return "Please provide a video URL or upload a file"
+        return "❌ Please provide a video URL or upload a file"
 
     if 'error' in result:
         return f"❌ Error: {result['error']}"
 
-    # Format results as markdown
     markdown = f"## 🎯 Results\n"
     markdown += f"**Predicted Accent:** {result['predicted_accent']}\n"
-    markdown += f"**Confidence:** {result['accent_confidence']:.1f}%\n"
+    markdown += f"**Confidence:** {result['confidence']:.1f}%\n"
     markdown += f"**English Confidence:** {result['english_confidence']:.1f}%\n\n"
     markdown += "### 📊 Probability Breakdown:\n"
 
-    for accent, prob in sorted(result['all_probabilities'].items(),
-                               key=lambda x: x[1], reverse=True):
+    for accent, prob in sorted(result['probabilities'].items(), key=lambda x: x[1], reverse=True):
         markdown += f"- {accent}: {prob:.1f}%\n"
 
     return markdown
@@ -40,11 +36,11 @@ interface = gr.Interface(
     ],
     outputs=gr.Markdown(label="Analysis Results"),
     examples=[
-        ["https://www.youtube.com/watch?v=NO5SbsvIjHE ", None],
-        [None, "/kaggle/input/test-video.mp4"]
+        ["https://www.youtube.com/watch?v=abc123 ", None],
+        [None, "example_video.mp4"]
     ],
     title="🎧 Video Accent Analyzer",
-    description="Analyze accents in videos from YouTube, Loom, or uploaded files. Supports English accents only."
+    description="Detect English accents in videos from YouTube, Loom, or uploaded files"
 )
 
 if __name__ == "__main__":

requirements.txt

@@ -4,4 +4,6 @@ soundfile
 transformers
 torch
 gradio
-ffmpeg-python
+ffmpeg-python
+matplotlib
+seaborn