Update app.py via AI Editor

app.py

@@ -179,6 +179,7 @@ def update_selected_doc_type(n_clicks_list, btn_ids):
         selected_type = btn_ids[idx[-1]]['index']
     else:
         selected_type = "Shred"
+    logging.info(f"Doc type selected: {selected_type}")
     return selected_type
 
 @app.callback(
@@ -229,6 +230,8 @@ def update_output(list_of_contents, list_of_names, existing_files):
             ]))
         if existing_files is None:
             existing_files = []
+        elif not isinstance(existing_files, list):
+            existing_files = [existing_files]
         shredded_document = None
         logging.info("Documents uploaded and file list updated.")
         return existing_files + new_files
@@ -245,11 +248,33 @@ def remove_file(n_clicks, existing_files):
     ctx = dash.callback_context
     if not ctx.triggered:
         raise dash.exceptions.PreventUpdate
-    removed_file = ctx.triggered[0]['prop_id'].split(',')[0].split(':')[-1].strip('}')
+    # Only proceed if a remove button was actually clicked
+    triggered_id = ctx.triggered[0]['prop_id'].split('.')[0]
+    if not triggered_id.startswith("{\"type\":\"remove-file\""):
+        raise dash.exceptions.PreventUpdate
+    try:
+        # Parse the file name from the triggered id
+        import json
+        triggered_dict = json.loads(triggered_id)
+        removed_file = triggered_dict['index']
+    except Exception as e:
+        logging.error(f"Could not parse removed file from callback context: {e}")
+        raise dash.exceptions.PreventUpdate
     uploaded_files.pop(removed_file, None)
     shredded_document = None
     logging.info(f"Removed file: {removed_file}")
-    return [file for file in existing_files if file['props']['children'][1]['props']['children'] != removed_file]
+    # Filter out the removed file from the list
+    filtered_files = []
+    if existing_files:
+        for file in existing_files:
+            # Defensive: file['props']['children'][1]['props']['children'] is the filename
+            try:
+                file_name = file['props']['children'][1]['props']['children']
+                if file_name != removed_file:
+                    filtered_files.append(file)
+            except Exception as e:
+                filtered_files.append(file)
+    return filtered_files
 
 @app.callback(
     Output('document-preview', 'children'),
@@ -286,6 +311,7 @@ def generate_any_doc(n_clicks_list, btn_ids, radio_values, upload_contents, uplo
             current_document = generated
             shredded_document = generated
             preview = spreadsheet_preview(generated, doc_type)
+            logging.info("Shred document generated.")
             return preview, "Shred generated"
         except Exception as e:
             logging.error(f"Error generating document: {str(e)}")
@@ -325,21 +351,25 @@ def generate_any_doc(n_clicks_list, btn_ids, radio_values, upload_contents, uplo
             pink_review_document = generated
             current_document = generated
             preview = spreadsheet_preview(generated, doc_type)
+            logging.info("Pink Review document generated.")
             return preview, f"{doc_type} generated"
         elif doc_type in ["Red Review", "Gold Review", "Virtual Board", "LOE"]:
             generated = generate_document(doc_type, [doc_content, shredded_document])
             current_document = generated
             preview = spreadsheet_preview(generated, doc_type)
+            logging.info(f"{doc_type} document generated.")
             return preview, f"{doc_type} generated"
         elif doc_type in ["Pink", "Red", "Gold"]:
             generated = generate_document(doc_type, [doc_content])
             current_document = generated
             preview = dcc.Markdown(generated)
+            logging.info(f"{doc_type} document generated.")
             return preview, f"{doc_type} generated"
         else:
             generated = generate_document(doc_type, [doc_content])
             current_document = generated
             preview = dcc.Markdown(generated)
+            logging.info(f"{doc_type} document generated.")
             return preview, f"{doc_type} generated"
     except Exception as e:
         logging.error(f"Error generating document: {str(e)}")