import gradio as gr
from transformers import pipeline

# Cargar un modelo de lenguaje
chatbot = pipeline("text-generation", model="gpt2")

# Función de respuesta
def responder_entrada(user_input):
    respuesta = chatbot(user_input, max_length=100, num_return_sequences=1)
    return respuesta[0]["generated_text"]

# Interfaz Gradio
with gr.Blocks() as interfaz:
    gr.Markdown("# Asistente Virtual")
    chat = gr.Chatbot()
    entrada_usuario = gr.Textbox(label="Escribe tu mensaje:")
    boton_enviar = gr.Button("Enviar")
    
    def enviar_mensaje(user_input, history):
        history = history or []
        respuesta = responder_entrada(user_input)
        history.append((user_input, respuesta))
        return history, ""

    boton_enviar.click(enviar_mensaje, [entrada_usuario, chat], [chat, entrada_usuario])

interfaz.launch()