Update app.py via AI Editor

app.py

@@ -150,6 +150,16 @@ def save_compliance_as_docx(compliance_text, rfp_filename):
     memf.seek(0)
     return memf.read()
 
+def save_virtual_board_as_docx(board_text, base_filename):
+    doc = Document()
+    doc.add_heading(f"Evaluation Board for {base_filename}", 0)
+    for line in board_text.split('\n'):
+        doc.add_paragraph(line)
+    memf = io.BytesIO()
+    doc.save(memf)
+    memf.seek(0)
+    return memf.read()
+
 def process_document(action, selected_filename=None, chat_input=None, rfp_decoded_bytes=None, selected_proposal_filename=None):
     global generated_response
 
@@ -224,7 +234,33 @@ def process_document(action, selected_filename=None, chat_input=None, rfp_decode
             return result, None, None, None, None
 
     elif action == 'virtual_board':
-        return "Virtual board not implemented yet.", None, None, None, None
+        # Implementation for extracting evaluation criteria (L & M) and evaluating proposal
+        if not selected_proposal_filename or selected_proposal_filename not in proposals:
+            return "No proposal document selected for evaluation board.", None, None, None, None
+        if not selected_filename or selected_filename not in uploaded_documents:
+            return "No RFP/SOW/PWS/RFI document selected for evaluation board.", None, None, None, None
+
+        rfp_text = uploaded_documents[selected_filename]
+        proposal_text = proposals[selected_proposal_filename]
+        logging.info(f"Evaluation Board: extracting criteria from RFP [{selected_filename}], evaluating proposal [{selected_proposal_filename}]")
+        prompt = (
+            "You are a federal acquisition evaluation board. Read the following RFP/SOW/PWS/RFI and extract ONLY Section L and Section M (evaluation criteria). "
+            "Then, for each extracted evaluation criterion, evaluate the provided proposal and rate it as either: unacceptable, acceptable, good, or outstanding (no other ratings allowed). "
+            "Return ONLY a markdown spreadsheet/table (NO intro, NO outro, NO comments, NO summary) with exactly these columns: | Evaluation Criteria | Evaluation (unacceptable/acceptable/good/outstanding) |. "
+            "Each row should have: the criterion exactly as extracted from the document, and the evaluation for how well the proposal meets or exceeds it. Do not rewrite or omit any criterion. Only the table, nothing else.\n\n"
+            f"---\nRFP/SOW/PWS/RFI ({selected_filename}):\n{rfp_text}\n"
+            "---\nProposal 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_virtual_board_as_docx(result, selected_filename)
+            board_docx_name = f"{os.path.splitext(selected_filename)[0]}_evaluation_board.docx"
+            uploaded_documents[board_docx_name] = result
+            shredded_documents[board_docx_name] = docx_bytes
+            return result, board_docx_name, docx_bytes, None, None
+        else:
+            return result, None, None, None, None
 
     elif action == 'proposal':
         if not doc_content:
@@ -623,7 +659,9 @@ def master_callback(
                 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)
+                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"})
         finally:
             gemini_lock.release()