Update app.py

app.py

@@ -55,8 +55,10 @@ def generate_audio():
         output_path = os.path.join(OUTPUT_PATH, output_filename)
         sf.write(output_path, output_audio, pipe.vae.sampling_rate)
         
-        # Send the file for download
-        return send_file(output_path, as_attachment=True)
+        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