Update app.py via AI Editor

app.py

@@ -150,7 +150,7 @@ def process_document(action, selected_filename=None, chat_input=None, rfp_decode
 
     doc_content = None
     doc_fileid = None
-    if action in ["shred", "compliance", "virtual_board", "proposal"]:
+    if action in ["shred", "compliance", "virtual_board", "proposal", "recover"]:
         if selected_filename and selected_filename in uploaded_documents:
             doc_content = uploaded_documents[selected_filename]
             doc_fileid = uploaded_documents_fileid.get(selected_filename)
@@ -224,7 +224,6 @@ def process_document(action, selected_filename=None, chat_input=None, rfp_decode
             return "No RFP/SOW/PWS/RFI document selected.", None, None, None, None
         rfp_filename = selected_filename
         rfp_fileid = uploaded_documents_fileid.get(selected_filename)
-        # Try to upload to Gemini if fileid missing and we have bytes
         if not rfp_fileid and rfp_filename in uploaded_documents_bytes:
             try:
                 fileid = upload_to_gemini_file(uploaded_documents_bytes[rfp_filename], rfp_filename)
@@ -255,7 +254,38 @@ def process_document(action, selected_filename=None, chat_input=None, rfp_decode
             return result, None, None, None, None
 
     elif action == 'recover':
-        return "Recovery not implemented yet.", None, None, None, None
+        # 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:
+            return "No proposal document selected for recovery.", None, None, None, None
+        if not selected_filename or selected_filename not in uploaded_documents:
+            return "No compliance check or shredded requirements document selected for recovery.", None, None, None, None
+        findings_content = uploaded_documents[selected_filename]
+        proposal_text = proposals[selected_proposal_filename]
+        logging.info(f"Recovery: fixing proposal [{selected_proposal_filename}] based on findings [{selected_filename}]")
+        prompt = (
+            "You are a proposal compliance recovery expert. Use the following findings and recommendations table, and the original proposal response. "
+            "Address ONLY those sections of the proposal that have a finding and a recommendation for improvement. "
+            "Do NOT change sections that are marked as fully compliant or already fully address the requirements. "
+            "For each section that needs fixing, revise it in the proposal to address the recommendation for compliance or to strengthen the response. "
+            "Return the full proposal with only the necessary sections revised, and leave all other sections untouched. "
+            "Do not add any introduction, summary, or comments. Return only the revised proposal in markdown, nothing else.\n\n"
+            "Findings and Recommendations Table or Requirements Table:\n"
+            f"{findings_content}\n"
+            "---\nOriginal Proposal:\n"
+            f"{proposal_text}\n"
+        )
+        result = gemini_generate_content(prompt, file_id=None, chat_input=chat_input)
+        if result and not result.startswith("Error"):
+            # Name for the recovered proposal: <original proposal base>_recovered.docx
+            base_name = os.path.splitext(selected_proposal_filename)[0]
+            recovered_docx_name = f"{base_name}_recovered.docx"
+            docx_bytes = save_proposal_as_docx(result, base_name)
+            proposals[recovered_docx_name] = result
+            proposals_fileid[recovered_docx_name] = None
+            return result, None, None, recovered_docx_name, docx_bytes
+        else:
+            return result, None, None, None, None
+
     elif action == 'loe':
         return "LOE estimation not implemented yet.", None, None, None, None
     return "Action not implemented yet.", None, None, None, None
@@ -608,15 +638,21 @@ def master_callback(
         try:
             if 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(
+                    action_name, doc_value, chat_input, selected_bytes, None
+                )
+                output_data_upload = dcc.Markdown(result, style={"whiteSpace": "pre-wrap", "wordWrap": "break-word"})
             elif triggered_id == "recover-action-btn":
                 action_name = "recover"
+                # For recover, use selected compliance/shred doc (doc_value) and selected proposal (proposal_value)
+                result, _, _, generated_filename, generated_docx_bytes = process_document(
+                    action_name, doc_value, chat_input, None, proposal_value
+                )
+                output_data_upload = dcc.Markdown(result, style={"whiteSpace": "pre-wrap", "wordWrap": "break-word"})
             elif triggered_id == "loe-action-btn":
                 action_name = "loe"
-            else:
-                action_name = None
-            # For proposal, always use the selected doc_value and its bytes if available
-            selected_bytes = uploaded_documents_bytes.get(doc_value, None)
-            if action_name:
+                selected_bytes = uploaded_documents_bytes.get(doc_value, None)
                 result, _, _, generated_filename, generated_docx_bytes = process_document(
                     action_name, doc_value, chat_input, selected_bytes, None
                 )