Update app.py

app.py

@@ -5,47 +5,38 @@ import gradio as gr
 import spaces
 from pydantic import BaseModel
 from fastapi import HTTPException, Header, Response
+from huggingface_hub import hf_hub_download
 from f5_tts.api import F5TTS
 
 SHARED_SECRET = os.environ.get("PRAIEIRO_SECRET", "praieiro-voz-2026")
 REF_AUDIO_PATH = "PraieiroVox.wav"
 
-print("Carregando modelo F5-TTS para clonagem de voz...")
-f5tts = F5TTS(model_type="F5-TTS_PT")
+print("Carregando modelo F5-TTS em português...")
+ckpt = hf_hub_download("firstpixel/F5-TTS-pt-br", "pt-br/model_last.safetensors")
+f5tts = F5TTS(model="F5TTS_Base", ckpt_file=ckpt)
 
-@spaces.GPU(duration=30)
+@spaces.GPU(duration=60)
 def clone_voice(text: str):
     if not text or not text.strip():
         return None
-    
-    # Gera a fala imitando exatamente o áudio PraieiroVox.wav
     wav, sr, _ = f5tts.infer(
         ref_file=REF_AUDIO_PATH,
         ref_text="",
         gen_text=text.strip(),
-        remove_silence=True
+        remove_silence=True,
     )
     return sr, wav
 
 with gr.Blocks(title="Praieiro Vox · Clonagem de Voz") as demo:
-    gr.Markdown("# 🌊 Praieiro Vox · Clonador de Voz do Agente")
-    gr.Markdown(f"Clonando a voz oficial a partir de **{REF_AUDIO_PATH}** em GPU.")
-    
-    with gr.Row():
-        text_input = gr.Textbox(
-            label="Texto a ser falado",
-            value="Fala meu brother! Seu pedido na praia já tá sendo preparado e chega voando!",
-            lines=3
-        )
-    
+    gr.Markdown("# 🌊 Praieiro Vox · Voz Oficial do Praieiro Bot")
+    text_input = gr.Textbox(
+        label="Texto a ser falado",
+        value="Fala meu brother! Seu pedido na praia já tá sendo preparado e chega voando!",
+        lines=3,
+    )
     btn = gr.Button("🎤 Gerar Fala com a Voz Clonada", variant="primary")
     audio_output = gr.Audio(label="Áudio Gerado", type="numpy")
-    
-    btn.click(
-        fn=clone_voice,
-        inputs=[text_input],
-        outputs=[audio_output]
-    )
+    btn.click(fn=clone_voice, inputs=[text_input], outputs=[audio_output], api_name="clone_voice")
 
 app = demo.app
 
@@ -57,11 +48,9 @@ def generate_speech_endpoint(req: TTSRequest, authorization: str = Header(None))
     token = authorization.replace("Bearer ", "") if authorization else ""
     if token != SHARED_SECRET:
         raise HTTPException(status_code=401, detail="Não autorizado")
-    
     res = clone_voice(req.input)
     if res is None:
         raise HTTPException(status_code=400, detail="Texto vazio ou inválido")
-        
     sr, audio_data = res
     buf = io.BytesIO()
     sf.write(buf, audio_data, sr, format="WAV")