Update app.py

app.py

@@ -5,11 +5,13 @@ os.environ["TOKENIZERS_PARALLELISM"] = "false"
 from PIL import Image, ImageDraw
 import traceback
 
+import pandas as pd
+
 import gradio as gr
 
 import torch
 from docquery import pipeline
-from docquery.document import load_document, ImageDocument
+from docquery.document import load_document, ImageDocumenta
 from docquery.ocr_reader import get_ocr_reader
 
 
@@ -87,8 +89,9 @@ examples = [
     ],
     [
         "SaleData.xlsx",
-        
-    ],
+        "What is the highest sale amount of televsion in east region?",
+
+    ]
     #    [
     #        "docquery.png",
     #        "How many likes does the space have?",
@@ -133,9 +136,12 @@ def process_path(path):
     )
 
 
-def process_upload(file):
+def process_upload(file, excel_file):
     if file:
         return process_path(file.name)
+    if excel_file:
+        excel_data = pd.read_excel(excel_file)
+        return process_path(excel_file.name)
     else:
         return (
             None,
@@ -145,6 +151,10 @@ def process_upload(file):
             gr.update(visible=False, value=None),
             None,
         )
+    
+
+    
+
 
 
 colors = ["#64A087", "green", "black"]
@@ -227,7 +237,7 @@ CSS = """
     padding-bottom: 2px !important;
     padding-left: 8px !important;
     padding-right: 8px !important;
-	margin-top: 10px;
+        margin-top: 10px;
 }
 .gradio-container .gr-button-primary {
     background: linear-gradient(180deg, #CDF9BE 0%, #AFF497 100%);
@@ -293,19 +303,15 @@ gradio-app h2, .gradio-app h2 {
 """
 
 with gr.Blocks(css=CSS) as demo:
-    gr.Markdown("# DocQuery: Document Query Engine")
-    gr.Markdown(
-        "DocQuery (created by [Impira](https://impira.com?utm_source=huggingface&utm_medium=referral&utm_campaign=docquery_space))"
-        " uses LayoutLMv1 fine-tuned on DocVQA, a document visual question"
-        " answering dataset, as well as SQuAD, which boosts its English-language comprehension."
-        " To use it, simply upload an image or PDF, type a question, and click 'submit', or "
-        " click one of the examples to load them."
-        " DocQuery is MIT-licensed and available on [Github](https://github.com/impira/docquery)."
-    )
+    gr.Markdown("# Document Query Engine")
+    
 
     document = gr.Variable()
     example_question = gr.Textbox(visible=False)
     example_image = gr.Image(visible=False)
+    excel_upload = gr.File(label="Upload Excel", type="xlsx", elem_id="excel-upload-box")
+
+    excel_process_button = gr.Button("Process Excel", variant="primary", elem_id="excel-process-button")
 
     with gr.Row(equal_height=True):
         with gr.Column():
@@ -428,6 +434,11 @@ with gr.Blocks(css=CSS) as demo:
         inputs=[example_image, example_question, model],
         outputs=[document, question, image, img_clear_button, output, output_text],
     )
+    upload.change(
+    fn=process_upload,
+    inputs=[upload, excel_upload],
+    outputs=[document, image_preview, img_clear_button, output, output_text, url_error],
+    )
 
 if __name__ == "__main__":
-    demo.launch(enable_queue=False)
+    demo.launch(enable_queue=False)