Update app.py via AI Editor

app.py

@@ -30,10 +30,9 @@ MAX_OUTPUT_TOKENS = 65536
 
 uploaded_documents = {}
 uploaded_documents_fileid = {}
-uploaded_proposals = {}
-uploaded_proposals_fileid = {}
+proposals = {}
+proposals_fileid = {}
 shredded_documents = {}
-shredded_document = None
 generated_response = None
 
 gemini_lock = Lock()
@@ -143,14 +142,14 @@ def save_compliance_as_docx(compliance_text, rfp_filename):
     memf.seek(0)
     return memf.read()
 
-def process_document(action, selected_filename=None, chat_input=None, rfp_decoded_bytes=None, selected_generated_filename=None):
-    global shredded_document, generated_response
+def process_document(action, selected_filename=None, chat_input=None, rfp_decoded_bytes=None, selected_proposal_filename=None):
+    global generated_response
 
     logging.info(f"Process document called with action: {action}")
 
     doc_content = None
     doc_fileid = None
-    if action in ["shred", "proposal", "compliance"]:
+    if action in ["shred", "compliance", "virtual_board"]:
         if selected_filename and selected_filename in uploaded_documents:
             doc_content = uploaded_documents[selected_filename]
             doc_fileid = uploaded_documents_fileid.get(selected_filename)
