Update app.py

app.py

@@ -20,6 +20,7 @@ st.title("Heartbeat Sound Classification")
 uploaded_file = st.file_uploader("Upload an audio file", type=["wav", "mp3"])
 
 if uploaded_file is not None:
+    st.subheader("Uploaded Audio File")
     # Load and display the audio file
     audio_bytes = uploaded_file.read()
     st.audio(audio_bytes, format='audio/wav')
@@ -32,10 +33,11 @@ if uploaded_file is not None:
     st.write("Classifying the audio...")
     results = classify_audio("temp_audio_file.wav")
     
-    # Display the classification results
-    st.write("Classification results:")
-    for label, score in results.items():
-        st.write(f"{label}: {score:.4f}")
+    # Display the classification results in a dedicated output box
+    st.subheader("Classification Results")
+    results_box = st.empty()
+    results_str = "\n".join([f"{label}: {score:.2f}" for label, score in results.items()])
+    results_box.text(results_str)
 
 # Examples of audio files for classification
 st.write("Examples:")
@@ -45,5 +47,6 @@ for example in examples:
     audio_bytes = open(example, 'rb').read()
     st.audio(audio_bytes, format='audio/wav')
     results = classify_audio(example)
-    for label, score in results.items():
-        st.write(f"{label}: {score:.4f}")
+    st.write("Results:")
+    results_str = "\n".join([f"{label}: {score:.2f}" for label, score in results.items()])
+    st.text(results_str)