Update xtts.py

xtts.py

@@ -48,6 +48,8 @@ def predict(text, sample=None, language="zh"):
             file_path=output
         )
 
+        output=to_mp3(output)
+        
         return upload(output)
 
         with io.BytesIO() as wav_buffer:
@@ -135,6 +137,23 @@ def trim_sample_audio(speaker_wav):
         traceback.print_exc()
         return speaker_wav
 
+def to_mp3(wav):
+    global ffmpeg
+    try:
+        mp3=tempfile.mktemp(suffix=".mp3")
+        shell_command = f"{ffmpeg} -i {wav} {mp3}".split(" ")
+        subprocess.run(
+            [item for item in shell_command],
+            capture_output=False,
+            text=True,
+            check=True,
+        )
+        return mp3
+    except:
+        traceback.print_exc()
+        return wav
+        
+
 from flask import Flask, request
 app = Flask(__name__)
 @app.route("/tts")