@@ -165,7 +164,7 @@ def process_document(action, selected_filename=None, chat_input=None, rfp_decode
     if action == 'shred':
         if not doc_content:
             logging.warning("No uploaded document found for shredding.")
-            return "No document uploaded.", None, None, None
+            return "No document uploaded.", None, None, None, None
         prompt = (
             "Analyze the following RFP/PWS/SOW/RFI and generate a requirements spreadsheet. Return ONLY the markdown spreadsheet and NOTHING else"
             "Identify requirements by action words like 'shall', 'will', 'perform', etc. Also analye the document and put a column for recommended win theme for each requirement identified. Organize by PWS section and requirement. "
@@ -175,21 +174,53 @@ def process_document(action, selected_filename=None, chat_input=None, rfp_decode
             prompt += f"User additional instructions: {chat_input}\n"
         prompt += f"\nFile Name: {selected_filename}\n\n"
         result = gemini_generate_content(prompt, file_id=doc_fileid, chat_input=chat_input)
-        shredded_document = result
         if result and not result.startswith("Error"):
             docx_bytes = save_shredded_as_docx(result, selected_filename)
             generated_docx_name = f"{os.path.splitext(selected_filename)[0]}_shredded.docx"
+            uploaded_documents[generated_docx_name] = result
             shredded_documents[generated_docx_name] = docx_bytes
-            if generated_docx_name not in uploaded_documents:
-                uploaded_documents[generated_docx_name] = result
-            return result, None, generated_docx_name, result
+            return result, generated_docx_name, docx_bytes, None, None
         else:
-            return result, None, None, result
+            return result, None, None, None, None
+
+    elif action == 'compliance':
+        if not selected_proposal_filename or selected_proposal_filename not in proposals:
+            return "No proposal document selected for compliance.", None, None, None, None
+        if not selected_filename or selected_filename not in uploaded_documents:
+            return "No RFP/SOW/PWS/RFI document selected for compliance.", None, None, None, None
+
+        rfp_text = uploaded_documents[selected_filename]
+        proposal_text = proposals[selected_proposal_filename]
+        logging.info(f"Compliance check: comparing proposal [{selected_proposal_filename}] to RFP [{selected_filename}]")
+        prompt = (
+            "You are a proposal compliance expert. Use the following RFP/SOW/PWS/RFI and the generated proposal response. "
+            "Compare the proposal to the RFP requirements and win themes. "
+            "For each PWS section/requirement, determine if the proposal fully, partially, or does not address the requirement and win theme. "
+            "For each, give a finding and a recommendation to recover for compliance or to strengthen the response. "
+            "Return ONLY a markdown table (NO comments, NO intro, NO outro, NO summary) with the following columns: "
+            "| PWS Section Number | Section Name | Requirement Description | Finding | Recommendation to Recover for Compliance |. "
+            "Be concise and focus on actionable compliance gaps. Only the table, nothing else.\n\n"
+            f"---\nRFP/SOW/PWS/RFI ({selected_filename}):\n{rfp_text}\n"
+            "---\nGenerated Proposal Document:\n"
+            f"{proposal_text}\n"
+        )
+        result = gemini_generate_content(prompt, file_id=None, chat_input=None)
+        if result and not result.startswith("Error"):
+            docx_bytes = save_compliance_as_docx(result, selected_filename)
+            compliance_docx_name = f"{os.path.splitext(selected_filename)[0]}_compliance_check.docx"
+            uploaded_documents[compliance_docx_name] = result
+            shredded_documents[compliance_docx_name] = docx_bytes
+            return result, compliance_docx_name, docx_bytes, None, None
+        else:
+            return result, None, None, None, None
+
+    elif action == 'virtual_board':
+        return "Virtual board not implemented yet.", None, None, None, None
 
     elif action == 'proposal':
         if not doc_content:
             logging.warning("No RFP/SOW/PWS/RFI document selected for proposal action.")
-            return "No RFP/SOW/PWS/RFI document selected.", None, None, None
+            return "No RFP/SOW/PWS/RFI document selected.", None, None, None, None
         rfp_filename = selected_filename
         rfp_fileid = uploaded_documents_fileid.get(selected_filename)
         if not rfp_fileid and rfp_decoded_bytes is not None:
@@ -215,67 +246,36 @@ def process_document(action, selected_filename=None, chat_input=None, rfp_decode
         if result and not result.startswith("Error"):
             docx_bytes = save_proposal_as_docx(result, rfp_filename)
             generated_docx_name = f"{os.path.splitext(rfp_filename)[0]}_proposal.docx"
-            uploaded_proposals[generated_docx_name] = result
-            uploaded_proposals_fileid[generated_docx_name] = None
-            return result, None, generated_docx_name, result
-        else:
-            return result, None, None, result
-
-    elif action == 'compliance':
-        if not selected_generated_filename or selected_generated_filename not in uploaded_proposals:
-            return "No generated document selected for compliance.", None, None, None
-        if not selected_filename or selected_filename not in uploaded_documents:
-            return "No RFP/SOW/PWS/RFI document selected for compliance.", None, None, None
-
-        rfp_text = uploaded_documents[selected_filename]
-        proposal_text = uploaded_proposals[selected_generated_filename]
-        logging.info(f"Compliance check: comparing proposal [{selected_generated_filename}] to RFP [{selected_filename}]")
-
-        prompt = (
-            "You are a proposal compliance expert. Use the following RFP/SOW/PWS/RFI and the generated proposal response. "
-            "Compare the proposal to the RFP requirements and win themes. "
-            "For each PWS section/requirement, determine if the proposal fully, partially, or does not address the requirement and win theme. "
-            "For each, give a finding and a recommendation to recover for compliance or to strengthen the response. "
-            "Return ONLY a markdown table (NO comments, NO intro, NO outro, NO summary) with the following columns: "
-            "| PWS Section Number | Section Name | Requirement Description | Finding | Recommendation to Recover for Compliance |. "
-            "Be concise and focus on actionable compliance gaps. Only the table, nothing else.\n\n"
-            f"---\nRFP/SOW/PWS/RFI ({selected_filename}):\n{rfp_text}\n"
-            "---\nGenerated Proposal Document:\n"
-            f"{proposal_text}\n"
-        )
-        result = gemini_generate_content(prompt, file_id=None, chat_input=None)
-        if result and not result.startswith("Error"):
-            docx_bytes = save_compliance_as_docx(result, selected_filename)
-            compliance_docx_name = f"{os.path.splitext(selected_filename)[0]}_compliance_check.docx"
-            uploaded_documents[compliance_docx_name] = result
-            shredded_documents[compliance_docx_name] = docx_bytes
-            return result, None, compliance_docx_name, result
+            proposals[generated_docx_name] = result
+            proposals_fileid[generated_docx_name] = None
+            return result, None, None, generated_docx_name, docx_bytes
         else:
-            return result, None, None, result
+            return result, None, None, None, None
 
     elif action == 'recover':
-        return "Recovery not implemented yet.", None, None, None
-    elif action == 'board':
-        return "Virtual board not implemented yet.", None, None, None
+        return "Recovery not implemented yet.", None, None, None, None
     elif action == 'loe':
-        return "LOE estimation not implemented yet.", None, None, None
-    return "Action not implemented yet.", None, None, None
-
-def get_uploaded_doc_list(docdict):
-    if not docdict:
-        return html.Div("No documents uploaded.", style={"wordWrap": "break-word"})
+        return "LOE estimation not implemented yet.", None, None, None, None
+    return "Action not implemented yet.", None, None, None, None
+
+def get_documents_list(docdict, shreddedict):
+    all_docs = {}
+    for filename, text in docdict.items():
+        all_docs[filename] = text
+    for filename, docx_bytes in shreddedict.items():
+        if filename not in all_docs:
+            all_docs[filename] = None
+    if not all_docs:
+        return html.Div("No documents uploaded or generated.", style={"wordWrap": "break-word"})
     doc_list = []
-    for filename in docdict:
-        file_content = docdict[filename]
-        mime = "application/vnd.openxmlformats-officedocument.wordprocessingml.document" if filename.lower().endswith('.docx') else "text/plain"
-        try:
-            if filename.lower().endswith('.docx') and filename in shredded_documents:
-                docx_bytes = shredded_documents[filename]
-                b64 = base64.b64encode(docx_bytes).decode('utf-8')
-            else:
-                b64 = base64.b64encode(file_content.encode('utf-8')).decode('utf-8')
-        except Exception:
-            b64 = base64.b64encode(file_content.encode('utf-8')).decode('utf-8')
+    for filename in all_docs:
+        if filename.lower().endswith('.docx') and filename in shreddedict:
+            b64 = base64.b64encode(shreddedict[filename]).decode('utf-8')
+            mime = "application/vnd.openxmlformats-officedocument.wordprocessingml.document"
+        else:
+            content = docdict.get(filename, "")
+            b64 = base64.b64encode(content.encode('utf-8')).decode('utf-8')
+            mime = "text/plain"
         download_link = html.A(
             filename,
             href=f"data:{mime};base64,{b64}",
@@ -286,44 +286,24 @@ def get_uploaded_doc_list(docdict):
         doc_list.append(
             dbc.ListGroupItem([
                 download_link,
-                dbc.Button("Delete", id={'type': 'delete-doc-btn', 'index': filename, 'group': 'rfp'}, size="sm", color="danger", className="float-end ms-2")
+                dbc.Button("Delete", id={'type': 'delete-doc-btn', 'index': filename, 'group': 'doc'}, size="sm", color="danger", className="float-end ms-2")
             ], className="d-flex justify-content-between align-items-center")
         )
     return dbc.ListGroup(doc_list, flush=True)
 
-def get_shredded_doc_list(shreddict):
-    if not shreddict:
-        return html.Div("No shredded requirements yet.", style={"wordWrap": "break-word"})
+def get_proposals_list(proposaldict):
+    if not proposaldict:
+        return html.Div("No proposals uploaded or generated.", style={"wordWrap": "break-word"})
     doc_list = []
-    for filename in shreddict:
-        b64 = base64.b64encode(shreddict[filename]).decode('utf-8')
-        download_link = html.A(
-            filename,
-            href=f"data:application/vnd.openxmlformats-officedocument.wordprocessingml.document;base64,{b64}",
-            download=filename,
-            target="_blank",
-            style={"wordWrap": "break-word", "marginRight": "10px", "textDecoration": "underline"}
-        )
-        doc_list.append(
-            dbc.ListGroupItem([
-                download_link,
-                dbc.Button("Delete", id={'type': 'delete-shredded-btn', 'index': filename, 'group': 'shredded'}, size="sm", color="danger", className="float-end ms-2")
-            ], className="d-flex justify-content-between align-items-center")
-        )
-    return dbc.ListGroup(doc_list, flush=True)
-
-def get_uploaded_proposal_list(docdict):
-    if not docdict:
-        return html.Div("No proposal documents uploaded.", style={"wordWrap": "break-word"})
-    doc_list = []
-    for filename in docdict:
-        file_content = docdict[filename]
-        mime = "application/vnd.openxmlformats-officedocument.wordprocessingml.document"
+    for filename in proposaldict:
+        file_content = proposaldict[filename]
         try:
             docx_bytes = save_proposal_as_docx(file_content, filename)
             b64 = base64.b64encode(docx_bytes).decode('utf-8')
+            mime = "application/vnd.openxmlformats-officedocument.wordprocessingml.document"
         except Exception:
             b64 = base64.b64encode(file_content.encode('utf-8')).decode('utf-8')
+            mime = "text/plain"
         download_link = html.A(
             filename,
             href=f"data:{mime};base64,{b64}",
@@ -343,17 +323,8 @@ app.layout = dbc.Container([
     dbc.Row([
         dbc.Col([
             dbc.Card([
-                dbc.CardHeader(html.H5("RFP/SOW/PWS/RFI")),
+                dbc.CardHeader(html.H5("Documents")),
                 dbc.CardBody([
-                    html.H6("Uploaded Documents"),
-                    html.Div(get_uploaded_doc_list(uploaded_documents), id='uploaded-doc-list'),
-                    dcc.Dropdown(
-                        id='select-document-dropdown',
-                        options=[{'label': fn, 'value': fn} for fn in uploaded_documents.keys()],
-                        placeholder="Select a document to work with",
-                        value=next(iter(uploaded_documents), None),
-                        style={"marginBottom": "10px"}
-                    ),
                     dcc.Upload(
                         id='upload-document',
                         children=html.Div([
@@ -372,23 +343,19 @@ app.layout = dbc.Container([
                         },
                         multiple=False
                     ),
-                    html.Hr(style={"marginTop": "20px", "marginBottom": "10px"}),
-                    html.H6("Shredded/Compliance Documents"),
-                    html.Div(get_shredded_doc_list(shredded_documents), id='shredded-doc-list')
+                    html.Div(id='documents-list', children=get_documents_list(uploaded_documents, shredded_documents)),
+                    dcc.Dropdown(
+                        id='select-document-dropdown',
+                        options=[{'label': fn, 'value': fn} for fn in uploaded_documents.keys()],
+                        placeholder="Select a document to work with",
+                        value=next(iter(uploaded_documents), None),
+                        style={"marginBottom": "10px"}
+                    ),
                 ])
             ], className="mb-3"),
             dbc.Card([
-                dbc.CardHeader(html.H5("Proposal")),
+                dbc.CardHeader(html.H5("Proposals")),
                 dbc.CardBody([
-                    html.H6("Generated Proposals"),
-                    html.Div(get_uploaded_proposal_list(uploaded_proposals), id='uploaded-proposal-list'),
-                    dcc.Dropdown(
-                        id='select-proposal-dropdown',
-                        options=[{'label': fn, 'value': fn} for fn in uploaded_proposals.keys()],
-                        placeholder="Select a proposal document",
-                        value=next(iter(uploaded_proposals), None),
-                        style={"marginBottom": "10px"}
-                    ),
                     dcc.Upload(
                         id='upload-proposal',
                         children=html.Div([
@@ -406,7 +373,15 @@ app.layout = dbc.Container([
                             'margin': '10px'
                         },
                         multiple=False
-                    )
+                    ),
+                    html.Div(id='proposals-list', children=get_proposals_list(proposals)),
+                    dcc.Dropdown(
+                        id='select-proposal-dropdown',
+                        options=[{'label': fn, 'value': fn} for fn in proposals.keys()],
+                        placeholder="Select a proposal document",
+                        value=next(iter(proposals), None),
+                        style={"marginBottom": "10px"}
+                    ),
                 ])
             ], className="mb-3"),
         ], style={'minWidth': '260px', 'width':'30vw','maxWidth':'30vw'}, width=3),
@@ -441,11 +416,10 @@ app.layout = dbc.Container([
 
 @app.callback(
     Output('output-data-upload', 'children'),
-    Output('uploaded-doc-list', 'children'),
+    Output('documents-list', 'children'),
     Output('select-document-dropdown', 'options'),
     Output('select-document-dropdown', 'value'),
-    Output('shredded-doc-list', 'children'),
-    Output('uploaded-proposal-list', 'children'),
+    Output('proposals-list', 'children'),
     Output('select-proposal-dropdown', 'options'),
     Output('select-proposal-dropdown', 'value'),
     [
@@ -454,14 +428,12 @@ app.layout = dbc.Container([
         Input('compliance-action-btn', 'n_clicks'),
         Input('upload-document', 'contents'),
         State('upload-document', 'filename'),
-        Input({'type': 'delete-doc-btn', 'index': ALL, 'group': 'rfp'}, 'n_clicks'),
+        Input({'type': 'delete-doc-btn', 'index': ALL, 'group': 'doc'}, 'n_clicks'),
         State('select-document-dropdown', 'value'),
         Input('upload-proposal', 'contents'),
         State('upload-proposal', 'filename'),
         Input({'type': 'delete-proposal-btn', 'index': ALL, 'group': 'proposal'}, 'n_clicks'),
         State('select-proposal-dropdown', 'value'),
-        Input({'type': 'delete-shredded-btn', 'index': ALL, 'group': 'shredded'}, 'n_clicks'),
-        State('shredded-doc-list', 'children'),
         State('chat-input', 'value'),
         Input('cancel-action-btn', 'n_clicks')
     ],
@@ -469,9 +441,8 @@ app.layout = dbc.Container([
 )
 def master_callback(
     shred_clicks, proposal_clicks, compliance_clicks,
-    rfp_content, rfp_filename, rfp_delete_clicks, selected_doc,
+    rfp_content, rfp_filename, doc_delete_clicks, selected_doc,
     proposal_content, proposal_filename, proposal_delete_clicks, selected_proposal,
-    shredded_delete_clicks, shredded_doc_children,
     chat_input, cancel_clicks
 ):
     ctx = callback_context
@@ -483,12 +454,8 @@ def master_callback(
         except Exception:
             return []
 
-    upload_triggered = False
-
-    rfp_delete_clicks = safe_get_n_clicks(ctx, 5)
+    doc_delete_clicks = safe_get_n_clicks(ctx, 5)
     proposal_delete_clicks = safe_get_n_clicks(ctx, 9)
-    shredded_delete_clicks = safe_get_n_clicks(ctx, 11)
-
     uploaded_rfp_decoded_bytes = None
 
     global gemini_lock
@@ -497,16 +464,14 @@ def master_callback(
         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()]
         doc_value = selected_doc if selected_doc in uploaded_documents else (next(iter(uploaded_documents), None) if uploaded_documents else None)
-        shredded_doc_list_items = get_shredded_doc_list(shredded_documents)
-        uploaded_doc_list = get_uploaded_doc_list(uploaded_documents)
-        uploaded_proposal_list = get_uploaded_proposal_list(uploaded_proposals)
-        proposal_options = [{'label': fn, 'value': fn} for fn in uploaded_proposals.keys()]
-        proposal_value = selected_proposal if selected_proposal in uploaded_proposals else (next(iter(uploaded_proposals), None) if uploaded_proposals else None)
+        proposals_list = get_proposals_list(proposals)
+        proposal_options = [{'label': fn, 'value': fn} for fn in proposals.keys()]
+        proposal_value = selected_proposal if selected_proposal in proposals else (next(iter(proposals), None) if proposals else None)
+        documents_list = get_documents_list(uploaded_documents, shredded_documents)
         return (
             output_data_upload,
-            uploaded_doc_list, doc_options, doc_value,
-            shredded_doc_list_items,
-            uploaded_proposal_list, proposal_options, proposal_value
+            documents_list, doc_options, doc_value,
+            proposals_list, proposal_options, proposal_value
         )
 
     if triggered_id == 'upload-document' and rfp_content is not None and rfp_filename:
@@ -524,7 +489,6 @@ def master_callback(
             logging.info(f"Document uploaded: {rfp_filename}")
         else:
             logging.error(f"Failed to decode uploaded document: {rfp_filename}")
-        upload_triggered = True
 
     if triggered_id == 'upload-proposal' and proposal_content is not None and proposal_filename:
         content_type, content_string = proposal_content.split(',')
@@ -534,16 +498,15 @@ def master_callback(
         if proposal_filename.lower().endswith(('.pdf', '.docx', '.xlsx', '.xls')):
             fileid = upload_to_gemini_file(decoded, proposal_filename)
         if text is not None:
-            uploaded_proposals[proposal_filename] = text
+            proposals[proposal_filename] = text
             if fileid:
-                uploaded_proposals_fileid[proposal_filename] = fileid
+                proposals_fileid[proposal_filename] = fileid
             logging.info(f"Proposal uploaded: {proposal_filename}")
         else:
             logging.error(f"Failed to decode uploaded proposal: {proposal_filename}")
-        upload_triggered = True
 
-    if triggered_id and isinstance(rfp_delete_clicks, list):
-        for i, n_click in enumerate(rfp_delete_clicks):
+    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[5][i]['id']
                 del_filename = btn_id['index']
@@ -556,99 +519,121 @@ def master_callback(
                             logging.warning(f"Failed to delete Gemini file {del_filename}: {e}")
                         del uploaded_documents_fileid[del_filename]
                     logging.info(f"Document deleted: {del_filename}")
-                    if selected_doc == del_filename:
-                        selected_doc = None
-                    upload_triggered = True
-                    break
+                if del_filename in shredded_documents:
+                    del shredded_documents[del_filename]
+                    logging.info(f"Shredded doc 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[9][i]['id']
                 del_filename = btn_id['index']
-                if del_filename in uploaded_proposals:
-                    del uploaded_proposals[del_filename]
-                    if del_filename in uploaded_proposals_fileid:
+                if del_filename in proposals:
+                    del proposals[del_filename]
+                    if del_filename in proposals_fileid:
                         try:
-                            genai.delete_file(uploaded_proposals_fileid[del_filename])
+                            genai.delete_file(proposals_fileid[del_filename])
                         except Exception as e:
                             logging.warning(f"Failed to delete Gemini proposal file {del_filename}: {e}")
-                        del uploaded_proposals_fileid[del_filename]
+                        del proposals_fileid[del_filename]
                     logging.info(f"Proposal deleted: {del_filename}")
-                    if selected_proposal == del_filename:
-                        selected_proposal = None
-                    upload_triggered = True
-                    break
-
-    if triggered_id and isinstance(shredded_delete_clicks, list):
-        for i, n_click in enumerate(shredded_delete_clicks):
-            if n_click:
-                btn_id = ctx.inputs_list[11][i]['id']
-                del_filename = btn_id['index']
-                if del_filename in shredded_documents:
-                    del shredded_documents[del_filename]
-                    logging.info(f"Shredded doc deleted: {del_filename}")
-                upload_triggered = True
+                if selected_proposal == del_filename:
+                    selected_proposal = None
                 break
 
     doc_options = [{'label': fn, 'value': fn} for fn in uploaded_documents.keys()]
     doc_value = selected_doc if selected_doc in uploaded_documents else (next(iter(uploaded_documents), None) if uploaded_documents else None)
-    shredded_doc_list_items = get_shredded_doc_list(shredded_documents)
-    uploaded_doc_list = get_uploaded_doc_list(uploaded_documents)
-    uploaded_proposal_list = get_uploaded_proposal_list(uploaded_proposals)
-    proposal_options = [{'label': fn, 'value': fn} for fn in uploaded_proposals.keys()]
-    proposal_value = selected_proposal if selected_proposal in uploaded_proposals else (next(iter(uploaded_proposals), None) if uploaded_proposals else None)
+    documents_list = get_documents_list(uploaded_documents, shredded_documents)
+    proposals_list = get_proposals_list(proposals)
+    proposal_options = [{'label': fn, 'value': fn} for fn in proposals.keys()]
+    proposal_value = selected_proposal if selected_proposal in proposals else (next(iter(proposals), None) if proposals else None)
 
     output_data_upload = html.Div("No action taken yet.", style={"wordWrap": "break-word"})
 
-    if triggered_id in ['shred-action-btn', 'proposal-action-btn', 'compliance-action-btn']:
+    if triggered_id in ['shred-action-btn', 'compliance-action-btn', 'board-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,
-                uploaded_doc_list, doc_options, doc_value,
-                shredded_doc_list_items,
-                uploaded_proposal_list, proposal_options, proposal_value
+                documents_list, doc_options, doc_value,
+                proposals_list, proposal_options, proposal_value
             )
         try:
-            action_name = "shred" if triggered_id=="shred-action-btn" else ("proposal" if triggered_id=="proposal-action-btn" else "compliance")
-            result, _, _, _ = process_document(action_name, doc_value, chat_input, uploaded_rfp_decoded_bytes, selected_proposal)
-            output_data_upload = dcc.Markdown(result, style={"whiteSpace": "pre-wrap", "wordWrap": "break-word"})
+            if triggered_id == "shred-action-btn":
+                action_name = "shred"
+                result, generated_filename, generated_docx_bytes, _, _ = process_document(action_name, doc_value, chat_input, uploaded_rfp_decoded_bytes, None)
+                output_data_upload = dcc.Markdown(result, style={"whiteSpace": "pre-wrap", "wordWrap": "break-word"})
+            elif triggered_id == "compliance-action-btn":
+                action_name = "compliance"
+                result, generated_filename, generated_docx_bytes, _, _ = process_document(
+                    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"})
+            elif triggered_id == "board-action-btn":
+                action_name = "virtual_board"
+                result, generated_filename, generated_docx_bytes, _, _ = process_document(action_name, doc_value, chat_input, uploaded_rfp_decoded_bytes, None)
+                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 uploaded_proposals.keys()]
-        proposal_value = proposal_value if proposal_value in uploaded_proposals else (next(iter(uploaded_proposals), None) if uploaded_proposals else None)
-        shredded_doc_list_items = get_shredded_doc_list(shredded_documents)
-        uploaded_doc_list = get_uploaded_doc_list(uploaded_documents)
-        uploaded_proposal_list = get_uploaded_proposal_list(uploaded_proposals)
+        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,
-            uploaded_doc_list, doc_options, doc_value,
-            shredded_doc_list_items,
-            uploaded_proposal_list, proposal_options, proposal_value
+            documents_list, doc_options, doc_value,
+            proposals_list, proposal_options, proposal_value
         )
 
-    if upload_triggered:
+    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
+            )
+        try:
+            if triggered_id == "proposal-action-btn":
+                action_name = "proposal"
+            elif triggered_id == "recover-action-btn":
+                action_name = "recover"
+            elif triggered_id == "loe-action-btn":
+                action_name = "loe"
+            else:
+                action_name = None
+            if action_name:
+                result, _, _, generated_filename, generated_docx_bytes = process_document(
+                    action_name, doc_value, chat_input, uploaded_rfp_decoded_bytes, None
+                )
+                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_value = proposal_value if proposal_value in uploaded_proposals else (next(iter(uploaded_proposals), None) if uploaded_proposals else None)
-        output_data_upload = html.Div("Upload/Delete completed.", style={"wordWrap": "break-word"})
+        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,
-            uploaded_doc_list, doc_options, doc_value,
-            shredded_doc_list_items,
-            uploaded_proposal_list, proposal_options, proposal_value
+            documents_list, doc_options, doc_value,
+            proposals_list, proposal_options, proposal_value
         )
 
     doc_value = doc_value if doc_value in uploaded_documents else (next(iter(uploaded_documents), None) if uploaded_documents else None)
-    proposal_value = proposal_value if proposal_value in uploaded_proposals else (next(iter(uploaded_proposals), None) if uploaded_proposals else None)
+    proposal_value = proposal_value if proposal_value in proposals else (next(iter(proposals), None) if proposals else None)
     return (
         output_data_upload,
-        uploaded_doc_list, doc_options, doc_value,
-        shredded_doc_list_items,
-        uploaded_proposal_list, proposal_options, proposal_value
+        documents_list, doc_options, doc_value,
+        proposals_list, proposal_options, proposal_value
     )
 
 if __name__ == '__main__':