import gradio as gr
import pdfquery

def consultar_pdf(pdf_file, consulta):
    # Cargar el PDF
    pdf = pdfquery.PDFQuery(pdf_file.name)
    pdf.load()
    
    # Realizar la consulta
    resultado = pdf.pq(consulta).text()

    # Devolver el resultado de la consulta
    return resultado

# Crear la interfaz de Gradio
iface = gr.Interface(
    fn=consultar_pdf, 
    inputs=[
        gr.inputs.File(label="Cargar PDF"),  # Entrada para cargar el archivo PDF
        gr.inputs.Textbox(label="Consulta", default="")  # Entrada para la consulta
    ], 
    outputs="text"  # Salida de texto con el resultado de la consulta
)

# Lanzar la interfaz
iface.launch()