upload mp3 format audio

app.py

@@ -55,11 +55,27 @@ def main():
 
     if audio_file:
         # 保存文件到本地：
+        # tmp_input_audio_file = os.path.join("/tmp/", audio_file.name)
+        # st.markdown(tmp_input_audio_file)
+        # with open(tmp_input_audio_file, "wb") as f:
+        #     f.write(audio_file.getbuffer())
+        # st.audio(tmp_input_audio_file, format="mp3/wav")
+
+        # Save file to local storage
         tmp_input_audio_file = os.path.join("/tmp/", audio_file.name)
-        st.markdown(tmp_input_audio_file)
         with open(tmp_input_audio_file, "wb") as f:
             f.write(audio_file.getbuffer())
-        st.audio(tmp_input_audio_file, format="mp3/wav")
+
+        # Convert MP3 to WAV using pydub
+        mp3_audio = AudioSegment.from_mp3(tmp_input_audio_file)
+        wav_output_file = tmp_input_audio_file.replace(".mp3", ".wav")
+        mp3_audio.export(wav_output_file, format="wav")
+
+        # Load the WAV file using torchaudio
+        wav, sample_rate = torchaudio.load(wav_output_file)
+
+        # Play the audio file (WAV format)
+        st.audio(wav_output_file, format="audio/wav")
         
         wav, sample_rate = torchaudio.load(audio_file, format="mp3/wav")
         st.markdown("SR")