Update app.py via AI Editor

app.py

@@ -342,7 +342,7 @@ def get_right_col_content(selected_type, store_data):
                     },
                     multiple=False
                 ),
-                html.Div(id=doc_name_id),
+                # 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=[
@@ -491,14 +491,6 @@ def update_right_col(selected_type):
         Output('store-virtual-board', 'data'),
         Output('file-list', 'children'),
         Output('uploaded-doc-name-shred', 'children'),
-        Output('uploaded-doc-name-pink', 'children'),
-        Output('uploaded-doc-name-pink_review', 'children'),
-        Output('uploaded-doc-name-red', 'children'),
-        Output('uploaded-doc-name-red_review', 'children'),
-        Output('uploaded-doc-name-gold', 'children'),
-        Output('uploaded-doc-name-gold_review', 'children'),
-        Output('uploaded-doc-name-loe', 'children'),
-        Output('uploaded-doc-name-virtual_board', 'children'),
         Output('document-preview', 'children'),
         Output('store-generated-doc', 'data'),
         Output('download-document', 'data'),
@@ -529,7 +521,7 @@ def unified_handler(contents_lists, n_clicks_generate_list, n_clicks_download, s
                     filenames_lists, shred, pink, pink_review, red, red_review, gold, gold_review, loe, virtual_board, prev_generated):
     stores = [shred, pink, pink_review, red, red_review, gold, gold_review, loe, virtual_board]
     file_list = []
-    doc_names = [None]*9
+    uploaded_doc_name_shred = dash.no_update
     preview = dash.no_update
     generated_doc = dash.no_update
     download_data = dash.no_update
@@ -539,15 +531,15 @@ def unified_handler(contents_lists, n_clicks_generate_list, n_clicks_download, s
     trigger = ctx.triggered[0]['prop_id'] if ctx.triggered else ""
 
     subtype_map = {
-        "shred": (0, 0),
-        "pink": (1, 1),
-        "pink_review": (2, 2),
-        "red": (3, 3),
-        "red_review": (4, 4),
-        "gold": (5, 5),
-        "gold_review": (6, 6),
-        "loe": (7, 7),
-        "virtual_board": (8, 8)
+        "shred": 0,
+        "pink": 1,
+        "pink_review": 2,
+        "red": 3,
+        "red_review": 4,
+        "gold": 5,
+        "gold_review": 6,
+        "loe": 7,
+        "virtual_board": 8
     }
 
     # Handle upload
@@ -556,7 +548,7 @@ def unified_handler(contents_lists, n_clicks_generate_list, n_clicks_download, s
             if contents and filenames:
                 id_dict = ctx.inputs_list[0][i]['id']
                 subtype = id_dict['subtype']
-                idx, doc_name_idx = subtype_map[subtype]
+                idx = subtype_map[subtype]
                 if isinstance(contents, str):
                     contents = [contents]
                     filenames = [filenames]
@@ -565,7 +557,6 @@ def unified_handler(contents_lists, n_clicks_generate_list, n_clicks_download, s
                 file_text = process_document(content, name)
                 set_uploaded_doc(subtype, file_text, name)
                 stores[idx] = file_text
-                doc_names[doc_name_idx] = name
                 if subtype == "shred":
                     file_list = [
                         dbc.Row([
@@ -579,6 +570,7 @@ 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":
@@ -589,7 +581,7 @@ def unified_handler(contents_lists, n_clicks_generate_list, n_clicks_download, s
                     preview = markdown_table_preview(file_text)
                 generated_doc = dash.no_update
                 download_data = dash.no_update
-                return stores + [file_list] + doc_names + [preview, generated_doc, download_data, progress]
+                return stores + [file_list, uploaded_doc_name_shred, preview, generated_doc, download_data, progress]
 
     # Handle Generate
     if any(n_clicks_generate_list):
@@ -634,7 +626,7 @@ def unified_handler(contents_lists, n_clicks_generate_list, n_clicks_download, s
             preview = markdown_narrative_preview(generated_doc)
         progress = "progress"
         download_data = dash.no_update
-        return stores + [file_list] + doc_names + [preview, generated_doc, download_data, progress]
+        return stores + [file_list, uploaded_doc_name_shred, preview, generated_doc, download_data, progress]
 
     # Handle Download
     if "btn-download" in trigger:
@@ -662,19 +654,19 @@ 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] + doc_names + [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, dash.no_update]
         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] + doc_names + [dash.no_update, dash.no_update, download_data, 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] + doc_names + [dash.no_update, dash.no_update, download_data, 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)
@@ -701,12 +693,12 @@ 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] + doc_names + [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, dash.no_update]
     if selected_type in spreadsheet_types:
         preview = markdown_table_preview(doc)
     else:
         preview = markdown_narrative_preview(doc)
-    return stores + [file_list] + doc_names + [preview, doc, dash.no_update, dash.no_update]
+    return stores + [file_list, uploaded_doc_name_shred, preview, doc, dash.no_update, dash.no_update]
 
 @app.callback(
     Output("main-progress-dot-output", "children"),