Update app.py via AI Editor

app.py

@@ -160,6 +160,16 @@ def save_virtual_board_as_docx(board_text, base_filename):
     memf.seek(0)
     return memf.read()
 
+def save_loe_as_docx(loe_text, proposal_filename):
+    doc = Document()
+    doc.add_heading(f"Level of Effort for {proposal_filename}", 0)
+    for line in loe_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
 
@@ -336,7 +346,39 @@ def process_document(action, selected_filename=None, chat_input=None, rfp_decode
             return result, None, None, None, None
 
     elif action == 'loe':
-        return "LOE estimation not implemented yet.", None, None, None, None
+        # LOE estimation implementation
+        if not selected_proposal_filename or selected_proposal_filename not in proposals:
+            logging.warning("No proposal document selected for LOE estimation.")
+            return "No proposal document selected for LOE estimation.", None, None, None, None
+        proposal_text = 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"
+        )
+        if chat_input:
+            prompt += f"User 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"):
+            loe_docx_name = f"{proposal_base_name}_loe.docx"
+            proposals[loe_docx_name] = result
+            proposals_fileid[loe_docx_name] = None
+            docx_bytes = save_loe_as_docx(result, proposal_base_name)
+            logging.info(f"LOE generated and saved as {loe_docx_name}")
+            return result, None, None, loe_docx_name, docx_bytes
+        else:
+            return result, None, None, None, None
+
     return "Action not implemented yet.", None, None, None, None
 
 def get_documents_list(docdict, shreddedict):
@@ -379,7 +421,10 @@ def get_proposals_list(proposaldict):
     for filename in proposaldict:
         file_content = proposaldict[filename]
         try:
-            docx_bytes = save_proposal_as_docx(file_content, filename)
+            if filename.lower().endswith('_loe.docx'):
+                docx_bytes = save_loe_as_docx(file_content, filename)
+            else:
+                docx_bytes = save_proposal_as_docx(file_content, filename)
             b64 = base64.b64encode(docx_bytes).decode('utf-8')
             mime = "application/vnd.openxmlformats-officedocument.wordprocessingml.document"
         except Exception:
@@ -743,9 +788,8 @@ def master_callback(
                 output_data_upload = dcc.Markdown(result, style={"whiteSpace": "pre-wrap", "wordWrap": "break-word"})
             elif triggered_id == "loe-action-btn":
                 action_name = "loe"
-                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
+                    action_name, None, chat_input, None, proposal_value
                 )
                 output_data_upload = dcc.Markdown(result, style={"whiteSpace": "pre-wrap", "wordWrap": "break-word"})
         finally: