import gradio as gr
from metrology_rag import MetrologyRAGPipeline

rag_pipeline = MetrologyRAGPipeline()

def process_query(query, pdf_files):
    if pdf_files:
        pdf_paths = [file.name for file in pdf_files]  # Works with type="filepath"
        rag_pipeline.load_pdfs(pdf_paths=pdf_paths)
    response = rag_pipeline.query(query)
    return response

interface = gr.Interface(
    fn=process_query,
    inputs=[
        gr.Textbox(label="Digite sua pergunta sobre metrologia"),
        gr.File(label="Carregar arquivos PDF (opcional)", file_count="multiple", type="filepath")
    ],
    outputs=gr.Textbox(label="Resposta do Agente de Metrologia"),
    title="Agente de Metrologia Avançado",
    description="Faça perguntas técnicas sobre metrologia e carregue PDFs para análise."
)

if __name__ == "__main__":
    interface.launch()