Update app.py via AI Editor

app.py

@@ -146,7 +146,7 @@ def save_proposal_as_docx(proposal_text, base_filename):
     memf.seek(0)
     return memf.read()
 
-def process_document(action, selected_filename=None, chat_input=None, rfp_decoded_bytes=None, cancel_event=None, stream=True):
+def process_document(action, selected_filename=None, chat_input=None, rfp_decoded_bytes=None, cancel_event=None, stream=True, selected_generated_filename=None):
     global shredded_document, generated_response, stream_buffer
 
     logging.info(f"Process document called with action: {action}")
@@ -277,7 +277,65 @@ def process_document(action, selected_filename=None, chat_input=None, rfp_decode
         return result_holder["text"], None, result_holder["docx_name"], result_holder["text"]
 
     elif action == 'compliance':
-        return "Compliance checking not implemented yet.", None, None, None
+        # --- COMPLIANCE FUNCTIONALITY ADDED HERE ---
+        # Get selected generated doc (proposal) and selected RFP doc
+        if not selected_generated_filename or selected_generated_filename not in generated_documents:
+            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
+
+        proposal_docx_bytes = generated_documents[selected_generated_filename]
+        rfp_text = uploaded_documents[selected_filename]
+        logging.info(f"Compliance check: comparing proposal [{selected_generated_filename}] to RFP [{selected_filename}]")
+
+        # Decode the proposal docx to text (best-effort, using python-docx)
+        proposal_text = ""
+        try:
+            file_stream = io.BytesIO(proposal_docx_bytes)
+            doc = Document(file_stream)
+            proposal_text = "\n".join([para.text for para in doc.paragraphs])
+            if not proposal_text.strip():
+                proposal_text = "[Unable to extract text from proposal docx]"
+        except Exception as e:
+            proposal_text = "[Error decoding proposal docx: {}]".format(str(e))
+            logging.error("Error extracting text from proposal docx: %s", e)
+
+        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_holder = {"text": None, "docx_name": None}
+
+        def thread_compliance():
+            global stream_buffer
+            stream_buffer["preview"] = ""
+            try:
+                logging.info("Starting Gemini generate_content for compliance check.")
+                for partial in gemini_generate_content_stream(prompt, file_id=None, chat_input=None, cancel_event=cancel_event):
+                    stream_buffer["preview"] = partial
+                    if cancel_event is not None and cancel_event.is_set():
+                        break
+                result = stream_buffer["preview"]
+                result_holder["text"] = result
+                logging.info("Compliance check completed.")
+            except Exception as e:
+                logging.error("Error during compliance check: %s", e)
+                result_holder["text"] = f"Error during compliance check: {e}"
+
+        stream_buffer["preview"] = ""
+        t = Thread(target=thread_compliance)
+        t.start()
+        t.join()
+        return result_holder["text"], None, None, result_holder["text"]
+
     elif action == 'recover':
         return "Recovery not implemented yet.", None, None, None
     elif action == 'board':
@@ -549,6 +607,7 @@ app.layout = dbc.Container([
     [
         Input('shred-action-btn', 'n_clicks'),
         Input('proposal-action-btn', 'n_clicks'),
+        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'),
@@ -573,7 +632,7 @@ app.layout = dbc.Container([
     prevent_initial_call=True
 )
 def master_callback(
-    shred_clicks, proposal_clicks,
+    shred_clicks, proposal_clicks, compliance_clicks,
     rfp_content, rfp_filename, rfp_delete_clicks, selected_doc,
     proposal_content, proposal_filename, proposal_delete_clicks, selected_proposal,
     generated_delete_clicks, selected_generated,
@@ -598,10 +657,10 @@ def master_callback(
     proposal_store = {'text': None, 'docx_name': None}
     streaming = False
 
-    rfp_delete_clicks = safe_get_n_clicks(ctx, 4)
-    proposal_delete_clicks = safe_get_n_clicks(ctx, 8)
-    generated_delete_clicks = safe_get_n_clicks(ctx, 10)
-    shredded_delete_clicks = safe_get_n_clicks(ctx, 12)
+    rfp_delete_clicks = safe_get_n_clicks(ctx, 5)
+    proposal_delete_clicks = safe_get_n_clicks(ctx, 9)
+    generated_delete_clicks = safe_get_n_clicks(ctx, 11)
+    shredded_delete_clicks = safe_get_n_clicks(ctx, 13)
 
     uploaded_rfp_decoded_bytes = None
 
@@ -692,7 +751,7 @@ def master_callback(
     if triggered_id and isinstance(rfp_delete_clicks, list):
         for i, n_click in enumerate(rfp_delete_clicks):
             if n_click:
-                btn_id = ctx.inputs_list[4][i]['id']
+                btn_id = ctx.inputs_list[5][i]['id']
                 del_filename = btn_id['index']
                 if del_filename in uploaded_documents:
                     del uploaded_documents[del_filename]
@@ -711,7 +770,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[8][i]['id']
+                btn_id = ctx.inputs_list[9][i]['id']
                 del_filename = btn_id['index']
                 if del_filename in uploaded_proposals:
                     del uploaded_proposals[del_filename]
@@ -730,7 +789,7 @@ def master_callback(
     if triggered_id and isinstance(generated_delete_clicks, list):
         for i, n_click in enumerate(generated_delete_clicks):
             if n_click:
-                btn_id = ctx.inputs_list[10][i]['id']
+                btn_id = ctx.inputs_list[11][i]['id']
                 del_filename = btn_id['index']
                 if del_filename in generated_documents:
                     del generated_documents[del_filename]
@@ -743,7 +802,7 @@ def master_callback(
     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[12][i]['id']
+                btn_id = ctx.inputs_list[13][i]['id']
                 del_filename = btn_id['index']
                 if del_filename in shredded_documents:
                     del shredded_documents[del_filename]
@@ -762,7 +821,7 @@ def master_callback(
 
     output_data_upload = html.Div("No action taken yet.", style={"wordWrap": "break-word"})
 
-    if triggered_id in ['shred-action-btn', 'proposal-action-btn']:
+    if triggered_id in ['shred-action-btn', 'proposal-action-btn', 'compliance-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"})
@@ -778,12 +837,13 @@ def master_callback(
         stream_event.clear()
         stream_buffer["preview"] = ""
         streaming = True
-        def stream_gemini_thread(action, doc_value, chat_input, rfp_decoded_bytes):
+        def stream_gemini_thread(action, doc_value, chat_input, rfp_decoded_bytes, selected_generated_filename):
             try:
-                process_document(action, doc_value, chat_input, rfp_decoded_bytes, cancel_event=stream_event, stream=True)
+                process_document(action, doc_value, chat_input, rfp_decoded_bytes, cancel_event=stream_event, stream=True, selected_generated_filename=selected_generated_filename)
             finally:
                 gemini_lock.release()
-        t = Thread(target=stream_gemini_thread, args=(('shred' if triggered_id=='shred-action-btn' else 'proposal'), doc_value, chat_input, uploaded_rfp_decoded_bytes))
+        action_name = "shred" if triggered_id=="shred-action-btn" else ("proposal" if triggered_id=="proposal-action-btn" else "compliance")
+        t = Thread(target=stream_gemini_thread, args=(action_name, doc_value, chat_input, uploaded_rfp_decoded_bytes, generated_doc_value))
         t.daemon = True
         t.start()
         output_data_upload = dcc.Markdown("Starting Gemini operation...", style={"whiteSpace": "pre-wrap", "wordWrap": "break-word"})