Update app.py

app.py

@@ -22,7 +22,7 @@ def ensure_list(x):
 
 CHECKPOINTS = {
     "CQI FLMv1 Base PT": "impira/layoutlm-document-qa",
-    "CQI FLMv1 for Invoices 💸": "impira/layoutlm-invoices",
+    "CQI FLMv1 para Faturas 💸": "impira/layoutlm-invoices",
     "CQI FDonut": "naver-clova-ix/donut-base-finetuned-docvqa",
     "CQI Assistant Eyes 🤖": "cloudqi/CQI_Visual_Question_Awnser_PT_v0"
 }
@@ -46,6 +46,8 @@ def run_pipeline(model, question, document, top_k):
     return pipeline(question=question, **document.context, top_k=top_k)
 
 
+# TODO: Move into docquery
+# TODO: Support words past the first page (or window?)
 def lift_word_boxes(document, page):
     return document.context["image"][page][1]
 
@@ -59,6 +61,7 @@ def expand_bbox(word_boxes):
     return [min_x, min_y, max_x, max_y]
 
 
+# LayoutLM boxes are normalized to 0, 1000
 def normalize_bbox(box, width, height, padding=0.005):
     min_x, min_y, max_x, max_y = [c / 1000 for c in box]
     if padding != 0:
@@ -72,22 +75,30 @@ def normalize_bbox(box, width, height, padding=0.005):
 examples = [
     [
         "invoice.png",
-        "What is the invoice number?",
+        "Qual é o número da fatura?",
     ],
     [
         "contract.jpeg",
-        "What is the purchase amount?",
+        "Qual é o valor da compra?",
     ],
     [
         "statement.png",
-        "What are the net sales for 2020?",
+        "Quais são as vendas líquidas para 2020?",
     ],
+    #    [
+    #        "docquery.png",
+    #        "How many likes does the space have?",
+    #    ],
+    #    [
+    #        "hacker_news.png",
+    #        "What is the title of post number 5?",
+    #    ],
 ]
 
 question_files = {
-    "What are the 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",
+    "Quais são as vendas líquidas para 2020?": "statement.pdf",
+    "Quantas curtidas tem o espaço?": "https://huggingface.co/spaces/impira/docquery",
+    "Qual é o título do post número 5?": "https://news.ycombinator.com",
 }
 
 
@@ -193,6 +204,17 @@ CSS = """
 #short-upload-box .w-full {
     min-height: 10rem !important;
 }
+/* I think something like this can be used to re-shape
+ * the table
+ */
+/*
+.gr-samples-table tr {
+    display: inline;
+}
+.gr-samples-table .p-2 {
+    width: 100px;
+}
+*/
 #select-a-file {
     width: 100%;
 }
@@ -267,7 +289,7 @@ gradio-app h2, .gradio-app h2 {
 """
 
 with gr.Blocks(css=CSS) as demo:
-    gr.Markdown("## Modular Visual Analytics for Prompt Engineering")
+    gr.Markdown("## Mecanismo de Consulta de Documentos")
 
     document = gr.Variable()
     example_question = gr.Textbox(visible=False)
@@ -276,9 +298,9 @@ with gr.Blocks(css=CSS) as demo:
     with gr.Row(equal_height=True):
         with gr.Column():
             with gr.Row():
-                gr.Markdown("## 1. Select a file", elem_id="select-a-file")
+                gr.Markdown("## 1. Selecione um arquivo", elem_id="select-a-file")
                 img_clear_button = gr.Button(
-                    "Clear", variant="secondary", elem_id="file-clear", visible=False
+                    "Limpar", variant="secondary", elem_id="file-clear", visible=False
                 )
             image = gr.Gallery(visible=False)
             with gr.Row(equal_height=True):
@@ -307,10 +329,10 @@ with gr.Blocks(css=CSS) as demo:
             )
 
         with gr.Column() as col:
-            gr.Markdown("## 2. Ask a question")
+            gr.Markdown("## 2. Faça uma pergunta")
             question = gr.Textbox(
                 label="Question",
-                placeholder="e.g. What is the invoice number?",
+                placeholder="e.g. Qual é o número da fatura?",
                 lines=1,
                 max_lines=1,
             )
@@ -323,11 +345,11 @@ with gr.Blocks(css=CSS) as demo:
             with gr.Row():
                 clear_button = gr.Button("Clear", variant="secondary")
                 submit_button = gr.Button(
-                    "Submit", variant="primary", elem_id="submit-button"
+                    "Enviar", variant="primary", elem_id="submit-button"
                 )
             with gr.Column():
                 output_text = gr.Textbox(
-                    label="Main Answer", visible=False, elem_id="answer"
+                    label="Resposta principal", visible=False, elem_id="answer"
                 )
                 output = gr.JSON(label="Output", visible=False)
 
@@ -396,4 +418,4 @@ with gr.Blocks(css=CSS) as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch(enable_queue=False)
+    demo.launch(enable_queue=False)