Zw07 commited on
Commit
7b3cbf2
·
verified ·
1 Parent(s): 8790743

upload mp3 format audio

Browse files
Files changed (1) hide show
  1. app.py +18 -2
app.py CHANGED
@@ -55,11 +55,27 @@ def main():
55
 
56
  if audio_file:
57
  # 保存文件到本地:
 
 
 
 
 
 
 
58
  tmp_input_audio_file = os.path.join("/tmp/", audio_file.name)
59
- st.markdown(tmp_input_audio_file)
60
  with open(tmp_input_audio_file, "wb") as f:
61
  f.write(audio_file.getbuffer())
62
- st.audio(tmp_input_audio_file, format="mp3/wav")
 
 
 
 
 
 
 
 
 
 
63
 
64
  wav, sample_rate = torchaudio.load(audio_file, format="mp3/wav")
65
  st.markdown("SR")
 
55
 
56
  if audio_file:
57
  # 保存文件到本地:
58
+ # tmp_input_audio_file = os.path.join("/tmp/", audio_file.name)
59
+ # st.markdown(tmp_input_audio_file)
60
+ # with open(tmp_input_audio_file, "wb") as f:
61
+ # f.write(audio_file.getbuffer())
62
+ # st.audio(tmp_input_audio_file, format="mp3/wav")
63
+
64
+ # Save file to local storage
65
  tmp_input_audio_file = os.path.join("/tmp/", audio_file.name)
 
66
  with open(tmp_input_audio_file, "wb") as f:
67
  f.write(audio_file.getbuffer())
68
+
69
+ # Convert MP3 to WAV using pydub
70
+ mp3_audio = AudioSegment.from_mp3(tmp_input_audio_file)
71
+ wav_output_file = tmp_input_audio_file.replace(".mp3", ".wav")
72
+ mp3_audio.export(wav_output_file, format="wav")
73
+
74
+ # Load the WAV file using torchaudio
75
+ wav, sample_rate = torchaudio.load(wav_output_file)
76
+
77
+ # Play the audio file (WAV format)
78
+ st.audio(wav_output_file, format="audio/wav")
79
 
80
  wav, sample_rate = torchaudio.load(audio_file, format="mp3/wav")
81
  st.markdown("SR")