new params

app.py

@@ -26,8 +26,19 @@ try:
 except Exception as e:
     print("Erreur lors du test du modèle :", e)
 
+
+SPEAKER_MAPPING = {
+    "Eden": "p339",
+    "voice2": "p340",
+    # Ajoutez d'autres correspondances si nécessaire
+}
+
+
 class TextInput(BaseModel):
     text: str
+    emotion: str = "neutral"
+    speaker: str = "Eden"
+
 
 @app.post("/tts")
 def generate_tts(input: TextInput):
@@ -35,8 +46,15 @@ def generate_tts(input: TextInput):
     if not input.text.strip():
         raise HTTPException(status_code=400, detail="Le texte ne peut pas être vide.")
     
+    speaker_code = SPEAKER_MAPPING.get(input.speaker)
+    if speaker_code is None:
+        raise HTTPException(
+            status_code=400, detail=f"Le speaker '{input.speaker}' n'est pas reconnu.")
+
+
     try:
-        print("generation starts", input.text.strip())
+        print("Generation starts:", input.text.strip(), "avec emotion:",
+              input.emotion, "et speaker:", input.speaker)
 
         with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as tmpfile:
             output_path = tmpfile.name
@@ -45,8 +63,15 @@ def generate_tts(input: TextInput):
 
         
         # Générer l'audio
-        tts.tts_to_file(text=input.text, speaker="p339", speaker_wav=None, age=22, gender="F", emotion="neutral", file_path=output_path)
-        
+        tts.tts_to_file(
+            text=input.text,
+            speaker=speaker_code,
+            speaker_wav=None,
+            age=22,
+            gender="F",
+            emotion=input.emotion,
+            file_path=output_path
+        )
         print("tmp file generated")
 
         return FileResponse(output_path, media_type="audio/wav", filename="output.wav")