Update app.py

app.py

@@ -128,10 +128,19 @@ def main():
 
         # Save file to local storage
         tmp_input_audio_file = os.path.join("/tmp/", audio_file.name)
-        with open("test.wav", "wb") as f:
-            f.write(audio_file.getbuffer())
+        file_extension = os.path.splitext(audio_file)[1].lower()
+        st.markdown(file_extension)
+        if file_extension in [".wav", ".flac"]:
+            with open("test.wav", "wb") as f:
+                f.write(audio_file.getbuffer())
 
-        st.audio("test.wav", format="audio/wav")
+            st.audio("test.wav", format="audio/wav")
+
+        elif file_extension == ".mp3":
+            with open("test.mp3", "wb") as f:
+                f.write(audio_file.getbuffer())
+
+            st.audio("test.wav", format="audio/mpeg")
 
         # # Convert MP3 to WAV using pydub
         # mp3_audio = AudioSegment.from_mp3(tmp_input_audio_file)