Update app.py via AI Editor

app.py

@@ -33,6 +33,26 @@ MAX_OUTPUT_TOKENS = 65536
 
 SESSION_STORE = {}
 
+PROMPT_FILES = {
+    'shred': 'prompt_shred',
+    'loe': 'prompt_LOE',
+    'compliance': 'prompt_compliance',
+    'proposal': 'prompt_proposal',
+    'recover': 'prompt_recover',
+    'virtual_board': 'prompt_virtual_board'
+}
+
+def read_prompt_file(prompt_type):
+    fname = PROMPT_FILES.get(prompt_type)
+    if not fname:
+        return f"[Error: No prompt file mapping for {prompt_type}]"
+    try:
+        with open(fname, 'r', encoding='utf-8') as f:
+            return f.read().strip()
+    except Exception as e:
+        logging.error(f"Could not read prompt file {fname}: {e}")
+        return f"[Error: Could not read prompt file for {prompt_type}]"
+
 def get_session_id_from_cookie(cookie_str):
     if not cookie_str:
         return None
@@ -256,13 +276,9 @@ def process_document(sess_data, action, selected_filename=None, chat_input=None,
         if not doc_content:
             logging.warning("No uploaded document found for shredding.")
             return "No document uploaded.", None, None, None, None
-        prompt = (
-            "Analyze the following RFP/PWS/SOW/RFI and generate a requirements spreadsheet. Return ONLY the markdown spreadsheet and NOTHING else"
-            "Identify requirements by action words like 'shall', 'will', 'perform', etc. Also analye the document and put a column for recommended win theme for each requirement identified. Organize by PWS section and requirement. "
-            "Do not write as if responding to the proposal Return the ONLY markdown spreadsheet and NOTHING else, no intro, no outro, no comments.\n"
-        )
+        prompt = read_prompt_file('shred')
         if chat_input:
-            prompt += f"User additional instructions: {chat_input}\n"
+            prompt += f"\nUser additional instructions: {chat_input}\n"
         prompt += f"\nFile Name: {selected_filename}\n\n"
         prompt += doc_content
         result = gemini_generate_content(prompt, file_id=doc_fileid, chat_input=chat_input)
@@ -284,18 +300,10 @@ def process_document(sess_data, action, selected_filename=None, chat_input=None,
         rfp_text = sess_data["uploaded_documents"][selected_filename]
         proposal_text = sess_data["proposals"][selected_proposal_filename]
         logging.info(f"Compliance check: comparing proposal [{selected_proposal_filename}] to RFP [{selected_filename}]")
-        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"
-        )
+        prompt = read_prompt_file('compliance')
+        prompt += f"\n---\nRFP/SOW/PWS/RFI ({selected_filename}):\n{rfp_text}\n"
+        prompt += "---\nGenerated Proposal Document:\n"
+        prompt += f"{proposal_text}\n"
         result = gemini_generate_content(prompt, file_id=None, chat_input=None)
         if result and not result.startswith("Error"):
             xlsx_bytes = save_markdown_as_xlsx(result, selected_filename)
@@ -315,17 +323,10 @@ def process_document(sess_data, action, selected_filename=None, chat_input=None,
         rfp_text = sess_data["uploaded_documents"][selected_filename]
         proposal_text = sess_data["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. Your job is to extract Section L and Section M (evaluation criteria) from the following RFP/SOW/PWS/RFI if they exist. "
-            "If Section L or Section M are not present, then read all requirements and develop a set of evaluation criteria that would be appropriate for this type of acquisition. "
-            "List each evaluation criterion exactly as it appears in the document or as you have developed. "
-            "For each extracted or developed evaluation criterion, evaluate the provided proposal and rate it as exactly one of: unacceptable, acceptable, good, or outstanding (no other ratings allowed). Add a column to explain the reason for the rating. Also include a column for recommendations to remediate the document so that it gets a better rating "
-            "Return ONLY a markdown spreadsheet/table (NO intro, NO outro, NO comments, NO summary) with exactly these columns: | Evaluation Criteria | Evaluation | Reason | Recommendation |"
-            "Each row should have the criterion exactly as extracted or developed, and the evaluation for how well the proposal meets or exceeds it. 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"
-        )
+        prompt = read_prompt_file('virtual_board')
+        prompt += f"\n---\nRFP/SOW/PWS/RFI ({selected_filename}):\n{rfp_text}\n"
+        prompt += "---\nProposal Document:\n"
+        prompt += f"{proposal_text}\n"
         result = gemini_generate_content(prompt, file_id=None, chat_input=None)
         if result and not result.startswith("Error"):
             xlsx_bytes = save_markdown_as_xlsx(result, selected_filename)
@@ -351,14 +352,9 @@ def process_document(sess_data, action, selected_filename=None, chat_input=None,
                     logging.info(f"RFP file {rfp_filename} uploaded to Gemini for proposal.")
             except Exception as e:
                 logging.error(f"Failed to upload RFP file {rfp_filename} for proposal: {e}")
-        prompt = (
-            "Respond to the following RFP/SOW/PWS/RFI by creating a highly detailed proposal response that follows each section and subsection header. "
-            "The response to each section and subsection will be compliant and compelling, focusing on describing the approach, and how the labor category uses a specific industry standard process in a workflow described in steps, and how technology is used. "
-            "You must show innovation in the approach. Refer to research that validates the approach and cite sources with measurable outcome. "
-            "Be sure to respond in paragraph format only, never use numbering, or bullets, only paragraph describing in detail.Return only the proposal response in markdown and nothing else, no comments, no intro, no outro, just the proposal\n"
-        )
+        prompt = read_prompt_file('proposal')
         if chat_input:
-            prompt += f"User additional instructions: {chat_input}\n"
+            prompt += f"\nUser additional instructions: {chat_input}\n"
         prompt += f"\n---\nRFP/SOW/PWS/RFI ({rfp_filename}):\n{doc_content}\n"
         result = gemini_generate_content(prompt, file_id=rfp_fileid, chat_input=chat_input)
         sess_data["generated_response"] = result
@@ -381,20 +377,11 @@ def process_document(sess_data, action, selected_filename=None, chat_input=None,
 
         findings_content = sess_data["uploaded_documents"][selected_filename]
         proposal_text = sess_data["proposals"][selected_proposal_filename]
-        prompt = (
-            "You are a proposal compliance recovery expert. Use the findings and recommendations table and the original proposal response provided below. "
-            "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"
-        )
+        prompt = read_prompt_file('recover')
         if chat_input:
-            prompt += f"User additional instructions: {chat_input}\n"
-
+            prompt += f"\nUser additional instructions: {chat_input}\n"
         prompt += f"\n---\nFindings and Recommendations Table ({selected_filename}):\n{findings_content}\n"
         prompt += f"\n---\nOriginal Proposal Document ({selected_proposal_filename}):\n{proposal_text}\n"
-
         result = gemini_generate_content(prompt, file_id=None, chat_input=chat_input)
         if result and not result.startswith("Error"):
             base_name = os.path.splitext(selected_proposal_filename)[0]
@@ -414,21 +401,9 @@ def process_document(sess_data, action, selected_filename=None, chat_input=None,
             return "No proposal document selected for LOE estimation.", None, None, None, None
         proposal_text = sess_data["proposals"][selected_proposal_filename]
         proposal_base_name = os.path.splitext(selected_proposal_filename)[0]
-        prompt = (
-            "You are a federal proposal cost and level of effort estimator. "
-            "Analyze the following proposal response which is structured to map to a government RFP, PWS, or SOW. "
-            "For each PWS task or requirement referenced in the proposal, estimate the labor categories and a conservative, overestimated number of hours for each labor category required to accomplish that task. "
-            "Include a management reserve to ensure the estimate is not underestimated. "
-            "Return ONLY a markdown table (NO comments, NO intro, NO outro, NO summary) with the following columns: "
-            "| PWS Task Number | Brief Task Description | Labor Category | Estimated Hours (including management reserve) | "
-            "For each task, if more than one labor category is needed, list each category on its own row under the same task. "
-            "Be sure to break down by detailed task, not just high level sections. "
-            "Be detailed and thorough in your estimation and make sure to overestimate hours to ensure sufficient coverage. "
-            "If the proposal references any assumed task number or section, use it in the table. "
-            "Return ONLY the markdown table, no other text, no intro, no outro.\n\n"
-        )
+        prompt = read_prompt_file('loe')
         if chat_input:
-            prompt += f"User additional instructions: {chat_input}\n"
+            prompt += f"\nUser additional instructions: {chat_input}\n"
         prompt += f"\n---\nProposal Document ({selected_proposal_filename}):\n{proposal_text}\n"
         result = gemini_generate_content(prompt, file_id=None, chat_input=chat_input)
         if result and not result.startswith("Error"):
@@ -737,7 +712,6 @@ def master_callback(
     uploaded_rfp_decoded_bytes = None
 
     if triggered_id == 'cancel-action-btn':
-        # Release Gemini lock if held, to allow cancellation of Gemini operation
         lock = sess_data.get("gemini_lock")
         if lock and lock.locked():
             try: