Changing Media view logic

app.py

@@ -1,8 +1,8 @@
 import streamlit as st
 from transformers import pipeline
 import scipy.io.wavfile
-import tempfile
 import numpy as np
+import io  # Import the io module for BytesIO
 
 # -----------------------------------------------------------
 def load_css(file_path):
@@ -31,21 +31,19 @@ def main():
             # Generate music with the synthesizer
             music = synthesizer(user_text, forward_params={"do_sample": True, "max_new_tokens": 500})
 
-            # Create a temporary file to store the generated audio
-            with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as tmp:
-                print(f"Temporary file created at: {tmp.name}")
-                
-                # Check if the audio data needs scaling and convert to int16
-                if music["audio"].dtype != np.int16:
-                    audio_data = np.int16(music["audio"] / np.max(np.abs(music["audio"])) * 32767)
-                else:
-                    audio_data = music["audio"]
-                
-                # Write the scaled audio data to the temporary WAV file
-                scipy.io.wavfile.write(tmp.name, rate=music["sampling_rate"], data=audio_data)
-                
-                # Use Streamlit's audio widget to display the audio player
-                st.audio(tmp.name, format="audio/wav")
+            # Check if the audio data needs scaling and convert to int16
+            if music["audio"].dtype != np.int16:
+                audio_data = np.int16(music["audio"] / np.max(np.abs(music["audio"])) * 32767)
+            else:
+                audio_data = music["audio"]
+
+            # Instead of writing to a temp file, write the audio data to a BytesIO object
+            audio_bytes_io = io.BytesIO()
+            scipy.io.wavfile.write(audio_bytes_io, rate=music["sampling_rate"], data=audio_data)
+            audio_bytes_io.seek(0)  # Go to the beginning of the BytesIO object
+
+            # Use Streamlit's audio widget to display the audio player with the BytesIO object
+            st.audio(audio_bytes_io, format="audio/wav")
 
     except Exception as e:
         st.error(f"An error occurred: {e}")