Basics work

README.md

@@ -4,7 +4,7 @@ emoji: 🦉
 colorFrom: purple
 colorTo: purple
 sdk: gradio
-sdk_version: 3.2
+sdk_version: 3.1.7
 app_file: app.py
 pinned: true
 ---

app.py

@@ -89,7 +89,14 @@ QUESTION_FILES = {}
 FIELDS = {
     "Vendor Name": ["Vendor Name - Logo?", "Vendor Name - Address?"],
     "Vendor Address": ["Vendor Address?"],
+    "Invoice Number": ["Invoice Number?"],
+    "Invoice Date": ["Invoice Date?"],
+    "Due Date": ["Due Date?"],
+    "Subtotal": ["Subtotal?"],
+    "Total Tax": ["Total Tax?"],
     "Invoice Total": ["Invoice Total?"],
+    "Amount Due": ["Amount Due?"],
+    "Payment Terms": ["Payment Terms?"],
 }
 
 
@@ -379,74 +386,73 @@ with gr.Blocks(css=CSS) as demo:
                 submit_button = gr.Button(
                     "Submit", variant="primary", elem_id="submit-button"
                 )
-            with gr.Column():
-                # with gr.Tab("Table"):
-                output_table = gr.Dataframe(
-                    headers=["Field", "Value"],
-                    value=[[name, None] for name in FIELDS.keys()],
-                )
+            with gr.Tabs():
+                with gr.TabItem("Table"):
+                    output_table = gr.Dataframe(
+                        headers=["Field", "Value"],
+                        value=[[name, None] for name in FIELDS.keys()],
+                    )
+
+                with gr.TabItem("JSON"):
+                    output = gr.JSON(label="Output", visible=False)
+
+    for cb in [img_clear_button, clear_button]:
+        cb.click(
+            lambda _: (
+                gr.update(visible=False, value=None),
+                None,
+                gr.update(visible=False, value=None),
+                gr.update(value=None),
+                gr.update(visible=False),
+                None,
+                None,
+                None,
+                gr.update(visible=False, value=None),
+                None,
+            ),
+            inputs=clear_button,
+            outputs=[
+                image,
+                document,
+                output,
+                output_table,
+                img_clear_button,
+                example_image,
+                upload,
+                url,
+                url_error,
+                question,
+            ],
+        )
+
+    upload.change(
+        fn=process_upload,
+        inputs=[upload],
+        outputs=[document, image, img_clear_button, output, output_table, url_error],
+    )
+    submit.click(
+        fn=process_path,
+        inputs=[url],
+        outputs=[document, image, img_clear_button, output, output_table, url_error],
+    )
 
-                # with gr.Tab("JSON"):
-                # XXX RENAME
-                output = gr.JSON(label="Output", visible=False)
-
-    #    for cb in [img_clear_button, clear_button]:
-    #        cb.click(
-    #            lambda _: (
-    #                gr.update(visible=False, value=None),
-    #                None,
-    #                gr.update(visible=False, value=None),
-    #                gr.update(value=None),
-    #                gr.update(visible=False),
-    #                None,
-    #                None,
-    #                None,
-    #                gr.update(visible=False, value=None),
-    #                None,
-    #            ),
-    #            inputs=clear_button,
-    #            outputs=[
-    #                image,
-    #                document,
-    #                output,
-    #                output_table,
-    #                img_clear_button,
-    #                example_image,
-    #                upload,
-    #                url,
-    #                url_error,
-    #                question,
-    #            ],
-    #        )
-    #
-    #    upload.change(
-    #        fn=process_upload,
-    #        inputs=[upload],
-    #        outputs=[document, image, img_clear_button, output, output_table, url_error],
-    #    )
-    #    submit.click(
-    #        fn=process_path,
-    #        inputs=[url],
-    #        outputs=[document, image, img_clear_button, output, output_table, url_error],
-    #    )
-    #
-    #    question.submit(
-    #        fn=process_question,
-    #        inputs=[question, document, model],
-    #        outputs=[image, output, output_table],
-    #    )
-    #
-    #    submit_button.click(
-    #        process_question,
-    #        inputs=[question, document, model],
-    #        outputs=[image, output, output_table],
-    #    )
-    #
-    #    model.change(
-    #        process_question,
-    #        inputs=[question, document, model],
-    #        outputs=[image, output, output_table],
-    #    )
+    question.submit(
+        fn=process_question,
+        inputs=[question, document, model],
+        outputs=[image, output, output_table],
+    )
+
+    submit_button.click(
+        process_question,
+        inputs=[question, document, model],
+        outputs=[image, output, output_table],
+    )
+
+    model.change(
+        process_question,
+        inputs=[question, document, model],
+        outputs=[image, output, output_table],
+    )
 
     example_image.change(
         fn=load_example_document,