refactor app.py

app.py

@@ -46,7 +46,6 @@ CSS = """
     --link-hover: #2b6cb0;
     --content-width: 1200px;
 }
-
 body {
     display: flex;
     justify-content: center;
@@ -54,7 +53,6 @@ body {
     color: var(--text-color);
     font-family: Arial, sans-serif;
 }
-
 .upload-section {
     width: 100%;
     margin: 0 auto 30px;
@@ -63,7 +61,6 @@ body {
     border-radius: 8px;
     box-shadow: var(--shadow);
 }
-
 .center-content {
     display: flex;
     flex-direction: column;
@@ -71,24 +68,20 @@ body {
     text-align: center;
     margin-bottom: 20px;
 }
-
 .header {
     margin-bottom: 30px;
     width: 100%;
 }
-
 .logo-container {
     width: 100%;
     margin-bottom: 20px;
 }
-
 .logo-img {
     width: 100%;
     max-width: var(--content-width);
     margin: 0 auto;
     display: block;
 }
-
 .nav-bar {
     display: flex;
     justify-content: center;
@@ -97,14 +90,12 @@ body {
     box-shadow: var(--shadow);
     margin-bottom: 20px;
 }
-
 .nav-links {
     display: flex;
     gap: 30px;
     width: 100%;
     justify-content: center;
 }
-
 .nav-link {
     color: var(--black);
     text-decoration: none;
@@ -112,12 +103,10 @@ body {
     font-size: 24px;
     transition: color 0.2s;
 }
-
 .nav-link:hover {
     color: var(--link-hover);
     text-decoration: none;
 }
-
 button {
     background-color: var(--text-color) !important;
     color: var(--black) !important;
@@ -125,7 +114,6 @@ button {
     border-radius: 4px;
     padding: 8px 16px;
 }
-
 .file-download {
     margin-top: 15px !important;
 }
@@ -138,12 +126,10 @@ button {
     animation: spin 1s linear infinite;
     margin: 20px auto;
 }
-
 @keyframes spin {
     0% { transform: rotate(0deg); }
     100% { transform: rotate(360deg); }
 }
-
 .loader-container {
     text-align: center;
     margin: 20px 0;
@@ -158,12 +144,10 @@ button {
     padding: 10px;
     border-radius: 4px;
 }
-
 .bold-label .gr-radio label {
     font-size: 14px;
     color: var(--black);
 }
-
 #analyze-btn {
     background-color: #FF5722 !important;
     color: white !important;
@@ -173,8 +157,6 @@ button {
     left: 3% !important;
     z-index: 1000 !important;
 }
-
-
 #unzip-btn {
     background-color: #4CAF50 !important;
     color: white !important;
@@ -184,19 +166,16 @@ button {
     left: 18% !important;
     z-index: 1000 !important;
 }
-
 #download_file {
     position: fixed !important;
     bottom: 1% !important;
     left: 22% !important;
     z-index: 1000 !important;
 }
-
 #analyze-btn:hover,#unzip-btn:hover{
     transform: translateY(-3px) !important;
     box-shadow: 0 4px 8px rgba(0,0,0,0.3) !important;
 }
-
 .square-pdf-btn {
     width: 90% !important;
     height: 3% !important;
@@ -207,48 +186,34 @@ button {
     justify-content: center !important;
     gap: 8px !important;
 }
-
-
 .square-pdf-btn img {
     width: 20% !important;
     height: 1% !important;
     margin: 0 !important;
 }
-
-
 .square-pdf-btn span {
     font-size: 14px !important;
     text-align: center !important;
 }
-
-
 .gradio-gallery-item:hover {
     background-color: transparent !important;
     filter: none !important;
     transform: none !important;
 }
-
 .custom-markdown h3 {
     font-size: 25px !important;
 }
-
 .tight-spacing {
     margin-bottom: -20px !important;
 }
-
 .tight-spacing-as {
     margin-top: 0px !important;
     margin-bottom: 0px !important;
 }
 
-.left-margin-column {
-    margin-left: 5%;
-}
-
 .image-container img {
     display: inline-block !important;
 }
-
 }
 """
 MAX_NUM_PAGES = 10
@@ -637,6 +602,7 @@ def toggle_spinner():
         gr.Column(visible=False),
         gr.File(visible=False),
         gr.update(visible=False),
+        gr.update(visible=False),
     )
 
 
