add audio player

app.py

@@ -69,11 +69,29 @@ st.write("Upload an audio file to generate a spectrogram and predict its class u
 uploaded_file = st.file_uploader("Choose an audio file", type=["wav", "mp3"])
 
 if uploaded_file is not None:
-    # Save the uploaded audio file to a temporary location
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as temp_audio_file:
-        temp_audio_file.write(uploaded_file.read())
+    # Read uploaded bytes so we can both play and save to temp for processing
+    file_bytes = uploaded_file.read()
+
+    # Determine a sensible suffix from the uploaded filename (fall back to .wav)
+    try:
+        ext = os.path.splitext(uploaded_file.name)[1]
+    except Exception:
+        ext = ".wav"
+    if not ext:
+        ext = ".wav"
+
+    # Save the uploaded audio file to a temporary location for librosa processing
+    with tempfile.NamedTemporaryFile(delete=False, suffix=ext) as temp_audio_file:
+        temp_audio_file.write(file_bytes)
         temp_audio_path = temp_audio_file.name
 
+    # Provide an audio player in Streamlit using the raw bytes
+    try:
+        st.audio(file_bytes, format=(uploaded_file.type if hasattr(uploaded_file, "type") else None))
+    except Exception:
+        # Fallback: try to let Streamlit infer format
+        st.audio(file_bytes)
+
     # Preprocess the audio into a spectrogram
     st.write("Processing audio into a spectrogram...")
     spectrogram = preprocess_audio(temp_audio_path)
@@ -102,8 +120,11 @@ if uploaded_file is not None:
     else:
         st.write("Failed to process the audio file. Please try again with a different file.")
 
-    # Optional: Clean up temporary file
-    os.remove(temp_audio_path)
+    # Clean up temporary file
+    try:
+        os.remove(temp_audio_path)
+    except Exception:
+        pass