Update app.py

app.py

@@ -85,11 +85,11 @@ model.load_state_dict(torch.load('model_weights.pth', map_location=device))
 model.eval()
 
 # Prediction function for Gradio
-def predict_bird(audio_file):
-    if audio_file is None:
+def predict_bird(file_path):
+    if file_path is None:
         return "Please upload an MP3 file."
     
-    predictions = infer_birdcall(model, audio_file, segment_length=500, device=str(device))
+    predictions = infer_birdcall(model, file_path, segment_length=500, device=str(device))
     
     # Format the predictions with numbering
     if not predictions:
@@ -98,13 +98,28 @@ def predict_bird(audio_file):
     numbered_predictions = [f"{i+1}. {bird}" for i, bird in enumerate(predictions)]
     return "\n".join(numbered_predictions)
 
+# Function to handle file upload and return path for audio player
+def process_upload(file_obj):
+    if file_obj is None:
+        return None
+    return file_obj
+
 # Create Gradio Blocks for more complex layout
 with gr.Blocks() as demo:
     gr.Markdown("# Bird Call Identification")
     
     with gr.Row():
-        with gr.Column():
-            audio_input = gr.Audio(type="filepath", label="Upload Bird Call Audio")
+        upload_file = gr.File(label="Upload MP3 file", file_types=[".mp3"])
+    
+    with gr.Row():
+        audio_player = gr.Audio(label="Listen to bird call", type="filepath", interactive=False)
+    
+    # Connect file upload to audio player
+    upload_file.change(
+        fn=process_upload,
+        inputs=upload_file,
+        outputs=audio_player
+    )
     
     with gr.Row():
         submit_btn = gr.Button("Identify Birds")
@@ -130,7 +145,7 @@ with gr.Blocks() as demo:
     # Set up the prediction event
     submit_btn.click(
         fn=predict_bird,
-        inputs=audio_input,
+        inputs=upload_file,
         outputs=output_text
     )