Update title

app.py

@@ -93,6 +93,7 @@ def process_path(path):
                 gr.update(visible=True, value=document.preview),
                 gr.update(visible=True),
                 gr.update(visible=False, value=None),
+                gr.update(visible=False, value=None),
             )
         except Exception:
             pass
@@ -101,6 +102,7 @@ def process_path(path):
         gr.update(visible=False, value=None),
         gr.update(visible=False),
         gr.update(visible=False, value=None),
+        gr.update(visible=False, value=None),
     )
 
 
@@ -113,6 +115,7 @@ def process_upload(file):
             gr.update(visible=False, value=None),
             gr.update(visible=False),
             gr.update(visible=False, value=None),
+            gr.update(visible=False, value=None),
         )
 
 
@@ -121,7 +124,7 @@ colors = ["#64A087", "green", "black"]
 
 def process_question(question, document, model=list(CHECKPOINTS.keys())[0]):
     if document is None:
-        return None, None
+        return None, None, None
 
     predictions = run_pipeline(model, question, document, 3)
     pages = [x.copy().convert("RGB") for x in document.preview]
@@ -143,16 +146,21 @@ def process_question(question, document, model=list(CHECKPOINTS.keys())[0]):
             )
             draw.rectangle(((x1, y1), (x2, y2)), fill=(0, 255, 0, int(0.4 * 255)))
 
-    return gr.update(visible=True, value=pages), gr.update(
-        visible=True, value=predictions
+    return (
+        gr.update(visible=True, value=pages),
+        gr.update(visible=True, value=predictions),
+        gr.update(
+            visible=True,
+            value=predictions[0]["answer"] if ensure_list(predictions) else None,
+        ),
     )
 
 
 def load_example_document(img, question, model):
     if img is not None:
         document = ImageDocument(Image.fromarray(img))
-        preview, answer = process_question(question, document, model)
-        return document, question, preview, gr.update(visible=True), answer
+        preview, answer, answer_text = process_question(question, document, model)
+        return document, question, preview, gr.update(visible=True), answer, answer_text
     else:
         return None, None, None, gr.update(visible=False), None
 
@@ -202,7 +210,7 @@ CSS = """
 """
 
 with gr.Blocks(css=CSS) as demo:
-    gr.Markdown("# DocQuery: Query Documents w/ NLP")
+    gr.Markdown("# DocQuery: Document Query Engine")
     gr.Markdown(
         "DocQuery uses LayoutLMv1 fine-tuned on DocVQA, a document visual question"
         " answering dataset, as well as SQuAD, which boosts its English-language comprehension."
@@ -261,6 +269,7 @@ with gr.Blocks(css=CSS) as demo:
                     "Submit", variant="primary", elem_id="submit-button"
                 )
             with gr.Column():
+                output_text = gr.Textbox(label="Top Answer", visible=False)
                 output = gr.JSON(label="Output", visible=False)
 
     img_clear_button.click(
@@ -268,13 +277,23 @@ with gr.Blocks(css=CSS) as demo:
             gr.update(visible=False, value=None),
             None,
             gr.update(visible=False, value=None),
+            gr.update(visible=False, value=None),
             gr.update(visible=False),
             None,
             None,
             None,
         ),
         inputs=img_clear_button,
-        outputs=[image, document, output, img_clear_button, example_image, upload, url],
+        outputs=[
+            image,
+            document,
+            output,
+            output_text,
+            img_clear_button,
+            example_image,
+            upload,
+            url,
+        ],
     )
     clear_button.click(
         lambda _: (
@@ -282,45 +301,57 @@ with gr.Blocks(css=CSS) as demo:
             None,
             None,
             gr.update(visible=False, value=None),
+            gr.update(visible=False, value=None),
             None,
             None,
             None,
         ),
         inputs=clear_button,
-        outputs=[image, document, question, output, example_image, upload, url],
+        outputs=[
+            image,
+            document,
+            question,
+            output,
+            output_text,
+            example_image,
+            upload,
+            url,
+        ],
     )
 
     upload.change(
         fn=process_upload,
         inputs=[upload],
-        outputs=[document, image, img_clear_button, output],
+        outputs=[document, image, img_clear_button, output, output_text],
     )
     url.change(
         fn=process_path,
         inputs=[url],
-        outputs=[document, image, img_clear_button, output],
+        outputs=[document, image, img_clear_button, output, output_text],
     )
 
     question.submit(
         fn=process_question,
         inputs=[question, document, model],
-        outputs=[image, output],
+        outputs=[image, output, output_text],
     )
 
     submit_button.click(
         process_question,
         inputs=[question, document, model],
-        outputs=[image, output],
+        outputs=[image, output, output_text],
     )
 
     model.change(
-        process_question, inputs=[question, document, model], outputs=[image, output]
+        process_question,
+        inputs=[question, document, model],
+        outputs=[image, output, output_text],
     )
 
     example_image.change(
         fn=load_example_document,
         inputs=[example_image, example_question, model],
-        outputs=[document, question, image, img_clear_button, output],
+        outputs=[document, question, image, img_clear_button, output, output_text],
     )
 
 if __name__ == "__main__":