Update app.py

app.py

@@ -0,0 +1,74 @@
+import io
+import os
+import soundfile as sf
+import torch
+import numpy as np
+from fastapi import FastAPI, HTTPException, Header, Response
+from pydantic import BaseModel
+import spaces
+from kokoro import KPipeline
+
+app = FastAPI(title="Praieiro Voice · Kokoro TTS")
+
+# Token de segurança do Praieiro (para ninguém gastar seu servidor)
+SHARED_SECRET = os.environ.get("PRAIEIRO_SECRET", "praieiro-voz-2026")
+
+# Carrega o pipeline em Português do Brasil ('p' = pt-br)
+print("Inicializando Kokoro-82M para Português do Brasil...")
+pipeline = KPipeline(lang_code="p")
+
+class SpeechRequest(BaseModel):
+    input: str
+    voice: str = "pm_alex"  # pm_alex (masculina natural) ou pf_dora (feminina)
+    speed: float = 1.0
+
+@app.get("/")
+def health():
+    return {
+        "status": "online",
+        "service": "Praieiro Kokoro Voice",
+        "lang": "pt-br",
+        "sample_rate": 24000
+    }
+
+@spaces.GPU
+def generate_audio_gpu(text: str, voice: str, speed: float):
+    generator = pipeline(text, voice=voice, speed=speed, split_pattern=r"\n+")
+    audio_chunks = []
+    for _, _, audio in generator:
+        audio_chunks.append(audio)
+    
+    if not audio_chunks:
+        raise ValueError("Nenhum áudio gerado")
+    
+    full_audio = np.concatenate(audio_chunks)
+    
+    buf = io.BytesIO()
+    sf.write(buf, full_audio, 24000, format="WAV")
+    buf.seek(0)
+    return buf.read()
+
+@app.post("/v1/audio/speech")
+@app.post("/tts")
+def speak(
+    req: SpeechRequest,
+    authorization: str = Header(None)
+):
+    expected = f"Bearer {SHARED_SECRET}"
+    if SHARED_SECRET and authorization != expected:
+        raise HTTPException(status_code=401, detail="Não autorizado")
+
+    text = req.input.strip()
+    if not text:
+        raise HTTPException(status_code=400, detail="Texto vazio")
+
+    try:
+        audio_bytes = generate_audio_gpu(text, req.voice, req.speed)
+        return Response(content=audio_bytes, media_type="audio/wav")
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)
+