Add Gradio Audio Inference w/ clean messages

app.py

@@ -26,6 +26,7 @@ def convert_to_wav(input_path: str) -> str:
 
 def classify_audio(audio_path):
     messages = []
+
     ext = Path(audio_path).suffix.lower()
     if ext != ".wav":
         messages.append(f"🔁 Converting `{ext}` file to WAV format...")
@@ -35,7 +36,7 @@ def classify_audio(audio_path):
         except Exception as e:
             return f"❌ Failed to convert file to WAV.\n\n**Error:** {str(e)}"
 
-    messages.append("🔎 Running inference using YAMNet + Fine-Tuned Model...")
+    messages.append("🔎 Running inference using YAMNet + fine-tuned model...")
     try:
         result = classifier.classify_file(audio_path)
     except Exception as e:
@@ -48,17 +49,25 @@ def classify_audio(audio_path):
     return "\n\n".join(messages)
 
 def upload_message(file):
-    return "📥 Audio submitted. Click ▶️ to listen. Then press 'Submit' to classify."
+    return (
+        "📥 **Audio uploaded successfully!**\n\n"
+        "▶️ You can now listen to the uploaded audio.\n"
+        "🧠 Click **Submit** and Inference will run shortly."
+    )
 
-# UI setup
+# Gradio UI
 with gr.Blocks() as demo:
-    gr.Markdown("🐝 **AI-Belha Classifier** - Upload audio and classify using YAMNet + fine-tuned model")
+    gr.Markdown("## 🐝 AI-Belha Classifier\nClassify Queen Bee Status in beehives using YAMNet + Fine-Tuned Model.")
+    
     with gr.Row():
-        audio_input = gr.Audio(type="filepath", label="Upload audio")
+        audio_input = gr.Audio(type="filepath", label="Upload audio file")
         output_box = gr.Markdown()
 
     audio_input.upload(fn=upload_message, inputs=audio_input, outputs=output_box)
-    submit_btn = gr.Button("Submit")
+
+    with gr.Row():
+        submit_btn = gr.Button("🚀 Submit for Inference")
+    
     submit_btn.click(fn=classify_audio, inputs=audio_input, outputs=output_box)
 
 demo.launch()