@@ -836,7 +802,7 @@ with gr.Blocks(css=CSS, title="Document Analysis System", theme=paddle_theme) as
                     elem_classes=["tight-spacing"],
                 )
                 with gr.Column(
-                    visible=False, elem_classes="left-margin-column"
+                    visible=False
                 ) as Module_Options:
                     use_chart_recognition_md = gr.Markdown(
                         "### Using the chart parsing module",
@@ -894,7 +860,7 @@ with gr.Blocks(css=CSS, title="Document Analysis System", theme=paddle_theme) as
                         elem_id="use_textline_orientation_rd",
                     )
                 with gr.Column(
-                    visible=True, elem_classes="left-margin-column"
+                    visible=True
                 ) as Subpipeline_Options:
                     use_seal_recognition_md = gr.Markdown(
                         "### Using the seal text recognition subpipeline",
@@ -930,7 +896,7 @@ with gr.Blocks(css=CSS, title="Document Analysis System", theme=paddle_theme) as
                         elem_id="use_table_recognition_rd",
                     )
                 with gr.Column(
-                    visible=False, elem_classes="left-margin-column"
+                    visible=False
                 ) as Layout_region_detection_Options:
                     layout_threshold_md = gr.Markdown(
                         "### Score threshold of layout region detection model",
@@ -970,7 +936,7 @@ with gr.Blocks(css=CSS, title="Document Analysis System", theme=paddle_theme) as
                         elem_id="layout_unclip_ratio_nb",
                     )
                 with gr.Column(
-                    visible=False, elem_classes="left-margin-column"
+                    visible=False
                 ) as Text_detection_Options:
                     text_det_limit_type_md = gr.Markdown(
                         "### Image side length restriction type for text detection",
@@ -1051,7 +1017,7 @@ with gr.Blocks(css=CSS, title="Document Analysis System", theme=paddle_theme) as
                     )
 
                 with gr.Column(
-                    visible=False, elem_classes="left-margin-column"
+                    visible=False
                 ) as Seal_text_recognition_Options:
                     seal_det_limit_type_md = gr.Markdown(
                         "### Image side length restriction type for seal text detection",
@@ -1130,7 +1096,7 @@ with gr.Blocks(css=CSS, title="Document Analysis System", theme=paddle_theme) as
                         elem_id="seal_rec_score_thresh_nb",
                     )
                 with gr.Column(
-                    visible=False, elem_classes="left-margin-column"
+                    visible=False
                 ) as Table_recognition_Options:
                     use_ocr_results_with_table_cells_md = gr.Markdown(
                         "### Cell OCR mode",
@@ -1202,11 +1168,7 @@ with gr.Blocks(css=CSS, title="Document Analysis System", theme=paddle_theme) as
                 process_btn = gr.Button(
                     "🚀 Parse Document", elem_id="analyze-btn", variant="primary"
                 )
-                download_all_btn = gr.Button(
-                    "📦 Download Full Results (ZIP)",
-                    elem_id="unzip-btn",
-                    variant="primary",
-                )
+
             gr.Markdown(
                 f"""
                 1. Only the first {MAX_NUM_PAGES} pages will be processed.
@@ -1399,6 +1361,12 @@ with gr.Blocks(css=CSS, title="Document Analysis System", theme=paddle_theme) as
                                     visible=False,
                                 )
                             )
+            download_all_btn = gr.Button(
+                    "📦 Download Full Results (ZIP)",
+                    elem_id="unzip-btn",
+                    variant="primary",
+                    visible=False,
+                )
     # # Navigation bar
     with gr.Column(elem_classes=["nav-bar"]):
         gr.HTML(
@@ -1436,7 +1404,7 @@ 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]
+        toggle_spinner, outputs=[loading_spinner, prepare_spinner, download_file, tabs, download_all_btn]
     ).then(
         process_file,
         inputs=[
@@ -1484,7 +1452,7 @@ 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)
 
     gallery_markdown.select(
         update_markdown,
@@ -1541,11 +1509,9 @@ with gr.Blocks(css=CSS, title="Document Analysis System", theme=paddle_theme) as
                     tooltip.style.left = x + "px";
                     tooltip.style.top = y + "px";
                 }}
-
                 function hideTooltip() {{
                     tooltip.style.opacity = "0";
                 }}
-
                 elem.addEventListener("mousemove", showTooltip);
                 elem.addEventListener("mouseleave", hideTooltip);
             }});