Update app.py

app.py

@@ -1,15 +1,17 @@
 import os
+import io
 
 os.environ["TOKENIZERS_PARALLELISM"] = "false"
 
 from PIL import Image, ImageDraw
 import traceback
 
+
 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
 
 
@@ -86,9 +88,10 @@ examples = [
         "What are net sales for 2020?",
     ],
     [
-        "SaleData.xlsx",
-        
-    ],
+        "saleData.pdf",
+        "What is the highest sale amount of televsion in east region?",
+
+    ]
     #    [
     #        "docquery.png",
     #        "How many likes does the space have?",
@@ -103,6 +106,8 @@ question_files = {
     "What are net sales for 2020?": "statement.pdf",
     "How many likes does the space have?": "https://huggingface.co/spaces/impira/docquery",
     "What is the title of post number 5?": "https://news.ycombinator.com",
+    "What is the highest sale amount of television in east region?": "SaleData.pdf",  # Add the PDF file here
+
 }
 
 
@@ -145,6 +150,10 @@ def process_upload(file):
             gr.update(visible=False, value=None),
             None,
         )
+    
+
+    
+
 
 
 colors = ["#64A087", "green", "black"]
@@ -227,7 +236,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 +302,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():
@@ -429,5 +434,6 @@ with gr.Blocks(css=CSS) as demo:
         outputs=[document, question, image, img_clear_button, output, output_text],
     )
 
+
 if __name__ == "__main__":
     demo.launch(enable_queue=False)