app.py

import gradio as gr
import rag_engine

def ask(query, top_k, umbral):

    docs = rag_engine.recuperar_documentos(query, top_k=top_k, umbral=umbral)
    respuesta = rag_engine.generar_respuesta(query, docs)

    docs_formateados = "\n\n---\n\n".join(docs)

    return respuesta, docs_formateados


with gr.Blocks() as demo:

    gr.Markdown("# Sistema de preguntas con RAG")
    gr.Markdown("Haz una pregunta sobre la base de conocimiento.")

    query = gr.Textbox(label="Tu pregunta (en inglés)", placeholder="Where is the hospital?")

    top_k = gr.Slider(1, 5, value=2, step=1, label="Top-k documentos")

    umbral = gr.Slider(0.0, 1.0, value=0.55, step=0.05, label="Umbral de similitud")

    respuesta = gr.Textbox(label="Respuesta", lines=3)

    docs = gr.Textbox(label="Documentos recuperados", lines=6)

    boton = gr.Button("Enviar")

    boton.click(
        ask,
        inputs=[query, top_k, umbral],
        outputs=[respuesta, docs]
    )

demo.launch()