edit server

server.py

@@ -2,6 +2,7 @@ from flask import Flask, request, send_file, jsonify, after_this_request
 from TTS.api import TTS
 import tempfile
 import os
+import uuid
 
 app = Flask(__name__)
 
@@ -9,42 +10,42 @@ app = Flask(__name__)
 MODEL_NAME = "tts_models/multilingual/multi-dataset/xtts_v2"
 tts = TTS(model_name=MODEL_NAME, gpu=False, progress_bar=False)
 
-voices = tts.speakers
-female_voice = voices[0]
-male_voice = voices[1] if len(voices) > 1 else voices[0]
 
-
-@app.route("/tts", methods=["GET"])
+@app.route("/tts", methods=["POST"])
 def tts_api():
-    text = request.args.get("text", "")
-    gender = request.args.get("gender", "female").lower()
-    language = request.args.get("lang", "en")
+    text = request.form.get("text", "")
+    language = request.form.get("lang", "en")
+    voice_file = request.files.get("voice")
 
     if not text:
         return jsonify({"error": "Text is required"}), 400
 
-    speaker = female_voice if gender == "female" else male_voice
+    if not voice_file:
+        return jsonify({"error": "Voice file is required"}), 400
+
+    voice_path = f"/tmp/{uuid.uuid4()}.wav"
+    out_path = f"/tmp/{uuid.uuid4()}.wav"
 
-    tmp = tempfile.NamedTemporaryFile(suffix=".wav", delete=False)
-    tmp_path = tmp.name
-    tmp.close()
+    voice_file.save(voice_path)
 
     tts.tts_to_file(
         text=text,
-        speaker=speaker,
+        speaker_wav=voice_path,
         language=language,
-        file_path=tmp_path
+        file_path=out_path
     )
 
+    os.remove(voice_path)
+
     @after_this_request
     def cleanup(response):
         try:
-            os.remove(tmp_path)
+            os.remove(out_path)
         except Exception as e:
             print("Cleanup error:", e)
         return response
 
-    return send_file(tmp_path, mimetype="audio/wav")
+    return send_file(out_path, mimetype="audio/wav")
 
 
 if __name__ == "__main__":