Update app.py via AI Editor

app.py

@@ -269,6 +269,9 @@ def get_left_col_content():
 
 def get_right_col_content(selected_type, store_data):
     controls = []
+    controls.append(
+        html.Div(id='progress-dot-area', style={'textAlign': 'center', 'marginBottom': '8px'})
+    )
     controls.append(dcc.Loading(
         id="loading-indicator",
         type="dot",
@@ -342,7 +345,6 @@ def get_right_col_content(selected_type, store_data):
                     },
                     multiple=False
                 ),
-                # Only the 'shred' one is present in layout, so don't add for others
                 dbc.RadioItems(
                     id={'type': 'radio-doc-source', 'subtype': src, 'index': selected_type},
                     options=[
@@ -414,7 +416,10 @@ app.layout = dbc.Container([
             width=3
         ),
         dbc.Col(
-            html.Div(id='right-col-content'),
+            html.Div([
+                html.Div(id='progress-dot-area', style={'textAlign': 'center', 'marginBottom': '8px'}),
+                html.Div(id='right-col-content')
+            ]),
             width=9
         )
     ])
@@ -525,7 +530,6 @@ def unified_handler(contents_lists, n_clicks_generate_list, n_clicks_download, s
     preview = dash.no_update
     generated_doc = dash.no_update
     download_data = dash.no_update
-    progress = dash.no_update
 
     ctx = callback_context
     trigger = ctx.triggered[0]['prop_id'] if ctx.triggered else ""
@@ -542,7 +546,8 @@ def unified_handler(contents_lists, n_clicks_generate_list, n_clicks_download, s
         "virtual_board": 8
     }
 
-    # Handle upload
+    # Show progress dot as soon as action is triggered
+    progress = "progress"
     if "upload-doc-type" in trigger:
         for i, (contents, filenames) in enumerate(zip(contents_lists, filenames_lists)):
             if contents and filenames:
@@ -571,7 +576,6 @@ def unified_handler(contents_lists, n_clicks_generate_list, n_clicks_download, s
                         ], id={'type': 'file-row', 'index': name}, align="center", className="mb-1")
                     ]
                     uploaded_doc_name_shred = name
-                progress = "progress"
                 # Set preview for this upload
                 if subtype == "shred":
                     preview = markdown_table_preview(file_text)
@@ -583,7 +587,6 @@ def unified_handler(contents_lists, n_clicks_generate_list, n_clicks_download, s
                 download_data = dash.no_update
                 return stores + [file_list, uploaded_doc_name_shred, preview, generated_doc, download_data, progress]
 
-    # Handle Generate
     if any(n_clicks_generate_list):
         logging.info(f"Generating document for type: {selected_type}")
         inputs = {
@@ -620,15 +623,14 @@ def unified_handler(contents_lists, n_clicks_generate_list, n_clicks_download, s
         t.join(timeout=60)
         generated_doc = result_holder.get('result', 'Error: No document generated.')
         set_generated_doc(selected_type, generated_doc)
+        # Always show preview after generation
         if selected_type in spreadsheet_types:
             preview = markdown_table_preview(generated_doc)
         else:
             preview = markdown_narrative_preview(generated_doc)
-        progress = "progress"
         download_data = dash.no_update
         return stores + [file_list, uploaded_doc_name_shred, preview, generated_doc, download_data, progress]
 
-    # Handle Download
     if "btn-download" in trigger:
         doc = prev_generated
         if not doc:
@@ -654,21 +656,18 @@ def unified_handler(contents_lists, n_clicks_generate_list, n_clicks_download, s
                 doc = stores[8]
         if not doc:
             logging.warning("No generated document to download.")
-            return stores + [file_list, uploaded_doc_name_shred, dash.no_update, dash.no_update, dash.no_update, dash.no_update]
+            return stores + [file_list, uploaded_doc_name_shred, dash.no_update, dash.no_update, dash.no_update, None]
         if selected_type in spreadsheet_types:
             xlsx_io = markdown_tables_to_xlsx(doc)
             logging.info("Spreadsheet document prepared for download.")
             download_data = dcc.send_bytes(xlsx_io.getvalue(), filename=f"{selected_type}_output.xlsx")
-            progress = "progress"
             return stores + [file_list, uploaded_doc_name_shred, dash.no_update, dash.no_update, download_data, progress]
         else:
             content = doc.encode('utf-8')
             logging.info("Narrative document prepared for download.")
             download_data = dcc.send_bytes(content, filename=f"{selected_type}_output.md")
-            progress = "progress"
             return stores + [file_list, uploaded_doc_name_shred, dash.no_update, dash.no_update, download_data, progress]
 
-    # Handle Doc Type Change/Preview
     doc = get_generated_doc(selected_type)
     if not doc:
         if selected_type == "Shred" and stores[0]:
@@ -693,18 +692,29 @@ def unified_handler(contents_lists, n_clicks_generate_list, n_clicks_download, s
             preview = markdown_narrative_preview(doc)
         else:
             preview = html.Div("No document loaded.")
-        return stores + [file_list, uploaded_doc_name_shred, preview, dash.no_update, dash.no_update, dash.no_update]
+        return stores + [file_list, uploaded_doc_name_shred, preview, dash.no_update, dash.no_update, None]
     if selected_type in spreadsheet_types:
         preview = markdown_table_preview(doc)
     else:
         preview = markdown_narrative_preview(doc)
-    return stores + [file_list, uploaded_doc_name_shred, preview, doc, dash.no_update, dash.no_update]
+    return stores + [file_list, uploaded_doc_name_shred, preview, doc, dash.no_update, None]
+
+@app.callback(
+    Output("progress-dot-area", "children"),
+    Input('progress-dot-store', 'data')
+)
+def update_right_progress_dot(progress):
+    if progress == "progress":
+        return html.Div("...", id="progress-blinker", style={'fontSize': '2rem', 'letterSpacing': '0.5rem', 'animation': 'blink 1s infinite'})
+    return ""
 
 @app.callback(
     Output("main-progress-dot-output", "children"),
     Input('progress-dot-store', 'data')
 )
 def update_main_progress_dot(progress):
+    if progress == "progress":
+        return html.Div("...", style={'fontSize': '2rem', 'letterSpacing': '0.5rem', 'animation': 'blink 1s infinite'})
     return ""
 
 if __name__ == '__main__':