Create app.py

app.py

@@ -0,0 +1,76 @@
+from fastapi import FastAPI, Response, HTTPException
+from fastapi.middleware.cors import CORSMiddleware
+from melo.api import TTS
+import os
+
+app = FastAPI()
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# Stockage des modèles chargés
+models = {}
+device = 'cpu'
+
+print("⏳ Initialisation de MeloTTS...")
+
+try:
+    # On pré-charge le Français et l'Anglais
+    # C'est rapide et léger avec Melo
+    print("  - Chargement FR...")
+    models['FR'] = TTS(language='FR', device=device)
+    
+    print("  - Chargement EN...")
+    models['EN'] = TTS(language='EN', device=device)
+    
+    print("✅ MeloTTS Prêt (FR + EN) !")
+except Exception as e:
+    print(f"❌ Erreur au chargement : {e}")
+
+@app.post("/tts")
+async def generate_speech(data: dict):
+    text = data.get("text", "")
+    lang = data.get("lang", "FR").upper() # Par défaut FR
+    speed = data.get("speed", 1.0)
+
+    if not text:
+        raise HTTPException(status_code=400, detail="Texte vide")
+
+    # Vérification si la langue est chargée
+    if lang not in models:
+        # Tentative de chargement dynamique (si vous demandez ES ou ZH par exemple)
+        try:
+            print(f"⏳ Chargement dynamique de {lang}...")
+            models[lang] = TTS(language=lang, device=device)
+        except Exception as e:
+            raise HTTPException(status_code=400, detail=f"Langue {lang} non supportée ou erreur: {e}")
+
+    model = models[lang]
+    speaker_ids = model.hps.data.spk2id
+    
+    # MeloTTS utilise souvent le code langue comme ID speaker (ex: 'FR')
+    # Mais parfois c'est 'EN-US', 'EN-BR', etc. pour l'anglais.
+    # On prend le premier dispo si la clé exacte n'existe pas.
+    spk_id = speaker_ids.get(lang, list(speaker_ids.values())[0])
+
+    output_path = f"/tmp/output_{os.getpid()}.wav"
+
+    try:
+        model.tts_to_file(text, spk_id, output_path, speed=speed)
+        
+        with open(output_path, "rb") as f:
+            audio_data = f.read()
+            
+        return Response(content=audio_data, media_type="audio/wav")
+
+    except Exception as e:
+        print(f"❌ Erreur génération : {e}")
+        return Response(content=str(e), status_code=500)
+        
+@app.get("/")
+def home():
+    return {"status": "MeloTTS API Ready", "loaded_langs": list(models.keys())}