import gradio as gr
from transformers import pipeline

# Definir función para obtener la respuesta
def obtener_respuesta(contexto, pregunta):
    # Cargar el modelo pre-entrenado
    model_name = "deepset/roberta-base-squad2"
    nlp = pipeline('question-answering', model=model_name, tokenizer=model_name)
    
    # Obtener la respuesta predicha
    res = nlp({'question': pregunta, 'context': contexto})
    return res['answer']

# Crear la interfaz de Gradio
interfaz = gr.Interface(
    fn=obtener_respuesta,
    inputs=[
        gr.Textbox(label="Contexto"),
        gr.Textbox(label="Pregunta")
    ],
    outputs=gr.Textbox(label="Respuesta"),
    title="QA con Gradio",
    description="Introduce el contexto y la pregunta para obtener la respuesta.",
    theme="glass"
)

# Lanzar la interfaz
interfaz.launch(share=True)