Update app.py via AI Editor

app.py

@@ -96,13 +96,7 @@ def process_document(action, selected_filename=None, chat_input=None, selected_p
         else:
             doc_content = None
     elif action == "proposal":
-        if selected_proposal and selected_proposal in uploaded_proposals:
-            doc_content = uploaded_proposals[selected_proposal]
-        elif uploaded_proposals:
-            doc_content = next(iter(uploaded_proposals.values()))
-            selected_proposal = next(iter(uploaded_proposals.keys()))
-        else:
-            doc_content = None
+        pass  # handled below
 
     if action == 'shred':
         if not doc_content:
@@ -373,7 +367,7 @@ app.layout = dbc.Container([
                     ]),
                     html.Div([
                         dbc.Button("Shred", id="shred-action-btn", className="me-3 mb-2 btn-primary"),
-                        dbc.Button("Proposal", id="generate-action-btn", className="me-3 mb-2 btn-secondary"),
+                        dbc.Button("Proposal", id="proposal-action-btn", className="me-3 mb-2 btn-secondary"),
                         dbc.Button("Compliance", id="compliance-action-btn", className="me-3 mb-2 btn-tertiary"),
                         dbc.Button("Recover", id="recover-action-btn", className="me-3 mb-2 btn-tertiary"),
                         dbc.Button("Virtual Board", id="board-action-btn", className="me-3 mb-2 btn-tertiary"),
@@ -415,7 +409,7 @@ app.layout = dbc.Container([
         State('select-generated-dropdown', 'value'),
         State('select-generated-dropdown', 'options'),
         Input('shred-action-btn', 'n_clicks'),
-        Input('generate-action-btn', 'n_clicks'),
+        Input('proposal-action-btn', 'n_clicks'),
         Input('compliance-action-btn', 'n_clicks'),
         Input('recover-action-btn', 'n_clicks'),
         Input('board-action-btn', 'n_clicks'),
@@ -432,7 +426,7 @@ def master_callback(
     rfp_content, rfp_filename, rfp_delete_clicks, selected_doc,
     proposal_content, proposal_filename, proposal_delete_clicks, selected_proposal,
     generated_delete_clicks, selected_generated, generated_options,
-    shred_clicks, generate_clicks, compliance_clicks, recover_clicks, board_clicks, loe_clicks,
+    shred_clicks, proposal_clicks, compliance_clicks, recover_clicks, board_clicks, loe_clicks,
     selected_generated_dropdown,
     chat_input, selected_filename, selected_proposal_dropdown, selected_generated_dropdown_state
 ):
@@ -516,13 +510,13 @@ def master_callback(
     output_data_upload = html.Div("No action taken yet.", style={"wordWrap": "break-word"})
 
     action_buttons = [
-        'shred-action-btn', 'generate-action-btn', 'compliance-action-btn',
+        'shred-action-btn', 'proposal-action-btn', 'compliance-action-btn',
         'recover-action-btn', 'board-action-btn', 'loe-action-btn'
     ]
 
-    # Show loading message for Proposal button before running the long operation
-    if triggered_id == 'generate-action-btn':
-        output_data_upload = html.Div("Generating response...", style={"wordWrap": "break-word"})
+    if triggered_id == 'proposal-action-btn':
+        output_data_upload = html.Div("Generating proposal from selected documents...", style={"wordWrap": "break-word"})
+        # This is a progress placeholder, real work will happen in the next callback (Dash pattern)
         return (
             uploaded_doc_list,
             doc_options,
@@ -549,9 +543,14 @@ def master_callback(
                 logging.info(f"Generated docx saved: {generated_docx_name}")
                 new_selected_generated = generated_docx_name
 
-        elif triggered_id == 'generate-action-btn':
-            # This block is now unreachable due to early return above
-            result, generated_docx_bytes, generated_docx_name = process_document('generate', selected_filename, chat_input)
+        elif triggered_id == 'proposal-action-btn':
+            result, generated_docx_bytes, generated_docx_name = process_document(
+                'proposal',
+                selected_filename,
+                chat_input,
+                None,
+                selected_generated_dropdown_state
+            )
             if generated_docx_bytes and generated_docx_name:
                 generated_documents[generated_docx_name] = generated_docx_bytes
                 logging.info(f"Generated proposal docx saved: {generated_docx_name}")