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() | |