Small cleanups

app.py

@@ -175,20 +175,22 @@ def annotate_page(prediction, pages, document):
 def process_question(
     question, document, img_gallery, model, fields, output, output_table
 ):
+    field_name = question
+    if field_name is not None:
+        fields = {field_name: [question], **fields}
+
     if not question or document is None:
-        return None, None, None, None
+        return None, document, fields, output, gr.update(value=output_table)
 
     text_value = None
     pages = [processing_utils.decode_base64_to_image(p) for p in img_gallery]
     prediction = run_pipeline(model, question, document, 1)
     annotate_page(prediction, pages, document)
 
-    field_name = question
-    fields = {field_name: [question], **fields}
     output = {field_name: prediction, **output}
     table = [[field_name, prediction.get("answer")]] + output_table.values.tolist()
     return (
-        None,    
+        None,
         gr.update(visible=True, value=pages),
         fields,
         output,
@@ -388,7 +390,8 @@ with gr.Blocks(css=CSS) as demo:
             with gr.Tabs():
                 with gr.TabItem("Table"):
                     output_table = gr.Dataframe(
-                        headers=["Field", "Value"], **empty_table(fields.value),
+                        headers=["Field", "Value"],
+                        **empty_table(fields.value),
                         elem_id="results-table"
                     )
 
@@ -470,23 +473,18 @@ with gr.Blocks(css=CSS) as demo:
         outputs=submit_outputs,
     )
 
-    question.submit(
-        fn=process_question,
-        inputs=[question, document, image, model, fields, output, output_table],
-        outputs=[question, image, fields, output, output_table],
-    )
-
-    submit_button.click(
-        process_question,
-        inputs=[question, document, model],
-        outputs=[image, output, output_table],
-    )
+    for action in [question.submit, submit_button.click]:
+        action(
+            fn=process_question,
+            inputs=[question, document, image, model, fields, output, output_table],
+            outputs=[question, image, fields, output, output_table],
+        )
 
-    model.change(
-        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,