Correction chargement Kokoro avec chemins explicites

app.py

@@ -8,29 +8,47 @@ from fastapi.staticfiles import StaticFiles
 from pydantic import BaseModel
 import soundfile as sf
 import time
+from huggingface_hub import hf_hub_download
 
 print("🚀 Démarrage API Kokoro-82M (ONNX rapide)")
 
 # -------------------------------------------------------------
-# Chargement du modèle via from_pretrained (recommandé)
+# Téléchargement des fichiers du modèle et des voix
 # -------------------------------------------------------------
-from kokoro_onnx import Kokoro
+REPO_ID = "onnx-community/Kokoro-82M-v1.0-ONNX"
+
+# Télécharger les deux fichiers nécessaires
+try:
+    model_path = hf_hub_download(
+        repo_id=REPO_ID,
+        filename="model.onnx",
+        cache_dir="/app/cache"
+    )
+    voices_path = hf_hub_download(
+        repo_id=REPO_ID,
+        filename="voices.json",
+        cache_dir="/app/cache"
+    )
+    print(f"✅ Modèle téléchargé : {model_path}")
+    print(f"✅ Voices téléchargées : {voices_path}")
+except Exception as e:
+    raise RuntimeError(f"Échec du téléchargement : {e}")
 
-# Modèle par défaut (sera téléchargé automatiquement depuis HF)
-MODEL_NAME = "onnx-community/Kokoro-82M-v1.0-ONNX"
+# -------------------------------------------------------------
+# Chargement de Kokoro avec les chemins explicites
+# -------------------------------------------------------------
+from kokoro_onnx import Kokoro
 
-tts = Kokoro.from_pretrained(MODEL_NAME, device="cpu")
+tts = Kokoro(model_path=model_path, voices_path=voices_path)
 print("✅ Modèle Kokoro chargé avec succès")
 
-# Dictionnaire pour les voix clonées (si supporté)
+# Dictionnaire pour les voix clonées (optionnel)
 cloned_voices = {}
 
 # -------------------------------------------------------------
 # Application FastAPI
 # -------------------------------------------------------------
-app = FastAPI(title="Kokoro TTS API", version="2.2")
-
-# Servir l'interface HTML (optionnel)
+app = FastAPI(title="Kokoro TTS API", version="2.3")
 app.mount("/static", StaticFiles(directory="static"), name="static")
 
 @app.get("/")
@@ -38,6 +56,7 @@ async def root():
     return {
         "status": "ok",
         "model": "Kokoro-82M (ONNX)",
+        "voices_available": list(tts.voices.keys()) if hasattr(tts, 'voices') else [],
         "endpoints": ["/tts", "/clone-voice", "/voices"]
     }
 
@@ -47,7 +66,7 @@ async def demo():
 
 class TTSRequest(BaseModel):
     text: str
-    voice: str = "af_heart"   # voix par défaut (anglais)
+    voice: str = "af_heart"
     speed: float = 1.0
     lang: str = "fr"
 
@@ -58,25 +77,22 @@ async def generate_speech(request: TTSRequest):
     if not text:
         raise HTTPException(400, "Texte vide")
 
-    # Choix de la voix : clonée ou intégrée
     voice = request.voice
     if voice in cloned_voices:
-        voice = cloned_voices[voice]  # peut être un chemin ou un identifiant
+        voice = cloned_voices[voice]
 
     try:
-        # Génération audio
         audio, sample_rate = tts.tts(
             text=text,
             voice=voice,
             speed=request.speed,
             lang=request.lang
         )
-        # Sauvegarde en mémoire
         buffer = io.BytesIO()
         sf.write(buffer, audio, sample_rate, format='WAV')
         buffer.seek(0)
 
-        print(f"✅ Généré en {time.time()-start:.2f}s | voix={voice} | langue={request.lang}")
+        print(f"✅ Généré en {time.time()-start:.2f}s | voix={voice}")
         return Response(content=buffer.read(), media_type="audio/wav")
     except Exception as e:
         raise HTTPException(500, f"Erreur TTS: {str(e)}")
@@ -86,21 +102,18 @@ async def clone_voice(audio_file: UploadFile = File(...), voice_name: str = Form
     if voice_name in cloned_voices:
         raise HTTPException(400, f"La voix '{voice_name}' existe déjà")
 
-    # Sauvegarde du fichier temporaire
     with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as tmp:
         content = await audio_file.read()
         tmp.write(content)
         tmp_path = tmp.name
 
     try:
-        # Certaines versions supportent clone_voice
         if hasattr(tts, 'clone_voice'):
             cloned_voice = tts.clone_voice(tmp_path)
+            cloned_voices[voice_name] = cloned_voice
+            return {"message": f"Voix '{voice_name}' clonée", "voice_name": voice_name}
         else:
-            # Fallback : on stocke le chemin (fonctionne si l'API l'accepte)
-            cloned_voice = tmp_path
-        cloned_voices[voice_name] = cloned_voice
-        return {"message": f"Voix '{voice_name}' clonée", "voice_name": voice_name, "total": len(cloned_voices)}
+            raise HTTPException(501, "Clonage non supporté par cette version de kokoro-onnx")
     except Exception as e:
         raise HTTPException(500, f"Erreur clonage: {str(e)}")
     finally:
@@ -109,7 +122,6 @@ async def clone_voice(audio_file: UploadFile = File(...), voice_name: str = Form
 
 @app.get("/voices")
 async def list_voices():
-    # Les voix intégrées sont dans tts.voices (si dispo)
     builtin = list(tts.voices.keys()) if hasattr(tts, 'voices') else []
     return {"builtin": builtin, "cloned": list(cloned_voices.keys())}