Update app.py via AI Editor

app.py

@@ -2,7 +2,7 @@ import os
 import base64
 import io
 import dash
-from dash import dcc, html, Input, Output, State, callback_context, MATCH, ALL
+from dash import dcc, html, Input, Output, State, callback_context, ALL
 import dash_bootstrap_components as dbc
 import pandas as pd
 import logging
@@ -679,20 +679,19 @@ def unified_master_callback(
     output_data_upload = html.Div("No action taken yet.", style={"wordWrap": "break-word"})
     uploaded_rfp_decoded_bytes = None
 
-    # NEW delete logic: use ctx.triggered[0]['id'] which is a dict for pattern-matched IDs
+    triggered_id = getattr(ctx, "triggered_id", None)
     is_doc_delete = False
     is_proposal_delete = False
     doc_del_filename = None
     proposal_del_filename = None
-    if ctx.triggered and ctx.triggered[0].get('id', None):
-        triggered_id_dict = ctx.triggered[0]['id']
-        if isinstance(triggered_id_dict, dict):
-            if triggered_id_dict.get('type') == 'delete-doc-btn' and triggered_id_dict.get('group') == 'doc':
-                is_doc_delete = True
-                doc_del_filename = triggered_id_dict.get('index')
-            elif triggered_id_dict.get('type') == 'delete-proposal-btn' and triggered_id_dict.get('group') == 'proposal':
-                is_proposal_delete = True
-                proposal_del_filename = triggered_id_dict.get('index')
+
+    if isinstance(triggered_id, dict):
+        if triggered_id.get('type') == 'delete-doc-btn' and triggered_id.get('group') == 'doc':
+            is_doc_delete = True
+            doc_del_filename = triggered_id.get('index')
+        elif triggered_id.get('type') == 'delete-proposal-btn' and triggered_id.get('group') == 'proposal':
+            is_proposal_delete = True
+            proposal_del_filename = triggered_id.get('index')
 
     if is_doc_delete and doc_del_filename:
         if doc_del_filename in sess_data["uploaded_documents"]:
@@ -749,7 +748,7 @@ def unified_master_callback(
     doc_delete_clicks = safe_get_n_clicks(ctx, 8)
     proposal_delete_clicks = safe_get_n_clicks(ctx, 12)
 
-    if not ctx.triggered or (ctx.triggered[0].get('prop_id', None) in ['session-id-store']):
+    if not ctx.triggered or (getattr(ctx, "triggered_id", None) == 'session-id-store'):
         doc_options = [{'label': truncate_filename(fn), 'value': fn} for fn in sess_data["uploaded_documents"].keys()]
         doc_value = next(iter(sess_data["uploaded_documents"]), None) if sess_data["uploaded_documents"] else None
         proposal_options = [{'label': truncate_filename(fn), 'value': fn} for fn in sess_data["proposals"].keys()]
@@ -763,7 +762,7 @@ def unified_master_callback(
             "expanded"
         )
 
-    if ctx.triggered[0].get('prop_id', None) == 'cancel-action-btn.n_clicks':
+    if getattr(ctx, "triggered_id", None) == 'cancel-action-btn':
         lock = sess_data.get("gemini_lock")
         if lock and lock.locked():
             try:
@@ -785,7 +784,7 @@ def unified_master_callback(
             "expanded"
         )
 
-    if ctx.triggered[0].get('prop_id', None) == 'upload-document.contents' and rfp_content is not None and rfp_filename:
+    if getattr(ctx, "triggered_id", None) == 'upload-document' and rfp_content is not None and rfp_filename:
         content_type, content_string = rfp_content.split(',')
         decoded = base64.b64decode(content_string)
         uploaded_rfp_decoded_bytes = decoded
@@ -802,7 +801,7 @@ def unified_master_callback(
         else:
             logging.error(f"[{sid}] Failed to decode uploaded document: {rfp_filename}")
 
-    if ctx.triggered[0].get('prop_id', None) == 'upload-proposal.contents' and proposal_content is not None and proposal_filename:
+    if getattr(ctx, "triggered_id", None) == 'upload-proposal' and proposal_content is not None and proposal_filename:
         content_type, content_string = proposal_content.split(',')
         decoded = base64.b64decode(content_string)
         text = decode_document(decoded)
@@ -829,7 +828,7 @@ def unified_master_callback(
         'recover-action-btn', 'board-action-btn', 'loe-action-btn'
     ]
 
-    if ctx.triggered[0].get('prop_id', '').split('.')[0] in action_btns:
+    if getattr(ctx, "triggered_id", None) in action_btns:
         got_lock = sess_data["gemini_lock"].acquire(blocking=False)
         if not got_lock:
             output_data_upload = html.Div("Another Gemini operation is in progress. Please wait or cancel.", style={"wordWrap": "break-word"})
@@ -840,7 +839,7 @@ def unified_master_callback(
                 "expanded"
             )
         try:
-            triggered_id = ctx.triggered[0].get('prop_id', '').split('.')[0]
+            triggered_id = getattr(ctx, "triggered_id", None)
             if triggered_id == "shred-action-btn":
                 action_name = "shred"
                 result, generated_filename, generated_xlsx_bytes, _, _ = process_document(sess_data, action_name, doc_value, chat_input, uploaded_rfp_decoded_bytes, None)