Update app.py

app.py

@@ -1,27 +1,29 @@
 import gradio as gr
 from gradio_pdf import PDF
-from pypdf import PdfReader
+from pdf2image import convert_from_path
+from transformers import pipeline
+from pathlib import Path
 
-def upload_file(files):
-    #file_paths = [file.name for file in files]
-    file_paths = files
-    return file_paths
-    
-def summarise_file(file):
-   # text = PdfReader(file)
-    
-    return file
+dir_ = Path(__file__).parent
 
-with gr.Blocks() as demo:
-    file_output = gr.File()
-    upload_button = gr.UploadButton("Click to Upload a File", file_types=["pdf", "doc"], file_count="single")
-    upload_button.upload(upload_file, upload_button, file_output)
-    
-    file_summary = gr.Textbox()			
-    sum_button = gr.Button("Click to summarise file")
-    
-    sum_button.click(summarise_file, file_output, file_summary)
+p = pipeline(
+    "document-question-answering",
+    model="impira/layoutlm-document-qa",
+)
 
+def qa(question: str, doc: str) -> str:
+    img = convert_from_path(doc)[0]
+    output = p(img, question)
+    return sorted(output, key=lambda x: x["score"], reverse=True)[0]['answer']
+
+
+demo = gr.Interface(
+    qa,
+    [gr.Textbox(label="Question"), PDF(label="Document")],
+    gr.Textbox(),
+    examples=[["What is the total gross worth?", str(dir_ / "invoice_2.pdf")],
+              ["Whos is being invoiced?", str(dir_ / "sample_invoice.pdf")]]
+)
 
 if __name__ == "__main__":
-    demo.queue(max_size=3).launch()
+    demo.launch()