import gradio as gr
from transformers import pipeline

# Modelo mejor que el DeepESP/gpt2-spanish  
generador = pipeline(
    task="text-generation",
    model="DeepESP/gpt2-spanish-medium",
    tokenizer="DeepESP/gpt2-spanish-medium"
)

def generar_consejo(tema):
    prompt_contexto = f"Consejo útil de salud/fuerza sobre {tema}:"

    resultado = generador(
        prompt_contexto,
        max_length=180,     # un poco más largo para textos más completos
        do_sample=True,
        temperature=0.8
    )

    return resultado[0]["generated_text"]

demo = gr.Interface(
    fn=generar_consejo,
    inputs=gr.Textbox(lines=2, placeholder="Ej: hidratación, fuerza, dormir bien...", label="Tema del consejo"),
    outputs=gr.Textbox(lines=8, label="Consejo generado"),
    title="Generador de Consejos Fitness (Mejorado)",
    description="Introduce un tema de salud o deporte y la IA te dará un consejo práctico."
)

if __name__ == "__main__":
    demo.launch()