Update app.py via AI Editor

app.py

@@ -240,7 +240,6 @@ def get_right_col_content(selected_type, store_data):
         children=[html.Div(id="loading-output")]
     ))
 
-    # For Shred, show upload
     if selected_type == "Shred":
         controls.append(
             html.Div([
@@ -265,7 +264,6 @@ def get_right_col_content(selected_type, store_data):
                 file_list_component()
             ])
         )
-    # For others, show dependency uploads and previews
     elif selected_type in doc_dependencies:
         sources = doc_dependencies[selected_type]["source"]
         for src in sources:
@@ -495,7 +493,6 @@ def handle_shred_upload(contents, filenames, current_shred, file_list):
     logging.info("Shred document uploaded and stored.")
     return latest_text, file_previews, filenames[-1]
 
-# Pattern-matching callback for all dependency uploads
 @app.callback(
     [
         Output('store-pink', 'data'),
@@ -506,7 +503,6 @@ def handle_shred_upload(contents, filenames, current_shred, file_list):
         Output('store-gold-review', 'data'),
         Output('store-loe', 'data'),
         Output('store-virtual-board', 'data'),
-        Output({'type': 'uploaded-doc-name', 'subtype': MATCH, 'index': MATCH}, 'children')
     ],
     [
         Input({'type': 'upload-doc-type', 'subtype': ALL, 'index': ALL}, 'contents')
@@ -516,11 +512,10 @@ def handle_shred_upload(contents, filenames, current_shred, file_list):
     ],
     prevent_initial_call=True
 )
-def handle_dep_upload(contents_list, filenames_list):
+def handle_dep_upload_to_stores(contents_list, filenames_list):
     stores = [dash.no_update]*8
-    upload_name = ""
     if not contents_list or not filenames_list:
-        return [dash.no_update]*8 + [upload_name]
+        return stores
     for i, (contents, filenames) in enumerate(zip(contents_list, filenames_list)):
         if contents and filenames:
             if isinstance(contents, str):
@@ -528,7 +523,7 @@ def handle_dep_upload(contents_list, filenames_list):
                 filenames = [filenames]
             for content, name in zip(contents, filenames):
                 file_text = process_document(content, name)
-                if 'pink' in name.lower():
+                if 'pink' in name.lower() and 'review' not in name.lower():
                     stores[0] = file_text
                 elif 'pink_review' in name.lower():
                     stores[1] = file_text
@@ -544,8 +539,21 @@ def handle_dep_upload(contents_list, filenames_list):
                     stores[6] = file_text
                 elif 'virtual' in name.lower():
                     stores[7] = file_text
-                upload_name = name
-    return stores + [upload_name]
+    return stores
+
+@app.callback(
+    Output({'type': 'uploaded-doc-name', 'subtype': MATCH, 'index': MATCH}, 'children'),
+    [
+        Input({'type': 'upload-doc-type', 'subtype': MATCH, 'index': MATCH}, 'filename')
+    ],
+    prevent_initial_call=True
+)
+def update_uploaded_doc_name(filename):
+    if not filename:
+        return ""
+    if isinstance(filename, list):
+        return filename[-1]
+    return filename
 
 @app.callback(
     Output('document-preview', 'children'),
@@ -569,7 +577,6 @@ def preview_or_generate_doc(n_clicks_list, selected_type, shred, pink, pink_revi
     ctx = callback_context
     trigger = ctx.triggered[0]['prop_id'] if ctx.triggered else ""
     logging.info(f"Document preview trigger: {trigger}")
-    # If generate button pressed
     if any(n_clicks_list):
         logging.info(f"Generating document for type: {selected_type}")
         inputs = {
@@ -612,12 +619,9 @@ def preview_or_generate_doc(n_clicks_list, selected_type, shred, pink, pink_revi
             preview = markdown_narrative_preview(generated_doc)
         logging.info("Document preview updated.")
         return preview
-    # If just switching doc type, show preview of loaded doc (from store)
     else:
-        # For Shred, show table preview
         if selected_type == "Shred" and shred:
             return markdown_table_preview(shred)
-        # For spreadsheet types, preview from store
         elif selected_type in spreadsheet_types:
             doc_store = {
                 "Pink Review": pink_review,
@@ -628,7 +632,6 @@ def preview_or_generate_doc(n_clicks_list, selected_type, shred, pink, pink_revi
             }
             doc = doc_store.get(selected_type, "")
             return markdown_table_preview(doc)
-        # For narrative types, preview from store
         elif selected_type in narrative_types:
             doc_store = {
                 "Pink": pink,
@@ -685,7 +688,6 @@ def update_generated_doc(n_clicks_list, selected_type, preview_content, shred, p
 )
 def download_document(n_clicks, selected_type, generated_doc, shred, pink, pink_review, red, red_review, gold, gold_review, loe, virtual_board):
     doc = generated_doc
-    # If not generated, try to get from store for preview/download
     if not doc:
         if selected_type == "Shred":
             doc = shred