Update app.py

app.py

@@ -709,6 +709,16 @@ css = """
 }
 """
 
+def analyze_emotions(audio_path, threshold):
+    if audio_path is None:
+        return "❌ Please upload a valid audio file.", None, None
+    try:
+        model_output = music2emo.predict(audio_path, threshold)
+        return format_prediction(model_output)
+    except Exception as e:
+        return f"⚠️ Error during prediction: {e}", None, None
+
+
 with gr.Blocks(css=css) as demo:
     gr.HTML(f"<h1 style='text-align: center;'>{title}</h1>")
     gr.Markdown(description_text)
@@ -750,14 +760,24 @@ with gr.Blocks(css=css) as demo:
                 mood_chart = gr.Plot(label="Mood Probabilities", scale=2, elem_classes=["gr-box"])
                 va_chart = gr.Plot(label="Valence-Arousal Space", scale=1, elem_classes=["gr-box"])
 
+    # predict_btn.click(
+    #     fn=lambda audio, thresh: format_prediction(music2emo.predict(audio, thresh)),
+    #     inputs=[input_audio, threshold],
+    #     outputs=[output_text, va_chart, mood_chart]
+    # )
     predict_btn.click(
-        fn=lambda audio, thresh: format_prediction(music2emo.predict(audio, thresh)),
+        fn=analyze_emotions,
         inputs=[input_audio, threshold],
         outputs=[output_text, va_chart, mood_chart]
     )
 
+    
+
 # Launch the App
 demo.queue().launch()
 
 
+
+
+