Update app.py

app.py

@@ -1,49 +1,98 @@
-from flask import Flask, render_template, request, jsonify
+import gradio as gr
+from detect import SimpleOfflineAccentClassifier
 import yt_dlp
 import os
-from detect import SimpleOfflineAccentClassifier
-
-app = Flask(__name__)
-classifier = SimpleOfflineAccentClassifier()
-
-@app.route('/')
-def home():
-    return render_template('index.html')
+import tempfile
 
-@app.route('/analyze', methods=['POST'])
-def analyze():
+def download_youtube_audio(url):
     try:
-        video_url = request.form['url']
+        # Geçici dosya yolu oluştur
+        temp_dir = tempfile.gettempdir()
+        temp_file = os.path.join(temp_dir, 'temp_audio.wav')
         
-        # YouTube'dan ses indir
         ydl_opts = {
             'format': 'bestaudio/best',
             'postprocessors': [{
                 'key': 'FFmpegExtractAudio',
                 'preferredcodec': 'wav',
             }],
-            'outtmpl': 'temp_audio',
+            'outtmpl': temp_file.replace('.wav', ''),
             'quiet': True,
             'no_warnings': True
         }
         
         with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-            ydl.download([video_url])
+            ydl.download([url])
         
-        # Ses dosyasını analiz et
-        result = classifier.predict_accent('temp_audio.wav')
+        return temp_file
+    except Exception as e:
+        print(f"Download error: {str(e)}")
+        return None
+
+def analyze_audio(audio_file, youtube_url):
+    try:
+        if youtube_url:
+            audio_file = download_youtube_audio(youtube_url)
+            if not audio_file:
+                return "Failed to download YouTube audio. Please check the URL and try again."
         
-        # Geçici dosyayı temizle
-        if os.path.exists('temp_audio.wav'):
-            os.remove('temp_audio.wav')
+        if not audio_file:
+            return "Please upload an audio file or provide a YouTube URL."
+        
+        classifier = SimpleOfflineAccentClassifier()
+        result = classifier.predict_accent(audio_file)
         
         if result is None:
-            return jsonify({'error': 'voice analyze failed.'})
+            return "Audio file processing failed."
+        
+        output = f"Predicted Accent: {result['accent']}\n"
+        output += f"Confidence Score: {result['confidence']:.2%}\n\n"
+        output += "All Probabilities:\n"
+        
+        sorted_probs = sorted(
+            result['all_probabilities'].items(), 
+            key=lambda x: x[1], 
+            reverse=True
+        )
         
-        return jsonify(result)
+        for accent, prob in sorted_probs:
+            bar = "█" * int(prob * 20)
+            output += f"- {accent}: {prob:.2%} {bar}\n"
+        
+        return output
         
     except Exception as e:
-        return jsonify({'error': str(e)})
+        return f"Error occurred: {str(e)}"
+
+# Create Gradio interface
+with gr.Blocks() as interface:
+    gr.Markdown("# AI Accent Classifier")
+    
+    with gr.Row():
+        with gr.Column():
+            audio_input = gr.Audio(
+                label="Upload Audio File",
+                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():
+            output_text = gr.Markdown(
+                label="Analysis Results",
+                value="Analysis results will appear here..."
+            )
+    
+    classify_btn.click(
+        fn=analyze_audio,
+        inputs=[audio_input, youtube_url],
+        outputs=output_text
+    )
 
-if __name__ == '__main__':
-    app.run(debug=True, port=5000)
+# Launch the interface
+interface.launch()