Update app.py

app.py

@@ -6,7 +6,6 @@ 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")
@@ -25,10 +24,6 @@ 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"])
@@ -49,20 +44,7 @@ def generate_audio():
             num_waveforms_per_prompt=1,
             generator=generator
         ).audios
-
-        # Create an absolute file path in the "static/generated_audio" directory
-        output_audio = audio_output[0].T.float().cpu().numpy()
-        output_filename = f"{prompt.replace(' ', '_')}_output.wav"
-        output_file_path = os.path.join(PUBLIC_DIRECTORY, output_filename)
-
-        # Write the generated audio to the file
-        sf.write(output_file_path, output_audio, pipe.vae.sampling_rate)
-
-        # 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
         output_io = io.BytesIO()
         output_audio = audio_output[0].T.float().cpu().numpy()
@@ -71,7 +53,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