import os
import gradio as gr
from TTS.api import TTS

MODEL_PATH = "./best_model.pth"

def tts_infer(text):
    if not os.path.exists(MODEL_PATH):
        return "Modelo no encontrado. Ejecuta primero: python finetune_xtts_peninsular_hf.py", None
    tts = TTS(MODEL_PATH)
    out_file = os.path.abspath("output.wav")
    tts.tts_to_file(text=text, file_path=out_file, language="es")
    return "Audio generado con acento peninsular", out_file

with gr.Blocks() as demo:
    gr.Markdown("# XTTSv2 español peninsular")
    text_input = gr.Textbox(label="Texto en español", lines=3)
    btn = gr.Button("Generar audio")
    status = gr.Label()
    audio_out = gr.Audio(label="Resultado", type="filepath")

    btn.click(fn=tts_infer, inputs=text_input, outputs=[status, audio_out])

demo.launch(server_name="0.0.0.0", server_port=7860)