Update app.py via AI Editor

app.py

@@ -254,7 +254,6 @@ def process_document(action, selected_filename=None, chat_input=None, rfp_decode
             return result, None, None, None, None
 
     elif action == 'recover':
-        # For recover: need compliance check or shred (selected_filename), and proposal (selected_proposal_filename)
         if not selected_proposal_filename or selected_proposal_filename not in proposals:
             logging.error("No proposal document selected for recovery.")
             return "No proposal document selected for recovery.", None, None, None, None
@@ -267,7 +266,6 @@ def process_document(action, selected_filename=None, chat_input=None, rfp_decode
         findings_fileid = uploaded_documents_fileid.get(selected_filename)
         proposal_fileid = proposals_fileid.get(selected_proposal_filename)
 
-        # Upload findings doc if not already uploaded
         if not findings_fileid:
             if selected_filename in shredded_documents:
                 findings_bytes = shredded_documents[selected_filename]
@@ -284,7 +282,6 @@ def process_document(action, selected_filename=None, chat_input=None, rfp_decode
                 except Exception as e:
                     logging.error(f"Failed to upload findings doc {selected_filename} for recover: {e}")
 
-        # Upload proposal doc if not already uploaded
         if not proposal_fileid:
             proposal_bytes = None
             if selected_proposal_filename in proposals:
@@ -315,7 +312,6 @@ def process_document(action, selected_filename=None, chat_input=None, rfp_decode
             "---\nOriginal Proposal:\n"
             f"{proposal_text}\n"
         )
-        # If both are present, use both as file context for Gemini
         used_fileid = None
         if findings_fileid:
             used_fileid = findings_fileid
@@ -507,6 +503,7 @@ app.layout = dbc.Container([
         Input('shred-action-btn', 'n_clicks'),
         Input('proposal-action-btn', 'n_clicks'),
         Input('compliance-action-btn', 'n_clicks'),
+        Input('recover-action-btn', 'n_clicks'),
         Input('upload-document', 'contents'),
         State('upload-document', 'filename'),
         Input({'type': 'delete-doc-btn', 'index': ALL, 'group': 'doc'}, 'n_clicks'),
@@ -521,7 +518,7 @@ app.layout = dbc.Container([
     prevent_initial_call=True
 )
 def master_callback(
-    shred_clicks, proposal_clicks, compliance_clicks,
+    shred_clicks, proposal_clicks, compliance_clicks, recover_clicks,
     rfp_content, rfp_filename, doc_delete_clicks, selected_doc,
     proposal_content, proposal_filename, proposal_delete_clicks, selected_proposal,
     chat_input, cancel_clicks
@@ -535,8 +532,8 @@ def master_callback(
         except Exception:
             return []
 
-    doc_delete_clicks = safe_get_n_clicks(ctx, 5)
-    proposal_delete_clicks = safe_get_n_clicks(ctx, 9)
+    doc_delete_clicks = safe_get_n_clicks(ctx, 6)
+    proposal_delete_clicks = safe_get_n_clicks(ctx, 10)
     uploaded_rfp_decoded_bytes = None
 
     global gemini_lock, uploaded_documents_bytes
@@ -590,7 +587,7 @@ def master_callback(
     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']
+                btn_id = ctx.inputs_list[6][i]['id']
                 del_filename = btn_id['index']
                 if del_filename in uploaded_documents:
                     del uploaded_documents[del_filename]
@@ -613,7 +610,7 @@ def master_callback(
     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']
+                btn_id = ctx.inputs_list[10][i]['id']
                 del_filename = btn_id['index']
                 if del_filename in proposals:
                     del proposals[del_filename]