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 tempfile
 
 # Load Hugging Face token securely
 HUGGINGFACE_TOKEN = os.getenv("HF_TOKEN")
@@ -48,28 +48,18 @@ def generate_audio():
             num_waveforms_per_prompt=1,
             generator=generator
         ).audios
-        '''
+        
         # 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)
         sf.write(output_path, output_audio, pipe.vae.sampling_rate)
 
+        # Absolute file path
+        absolute_output_path = os.path.abspath(output_path)
+        
         # Return the WAV file
         return jsonify({"file_path": output_path}), 200
-        '''
-        # Convert to numpy
-        output_audio = audio_output[0].T.float().cpu().numpy()
-
-        # Create a temporary file to store the output audio
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as temp_file:
-            temp_file_path = temp_file.name
-            # Save the audio to the temporary file
-            sf.write(temp_file_path, output_audio, pipe.vae.sampling_rate)
-            
-            # Return the file as an attachment
-            return send_file(temp_file_path, as_attachment=True, download_name="output.wav", mimetype="audio/wav")
-
 
         
     except Exception as e: