Create app.py

app.py

@@ -0,0 +1,45 @@
+from fastapi import FastAPI, Response
+from fastapi.middleware.cors import CORSMiddleware
+from pocket_tts import TTSModel
+import scipy.io.wavfile
+import io
+import torch
+
+app = FastAPI()
+
+# IMPORTANT : Autoriser votre site web à accéder à ce serveur
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# --- CHARGEMENT DU MODÈLE (Au démarrage) ---
+print("⏳ Chargement de Kyutai Pocket TTS...")
+try:
+    tts_model = TTSModel.load_model()
+    # On charge la voix "Alba" par défaut
+    voice_state = tts_model.get_state_for_audio_prompt("hf://kyutai/tts-voices/alba-mackenna/casual.wav")
+    print("✅ Modèle chargé !")
+except Exception as e:
+    print(f"❌ Erreur au chargement du modèle: {e}")
+
+@app.post("/tts")
+async def generate_speech(data: dict):
+    text = data.get("text", "")
+    
+    # Génération
+    audio_tensor = tts_model.generate_audio(voice_state, text)
+    
+    # Conversion en fichier WAV en mémoire
+    buffer = io.BytesIO()
+    scipy.io.wavfile.write(buffer, tts_model.sample_rate, audio_tensor.numpy())
+    buffer.seek(0)
+    
+    return Response(content=buffer.read(), media_type="audio/wav")
+
+@app.get("/")
+def home():
+    return {"status": "Pocket TTS is running"}
+