Update app.py

app.py

@@ -25,6 +25,10 @@ pipe = pipe.to(device)
 # Initialize Flask app
 app = Flask(__name__)
 
+# Set up a folder for public files (you can choose a different directory)
+PUBLIC_DIRECTORY = os.path.join(os.getcwd(), 'static', 'generated_audio')
+os.makedirs(PUBLIC_DIRECTORY, exist_ok=True)
+
 # Route to generate audio
 
 @app.route("/generate", methods=["GET"])
@@ -46,17 +50,16 @@ def generate_audio():
             generator=generator
         ).audios
 
-        # Create a temporary file to store the generated audio
-        output_audio = audio_output[0].T.float().cpu().numpy()
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as tmpfile:
-            # Write the audio to the temporary file
-            sf.write(tmpfile.name, output_audio, pipe.vae.sampling_rate)
+        # Create an absolute file path in the "static/generated_audio" directory
+        output_filename = f"{prompt.replace(' ', '_')}_output.wav"
+        output_file_path = os.path.join(PUBLIC_DIRECTORY, output_filename)
 
-            # Store the temporary file path
-            temp_file_path = tmpfile.name
+        # Write the generated audio to the file
+        sf.write(output_file_path, audio_output.T.cpu().numpy(), pipe.vae.sampling_rate)
 
-        # Create a downloadable URL by returning the temporary file path
-        return jsonify({"download_url": f"/download/{os.path.basename(temp_file_path)}"}), 200
+        # Return the absolute URL to the file
+        download_url = f"/download/{output_filename}"
+        return jsonify({"download_url": download_url}), 200
         
         '''
         # Convert audio to BytesIO in memory