Update app.py

app.py

@@ -5,7 +5,7 @@ import soundfile as sf
 from huggingface_hub import login
 from diffusers import StableAudioPipeline
 import os
-#import tempfile
+import io
 
 # Load Hugging Face token securely
 HUGGINGFACE_TOKEN = os.getenv("HF_TOKEN")
@@ -22,8 +22,8 @@ pipe = StableAudioPipeline.from_pretrained("stabilityai/stable-audio-open-1.0",
 pipe = pipe.to(device)
 
 # Path to store generated audio files
-OUTPUT_PATH = "./generated_audio"
-os.makedirs(OUTPUT_PATH, exist_ok=True)
+#OUTPUT_PATH = "./generated_audio"
+#os.makedirs(OUTPUT_PATH, exist_ok=True)
 
 # Initialize Flask app
 app = Flask(__name__)
@@ -48,17 +48,29 @@ def generate_audio():
             num_waveforms_per_prompt=1,
             generator=generator
         ).audios
-        
+
+        # Convert audio to BytesIO in memory
+        output_io = io.BytesIO()
+        output_audio = audio_output[0].T.float().cpu().numpy()
+        sf.write(output_io, output_audio, pipe.vae.sampling_rate)  # Save as WAV or your preferred format
+        output_io.seek(0)  # Reset buffer pointer to beginning
+
+        # Send the file in response as attachment for download
+        return send_file(output_io, as_attachment=True, download_name="output.wav", mimetype='audio/wav')
+
+        '''
         # Convert to numpy and save to a WAV file
         output_audio = audio_output[0].T.float().cpu().numpy()
         output_filename = "output.wav"
-        output_path = os.path.join(OUTPUT_PATH, output_filename)
+        #output_path = os.path.join(OUTPUT_PATH, output_filename)
         sf.write(output_path, output_audio, pipe.vae.sampling_rate)
         
         if os.path.exists(output_path):
             return send_file(output_path, as_attachment=True)
         else:
             return {"error": f"File not found at {output_path}"}, 404
+
+        '''
         
         # Return the WAV file
         #return jsonify({"file_path": absolute_output_path}), 200