# Importar las bibliotecas necesarias
from transformers import pipeline
import gradio as gr

# Crear la pipeline de generación de texto con el modelo español
pipe = pipeline("text-generation", model="datificate/gpt2-small-spanish")

# Función que el chatbot utilizará para responder
def chatbot_response(prompt):
    # Genera la respuesta del modelo
    result = pipe(prompt, max_length=50)
    # Retorna solo el texto de la respuesta generada
    return result[0]['generated_text']

# Crear la interfaz de Gradio
iface = gr.Interface(
    fn=chatbot_response, 
    inputs="text", 
    outputs="text",
    title="Chatbot en Español",
    description="Este chatbot responde a tus preguntas. Está basado en el modelo GPT-2 pequeño en español."
)

# Lanzar la interfaz
iface.launch()