Update app.py via AI Editor

app.py

@@ -346,7 +346,6 @@ def process_document(action, selected_filename=None, chat_input=None, rfp_decode
             return result, None, None, None, None
 
     elif action == 'loe':
-        # LOE estimation implementation
         if not selected_proposal_filename or selected_proposal_filename not in proposals:
             logging.warning("No proposal document selected for LOE estimation.")
             return "No proposal document selected for LOE estimation.", None, None, None, None
@@ -617,12 +616,7 @@ def master_callback(
 
     global gemini_lock, uploaded_documents_bytes
 
-    if triggered_id in [
-        'shred-action-btn', 'proposal-action-btn', 'compliance-action-btn',
-        'recover-action-btn', 'board-action-btn', 'loe-action-btn'
-    ] and preview_window_state != "shrunk":
-        return dash.no_update, dash.no_update, dash.no_update, dash.no_update, dash.no_update, dash.no_update, dash.no_update, "shrunk"
-
+    # Cancel action
     if triggered_id == 'cancel-action-btn':
         output_data_upload = html.Div("[Cancelled by user]\n", style={"wordWrap": "break-word"})
         doc_options = [{'label': fn, 'value': fn} for fn in uploaded_documents.keys()]
@@ -638,6 +632,7 @@ def master_callback(
             "expanded"
         )
 
+    # Handle 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)
@@ -670,6 +665,7 @@ def master_callback(
         else:
             logging.error(f"Failed to decode uploaded proposal: {proposal_filename}")
 
+    # Handle deletes
     if triggered_id and isinstance(doc_delete_clicks, list):
         for i, n_click in enumerate(doc_delete_clicks):
             if n_click:
@@ -720,7 +716,12 @@ 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_btns = [
+        'shred-action-btn', 'proposal-action-btn', 'compliance-action-btn',
+        'recover-action-btn', 'board-action-btn', 'loe-action-btn'
+    ]
+
+    if triggered_id in action_btns:
         got_lock = 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"})
@@ -747,33 +748,7 @@ def master_callback(
                     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:
-            gemini_lock.release()
-        doc_options = [{'label': fn, 'value': fn} for fn in uploaded_documents.keys()]
-        doc_value = doc_value if doc_value in uploaded_documents else (next(iter(uploaded_documents), None) if uploaded_documents else None)
-        proposal_options = [{'label': fn, 'value': fn} for fn in proposals.keys()]
-        proposal_value = proposal_value if proposal_value in proposals else (next(iter(proposals), None) if proposals else None)
-        documents_list = get_documents_list(uploaded_documents, shredded_documents)
-        proposals_list = get_proposals_list(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 = 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"})
-            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 = uploaded_documents_bytes.get(doc_value, None)
                 result, _, _, generated_filename, generated_docx_bytes = process_document(
@@ -804,7 +779,7 @@ def master_callback(
             output_data_upload,
             documents_list, doc_options, doc_value,
             proposals_list, proposal_options, proposal_value,
-            "expanded"
+            "shrunk"
         )
 
     doc_value = doc_value if doc_value in uploaded_documents else (next(iter(uploaded_documents), None) if uploaded_documents else None)