Update app.py

app.py

@@ -6,6 +6,7 @@ from huggingface_hub import login
 from diffusers import StableAudioPipeline
 import os
 import io
+import tempfile
 
 # Load Hugging Face token securely
 HUGGINGFACE_TOKEN = os.getenv("HF_TOKEN")
@@ -45,6 +46,18 @@ def generate_audio():
             generator=generator
         ).audios
 
+        # Create a temporary file to store the generated audio
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as tmpfile:
+            # Write the audio to the temporary file
+            sf.write(tmpfile.name, audio_output.T.cpu().numpy(), pipe.vae.sampling_rate)
+
+            # Store the temporary file path
+            temp_file_path = tmpfile.name
+
+        # Create a downloadable URL by returning the temporary file path
+        return jsonify({"download_url": f"/download/{os.path.basename(temp_file_path)}"}), 200
+        
+        '''
         # Convert audio to BytesIO in memory
         output_io = io.BytesIO()
         output_audio = audio_output[0].T.float().cpu().numpy()
@@ -53,6 +66,7 @@ def generate_audio():
 
         # Send the file in response as attachment for download
         return send_file(output_io, as_attachment=False, download_name="output.wav", mimetype='audio/wav')
+        '''
         
     except Exception as e:
         return jsonify({"error": str(e)}), 500