Update app.py

app.py

@@ -86,6 +86,15 @@ def create_default_value():
         def_val_str = "".join([str(i) for i in def_val_npy])
         st.session_state.def_value = def_val_str
 
+def download_sample_audio():
+    url = "https://keithito.com/LJ-Speech-Dataset/LJ037-0171.wav"
+    with open("test.wav", "wb") as f:
+        resp = urllib.request.urlopen(url)
+        f.write(resp.read())
+    
+    wav, sample_rate = torchaudio.load("test.wav")
+    return wav, sample_rate
+
 # Main web app
 def main():
     create_default_value()
@@ -114,24 +123,26 @@ def main():
         #     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)
-        with open(tmp_input_audio_file, "wb") as f:
+        with open("test.wav", "wb") as f:
             f.write(audio_file.getbuffer())
 
-        # 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")
+        # # 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)
+        wav, sample_rate = torchaudio.load("test.wav")
         st.markdown("Before unsquuezewav")
         st.markdown(wav)
         wav= wav.unsqueeze(0)
 
         # Play the audio file (WAV format)
-        st.audio(wav_output_file, format="audio/wav")
+        st.audio(wav, format="audio/wav")
         
         # wav, sample_rate = torchaudio.load(audio_file, format="mp3/wav")
         st.markdown("SR")