Update app.py via AI Editor

app.py

@@ -652,25 +652,6 @@ def master_callback(
             del SESSION_LOCKS[sid]
         return html.Div("Session ended. All data cleared.", style={"wordWrap": "break-word"}), None, [], None, None, [], None, "expanded"
 
-    # ACTION BUTTONS: If not shrunk, just shrink and clear preview output, do NOT process action
-    if triggered_id in [
-        'shred-action-btn', 'proposal-action-btn', 'compliance-action-btn',
-        'recover-action-btn', 'board-action-btn', 'loe-action-btn', 'cancel-action-btn'
-    ] and preview_window_state != "shrunk":
-        # Don't alter session, don't clear docs, don't process action
-        doc_options = [{'label': fn, 'value': fn} for fn in session_dict['uploaded_documents'].keys()]
-        doc_value = selected_doc if selected_doc in session_dict['uploaded_documents'] else (next(iter(session_dict['uploaded_documents']), None) if session_dict['uploaded_documents'] else None)
-        proposal_options = [{'label': fn, 'value': fn} for fn in session_dict['proposals'].keys()]
-        proposal_value = selected_proposal if selected_proposal in session_dict['proposals'] else (next(iter(session_dict['proposals']), None) if session_dict['proposals'] else None)
-        documents_list = get_documents_list(session_dict['uploaded_documents'], session_dict['shredded_documents'])
-        proposals_list = get_proposals_list(session_dict['proposals'])
-        return (
-            html.Div("", style={"wordWrap": "break-word"}),
-            documents_list, doc_options, doc_value,
-            proposals_list, proposal_options, proposal_value,
-            "shrunk"
-        )
-
     # Cancel should just clear the preview and expand window, not clear session/docs
     if triggered_id == 'cancel-action-btn':
         doc_options = [{'label': fn, 'value': fn} for fn in session_dict['uploaded_documents'].keys()]
@@ -686,6 +667,7 @@ def master_callback(
             "expanded"
         )
 
+    # File uploads
     if triggered_id == 'upload-document' and rfp_content is not None and rfp_filename:
         content_type, content_string = rfp_content.split(',')
         decoded = base64.b64decode(content_string)
@@ -718,6 +700,7 @@ def master_callback(
         else:
             logging.error(f"Failed to decode uploaded proposal: {proposal_filename}")
 
+    # Deletion handling
     if triggered_id and isinstance(doc_delete_clicks, list):
         for i, n_click in enumerate(doc_delete_clicks):
             if n_click:
@@ -768,7 +751,9 @@ def master_callback(
 
     output_data_upload = html.Div("No action taken yet.", style={"wordWrap": "break-word"})
 
-    if triggered_id in ['shred-action-btn', 'compliance-action-btn', 'board-action-btn']:
+    # Action buttons: always process immediately
+    if triggered_id in ['shred-action-btn', 'compliance-action-btn', 'board-action-btn',
+                        'proposal-action-btn', 'recover-action-btn', 'loe-action-btn']:
         got_lock = 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"})
@@ -795,33 +780,7 @@ def master_callback(
                     session_dict, action_name, doc_value, chat_input, uploaded_rfp_decoded_bytes, proposal_value
                 )
                 output_data_upload = dcc.Markdown(result, style={"whiteSpace": "pre-wrap", "wordWrap": "break-word"})
-        finally:
-            lock.release()
-        doc_options = [{'label': fn, 'value': fn} for fn in session_dict['uploaded_documents'].keys()]
-        doc_value = doc_value if doc_value in session_dict['uploaded_documents'] else (next(iter(session_dict['uploaded_documents']), None) if session_dict['uploaded_documents'] else None)
-        proposal_options = [{'label': fn, 'value': fn} for fn in session_dict['proposals'].keys()]
-        proposal_value = proposal_value if proposal_value in session_dict['proposals'] else (next(iter(session_dict['proposals']), None) if session_dict['proposals'] else None)
-        documents_list = get_documents_list(session_dict['uploaded_documents'], session_dict['shredded_documents'])
-        proposals_list = get_proposals_list(session_dict['proposals'])
-        return (
-            output_data_upload,
-            documents_list, doc_options, doc_value,
-            proposals_list, proposal_options, proposal_value,
-            "expanded"
-        )
-
-    if triggered_id in ['proposal-action-btn', 'recover-action-btn', 'loe-action-btn']:
-        got_lock = 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"})
-            return (
-                output_data_upload,
-                documents_list, doc_options, doc_value,
-                proposals_list, proposal_options, proposal_value,
-                "expanded"
-            )
-        try:
-            if triggered_id == "proposal-action-btn":
+            elif triggered_id == "proposal-action-btn":
                 action_name = "proposal"
                 selected_bytes = session_dict['uploaded_documents_bytes'].get(doc_value, None)
                 result, _, _, generated_filename, generated_docx_bytes = process_document(
@@ -842,6 +801,7 @@ def master_callback(
                 output_data_upload = dcc.Markdown(result, style={"whiteSpace": "pre-wrap", "wordWrap": "break-word"})
         finally:
             lock.release()
+        # Update dropdowns and lists after action (in case doc/proposal was auto-created)
         doc_options = [{'label': fn, 'value': fn} for fn in session_dict['uploaded_documents'].keys()]
         doc_value = doc_value if doc_value in session_dict['uploaded_documents'] else (next(iter(session_dict['uploaded_documents']), None) if session_dict['uploaded_documents'] else None)
         proposal_options = [{'label': fn, 'value': fn} for fn in session_dict['proposals'].keys()]