refactor app.py

app.py

@@ -801,9 +801,7 @@ with gr.Blocks(css=CSS, title="Document Analysis System", theme=paddle_theme) as
                     scale=0,
                     elem_classes=["tight-spacing"],
                 )
-                with gr.Column(
-                    visible=False
-                ) as Module_Options:
+                with gr.Column(visible=False) as Module_Options:
                     use_chart_recognition_md = gr.Markdown(
                         "### Using the chart parsing module",
                         elem_id="use_chart_recognition_md",
@@ -859,9 +857,7 @@ with gr.Blocks(css=CSS, title="Document Analysis System", theme=paddle_theme) as
                         show_label=False,
                         elem_id="use_textline_orientation_rd",
                     )
-                with gr.Column(
-                    visible=True
-                ) as Subpipeline_Options:
+                with gr.Column(visible=True) as Subpipeline_Options:
                     use_seal_recognition_md = gr.Markdown(
                         "### Using the seal text recognition subpipeline",
                         elem_id="use_seal_recognition_md",
@@ -895,9 +891,7 @@ with gr.Blocks(css=CSS, title="Document Analysis System", theme=paddle_theme) as
                         show_label=False,
                         elem_id="use_table_recognition_rd",
                     )
-                with gr.Column(
-                    visible=False
-                ) as Layout_region_detection_Options:
+                with gr.Column(visible=False) as Layout_region_detection_Options:
                     layout_threshold_md = gr.Markdown(
                         "### Score threshold of layout region detection model",
                         elem_id="layout_threshold_md",
@@ -935,9 +929,7 @@ with gr.Blocks(css=CSS, title="Document Analysis System", theme=paddle_theme) as
                         show_label=False,
                         elem_id="layout_unclip_ratio_nb",
                     )
-                with gr.Column(
-                    visible=False
-                ) as Text_detection_Options:
+                with gr.Column(visible=False) as Text_detection_Options:
                     text_det_limit_type_md = gr.Markdown(
                         "### Image side length restriction type for text detection",
                         elem_id="text_det_limit_type_md",
@@ -1016,9 +1008,7 @@ with gr.Blocks(css=CSS, title="Document Analysis System", theme=paddle_theme) as
                         elem_id="text_rec_score_thresh_nb",
                     )
 
-                with gr.Column(
-                    visible=False
-                ) as Seal_text_recognition_Options:
+                with gr.Column(visible=False) as Seal_text_recognition_Options:
                     seal_det_limit_type_md = gr.Markdown(
                         "### Image side length restriction type for seal text detection",
                         elem_id="seal_det_limit_type_md",
@@ -1095,9 +1085,7 @@ with gr.Blocks(css=CSS, title="Document Analysis System", theme=paddle_theme) as
                         show_label=False,
                         elem_id="seal_rec_score_thresh_nb",
                     )
-                with gr.Column(
-                    visible=False
-                ) as Table_recognition_Options:
+                with gr.Column(visible=False) as Table_recognition_Options:
                     use_ocr_results_with_table_cells_md = gr.Markdown(
                         "### Cell OCR mode",
                         elem_id="use_ocr_results_with_table_cells_md",
@@ -1362,11 +1350,11 @@ with gr.Blocks(css=CSS, title="Document Analysis System", theme=paddle_theme) as
                                 )
                             )
             download_all_btn = gr.Button(
-                    "📦 Download Full Results (ZIP)",
-                    elem_id="unzip-btn",
-                    variant="primary",
-                    visible=False,
-                )
+                "📦 Download Full Results (ZIP)",
+                elem_id="unzip-btn",
+                variant="primary",
+                visible=False,
+            )
     # # Navigation bar
     with gr.Column(elem_classes=["nav-bar"]):
         gr.HTML(
@@ -1404,7 +1392,14 @@ with gr.Blocks(css=CSS, title="Document Analysis System", theme=paddle_theme) as
     """,
     )
     process_btn.click(
-        toggle_spinner, outputs=[loading_spinner, prepare_spinner, download_file, tabs, download_all_btn]
+        toggle_spinner,
+        outputs=[
+            loading_spinner,
+            prepare_spinner,
+            download_file,
+            tabs,
+            download_all_btn,
+        ],
     ).then(
         process_file,
         inputs=[
@@ -1452,7 +1447,9 @@ with gr.Blocks(css=CSS, title="Document Analysis System", theme=paddle_theme) as
         + output_json_list
         + markdown_display_list
         + gallery_list,
-    ).then(lambda: gr.update(visible=True), outputs=download_all_btn)
+    ).success(
+        lambda: gr.update(visible=True), outputs=download_all_btn
+    )
 
     gallery_markdown.select(
         update_markdown,