Update app.py

app.py

@@ -1,5 +1,5 @@
 import os
-from flask import Flask, request, jsonify, send_file
+from flask import Flask, request, jsonify, send_file, Response
 import torch
 import torchaudio
 import librosa
@@ -13,6 +13,7 @@ from modules.bigvgan import bigvgan
 from transformers import AutoFeatureExtractor, WhisperModel
 from modules.audio import mel_spectrogram
 from modules.rmvpe import RMVPE
+from io import BytesIO
 
 # Initialize Flask app
 app = Flask(__name__)
@@ -134,13 +135,14 @@ def voice_conversion_api():
     output_wave = (output_wave * 32768.0).astype(np.int16)
 
     # Convert to MP3 and send the response
-    mp3_bytes = AudioSegment(
+    mp3_file = BytesIO()
+    AudioSegment(
         output_wave.tobytes(), frame_rate=sr,
         sample_width=output_wave.dtype.itemsize, channels=1
-    ).export(format="mp3", bitrate="320k").read()
+    ).export(mp3_file, format="mp3", bitrate="320k")
+    mp3_file.seek(0)  # Ensure the stream is at the beginning
 
-    # Send the MP3 as a response
-    return send_file(mp3_bytes, mimetype="audio/mpeg", as_attachment=True, download_name="converted_audio.mp3")
+    return send_file(mp3_file, mimetype="audio/mpeg", as_attachment=True, download_name="converted_audio.mp3")
 
 if __name__ == "__main__":
     # Run the Flask app