ocr / app.py
GAS17's picture
Create app.py
990b2f4 verified
raw
history blame
663 Bytes
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()