Update app.py via AI Editor

app.py

@@ -680,6 +680,68 @@ def unified_master_callback(
     output_data_upload = html.Div("No action taken yet.", style={"wordWrap": "break-word"})
     uploaded_rfp_decoded_bytes = None
 
+    # --- Fix for delete buttons using callback_context.triggered ---
+    # Check if a delete button was pressed (documents or proposals)
+    if ctx.triggered and ctx.triggered[0]['prop_id']:
+        prop_id_str = ctx.triggered[0]['prop_id'].split('.')[0]
+        if prop_id_str == "{'type': 'delete-doc-btn', 'index': '*', 'group': 'doc'}" or prop_id_str == "{'type': 'delete-proposal-btn', 'index': '*', 'group': 'proposal'}":
+            # fallback for some Dash versions
+            prop_id = eval(ctx.triggered[0]['prop_id'].split('.')[0])
+        else:
+            try:
+                import json
+                prop_id = json.loads(prop_id_str.replace("'", '"'))
+            except Exception:
+                prop_id = ctx.triggered[0]['prop_id']
+        if isinstance(prop_id, dict) and prop_id.get('type') == 'delete-doc-btn' and prop_id.get('group') == 'doc':
+            del_filename = prop_id.get('index')
+            if del_filename in sess_data["uploaded_documents"]:
+                del sess_data["uploaded_documents"][del_filename]
+            if del_filename in sess_data["uploaded_documents_fileid"]:
+                try:
+                    genai.delete_file(sess_data["uploaded_documents_fileid"][del_filename])
+                except Exception as e:
+                    logging.warning(f"[{sid}] Failed to delete Gemini file {del_filename}: {e}")
+                del sess_data["uploaded_documents_fileid"][del_filename]
+            if del_filename in sess_data["uploaded_documents_bytes"]:
+                del sess_data["uploaded_documents_bytes"][del_filename]
+            if del_filename in sess_data["shredded_documents"]:
+                del sess_data["shredded_documents"][del_filename]
+            # If there are temp files on disk, attempt to remove them
+            tempdir = sess_data.get("session_tempdir")
+            if tempdir and os.path.isdir(tempdir):
+                try:
+                    file_path = os.path.join(tempdir, del_filename)
+                    if os.path.exists(file_path):
+                        os.remove(file_path)
+                except Exception:
+                    pass
+            if selected_doc == del_filename:
+                selected_doc = None
+            logging.info(f"[{sid}] Document deleted: {del_filename}")
+        elif isinstance(prop_id, dict) and prop_id.get('type') == 'delete-proposal-btn' and prop_id.get('group') == 'proposal':
+            del_filename = prop_id.get('index')
+            if del_filename in sess_data["proposals"]:
+                del sess_data["proposals"][del_filename]
+            if del_filename in sess_data["proposals_fileid"]:
+                try:
+                    genai.delete_file(sess_data["proposals_fileid"][del_filename])
+                except Exception as e:
+                    logging.warning(f"[{sid}] Failed to delete Gemini proposal file {del_filename}: {e}")
+                del sess_data["proposals_fileid"][del_filename]
+            # If there are temp files on disk, attempt to remove them
+            tempdir = sess_data.get("session_tempdir")
+            if tempdir and os.path.isdir(tempdir):
+                try:
+                    file_path = os.path.join(tempdir, del_filename)
+                    if os.path.exists(file_path):
+                        os.remove(file_path)
+                except Exception:
+                    pass
+            if selected_proposal == del_filename:
+                selected_proposal = None
+            logging.info(f"[{sid}] Proposal deleted: {del_filename}")
+
     def safe_get_n_clicks(ctx, idx):
         try:
             return ctx.inputs_list[idx]
@@ -757,53 +819,6 @@ def unified_master_callback(
         else:
             logging.error(f"[{sid}] Failed to decode uploaded proposal: {proposal_filename}")
 
-    # --- Fix for delete buttons below ---
-    if triggered_id and isinstance(doc_delete_clicks, list):
-        for i, n_click in enumerate(doc_delete_clicks):
-            if n_click:
-                btn_id = ctx.inputs_list[8][i]['id']
-                del_filename = btn_id['index']
-                # Remove from uploaded_documents
-                if del_filename in sess_data["uploaded_documents"]:
-                    del sess_data["uploaded_documents"][del_filename]
-                # Remove from uploaded_documents_fileid and delete Gemini file if exists
-                if del_filename in sess_data["uploaded_documents_fileid"]:
-                    try:
-                        genai.delete_file(sess_data["uploaded_documents_fileid"][del_filename])
-                    except Exception as e:
-                        logging.warning(f"[{sid}] Failed to delete Gemini file {del_filename}: {e}")
-                    del sess_data["uploaded_documents_fileid"][del_filename]
-                # Remove from uploaded_documents_bytes
-                if del_filename in sess_data["uploaded_documents_bytes"]:
-                    del sess_data["uploaded_documents_bytes"][del_filename]
-                # Remove from shredded_documents
-                if del_filename in sess_data["shredded_documents"]:
-                    del sess_data["shredded_documents"][del_filename]
-                logging.info(f"[{sid}] Document deleted: {del_filename}")
-                if selected_doc == del_filename:
-                    selected_doc = None
-                break
-
-    if triggered_id and isinstance(proposal_delete_clicks, list):
-        for i, n_click in enumerate(proposal_delete_clicks):
-            if n_click:
-                btn_id = ctx.inputs_list[12][i]['id']
-                del_filename = btn_id['index']
-                # Remove from proposals
-                if del_filename in sess_data["proposals"]:
-                    del sess_data["proposals"][del_filename]
-                # Remove from proposals_fileid and delete Gemini file if exists
-                if del_filename in sess_data["proposals_fileid"]:
-                    try:
-                        genai.delete_file(sess_data["proposals_fileid"][del_filename])
-                    except Exception as e:
-                        logging.warning(f"[{sid}] Failed to delete Gemini proposal file {del_filename}: {e}")
-                    del sess_data["proposals_fileid"][del_filename]
-                logging.info(f"[{sid}] Proposal deleted: {del_filename}")
-                if selected_proposal == del_filename:
-                    selected_proposal = None
-                break
-
     doc_options = [{'label': truncate_filename(fn), 'value': fn} for fn in sess_data["uploaded_documents"].keys()]
     doc_value = selected_doc if selected_doc in sess_data["uploaded_documents"] else (next(iter(sess_data["uploaded_documents"]), None) if sess_data["uploaded_documents"] else None)
     documents_list = get_documents_list(sess_data["uploaded_documents"], sess_data["shredded_documents"])