Update app.py

app.py

@@ -126,49 +126,52 @@ 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)
-        file_extension = os.path.splitext(tmp_input_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())
+        #1st attempt
+        audio_path = "C:\Users\Zw\Downloads\example.mp3"
 
-            st.audio("test.wav", format="audio/wav")
+        audio, sr = torchaudio.load(audio_path)
+        st.audio(audio_path, format="audio/mpeg")
+        audio= audio.unsqueeze(0)
+        
+        
+        #2nd attempt
+        # Save file to local storage
+        # tmp_input_audio_file = os.path.join("/tmp/", audio_file.name)
+        # file_extension = os.path.splitext(tmp_input_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())
 
-        elif file_extension == ".mp3":
-            with open("test.mp3", "wb") as f:
-                f.write(audio_file.getbuffer())
+        #     st.audio("test.wav", format="audio/wav")
 
-            st.audio("test.mp3", format="audio/mpeg")
+        # elif file_extension == ".mp3":
+        #     with open("test.mp3", "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")
+        #     st.audio("test.mp3", format="audio/mpeg")
 
         # Load the WAV file using torchaudio
-        if file_extension in [".wav", ".flac"]: 
-            wav, sample_rate = torchaudio.load("test.wav")
-            st.markdown("Before unsquueze wav")
-            st.markdown(wav)
+        # if file_extension in [".wav", ".flac"]: 
+        #     wav, sample_rate = torchaudio.load("test.wav")
+        #     st.markdown("Before unsquueze wav")
+        #     st.markdown(wav)
 
-            #Unsqueeze for line 176
-            wav= wav.unsqueeze(0)
+        #     #Unsqueeze for line 176
+        #     wav= wav.unsqueeze(0)
                 
-        elif file_extension == ".mp3":
-            wav3, sample_rate = librosa.load("test.mp3")
-            st.markdown(wav3)
-            #RuntimeError: Could not infer dtype of numpy.float32
-            #wav = torch.tensor(wav3).float() / 32768.0
+        # elif file_extension == ".mp3":
+        #     wav3, sample_rate = librosa.load("test.mp3")
+        #     st.markdown(wav3)
+        #     #RuntimeError: Could not infer dtype of numpy.float32
+        #     #wav = torch.tensor(wav3).float() / 32768.0
             
-            #RuntimeError: Numpy is not available
-            wav = torch.from_numpy(wav3) #/32768.0
-            wav = wav.unsqueeze(0).unsqueeze(0)
-            st.markdown("Before unsqueeze mp3")
-            st.markdown(wav)
+        #     #RuntimeError: Numpy is not available
+        #     wav = torch.from_numpy(wav3) #/32768.0
+        #     wav = wav.unsqueeze(0).unsqueeze(0)
+        #     st.markdown("Before unsqueeze mp3")
+        #     st.markdown(wav)
         
         # #Unsqueeze for line 176
         # wav= wav.unsqueeze(0)