Update app.py

app.py

@@ -26,20 +26,25 @@ if uploaded_file:
     # Convert audio to tensor
     audio_tensor = torch.tensor(audio).float()
 
-    with st.spinner("Inverting audio..."):
-        # Invert the audio using the modified function
-        inverted_audio_tensor = invert_audio(model, processor, audio_tensor, sr)
-        
-        # Convert tensor back to numpy
-        inverted_audio_np = inverted_audio_tensor.numpy()
-
-        # Convert numpy audio to bytes and play
-        with io.BytesIO() as out_io:
-            sf.write(out_io, inverted_audio_np, sr, format="wav")
-            st.audio(out_io.getvalue(), format="audio/wav")
+    # Use Streamlit's session state to prevent re-inversion
+    if "inverted_audio" not in st.session_state:
+        with st.spinner("Inverting audio..."):
+            # Invert the audio using the modified function
+            inverted_audio_tensor = invert_audio(model, processor, audio_tensor, sr)
+            
+            # Convert tensor back to numpy
+            inverted_audio_np = inverted_audio_tensor.numpy()
+
+            # Store inverted audio in session state
+            st.session_state.inverted_audio = inverted_audio_np
+
+    # Play inverted audio from session state
+    with io.BytesIO() as out_io:
+        sf.write(out_io, st.session_state.inverted_audio, sr, format="wav")
+        st.audio(out_io.getvalue(), format="audio/wav")
 
     # Offer a download button for the inverted audio
     if st.button("Download Inverted Audio"):
         with io.BytesIO() as out_io:
-            sf.write(out_io, inverted_audio_np, sr, format="wav")
+            sf.write(out_io, st.session_state.inverted_audio, sr, format="wav")
             st.download_button("Download Inverted Audio", data=out_io.getvalue(), file_name="inverted_output.wav", mime="audio/wav")