import gradio as gr
from transformers import pipeline

# Cargar un modelo preentrenado de Hugging Face
model = pipeline("text-generation", model="gpt2")

# Función para generar texto basado en una pregunta
def generar_texto(pregunta):
    respuesta = model(pregunta, max_length=200, num_return_sequences=1)
    return respuesta[0]["generated_text"]

# Interfaz de usuario con Gradio
interfaz = gr.Interface(
    fn=generar_texto,  # Función que se ejecuta
    inputs="text",     # Entrada: texto (pregunta)
    outputs="text",    # Salida: texto (respuesta)
    title="Asistente Legal y Docente",
    description="Haz preguntas sobre legislación o temas docentes."
)

# Lanzar la aplicación
interfaz.launch()