Update app.py via AI Editor

app.py

@@ -161,6 +161,7 @@ def gemini_generate_content(prompt, file_id=None, chat_input=None, file_ids=None
             content_list.extend(files)
             content_list.append("\n\n")
         content_list.append(prompt)
+        logging.info(f"Prompt sent to Gemini: {prompt[:500]}...")  # log first 500 chars
         model = genai.GenerativeModel(GEMINI_MODEL)
         response = model.generate_content(
             contents=content_list,
@@ -169,6 +170,7 @@ def gemini_generate_content(prompt, file_id=None, chat_input=None, file_ids=None
             )
         )
         result = response.text if hasattr(response, "text") else str(response)
+        logging.info(f"Gemini response (first 500 chars): {str(result)[:500]}...")
         return result
     except Exception as e:
         logging.error("Error during Gemini generate_content: %s", e)
@@ -260,17 +262,16 @@ def save_loe_as_docx(loe_text, proposal_filename):
 def process_document(sess_data, action, selected_filename=None, chat_input=None, rfp_decoded_bytes=None, selected_proposal_filename=None):
     doc_content = None
     doc_fileid = None
-    if action in ["shred", "compliance", "virtual_board", "proposal", "recover"]:
-        if selected_filename and selected_filename in sess_data["uploaded_documents"]:
-            doc_content = sess_data["uploaded_documents"][selected_filename]
-            doc_fileid = sess_data["uploaded_documents_fileid"].get(selected_filename)
-        elif sess_data["uploaded_documents"]:
-            doc_content = next(iter(sess_data["uploaded_documents"].values()))
-            selected_filename = next(iter(sess_data["uploaded_documents"].keys()))
-            doc_fileid = sess_data["uploaded_documents_fileid"].get(selected_filename)
-        else:
-            doc_content = None
-            doc_fileid = None
+    proposal_content = None
+    proposal_fileid = None
+
+    # Always find document and proposal content if possible
+    if selected_filename and selected_filename in sess_data["uploaded_documents"]:
+        doc_content = sess_data["uploaded_documents"][selected_filename]
+        doc_fileid = sess_data["uploaded_documents_fileid"].get(selected_filename)
+    if selected_proposal_filename and selected_proposal_filename in sess_data["proposals"]:
+        proposal_content = sess_data["proposals"][selected_proposal_filename]
+        proposal_fileid = sess_data["proposals_fileid"].get(selected_proposal_filename)
 
     if action == 'shred':
         if not doc_content:
@@ -281,6 +282,7 @@ def process_document(sess_data, action, selected_filename=None, chat_input=None,
             prompt += f"\nUser additional instructions: {chat_input}\n"
         prompt += f"\nFile Name: {selected_filename}\n\n"
         prompt += doc_content
+        logging.info(f"[SHRED] Sending document {selected_filename} to Gemini for shredding.")
         result = gemini_generate_content(prompt, file_id=doc_fileid, chat_input=chat_input)
         if result and not result.startswith("Error"):
             xlsx_bytes = save_markdown_as_xlsx(result, selected_filename)
@@ -292,18 +294,16 @@ def process_document(sess_data, action, selected_filename=None, chat_input=None,
             return result, None, None, None, None
 
     elif action == 'compliance':
-        if not selected_proposal_filename or selected_proposal_filename not in sess_data["proposals"]:
+        if not proposal_content:
             return "No proposal document selected for compliance.", None, None, None, None
-        if not selected_filename or selected_filename not in sess_data["uploaded_documents"]:
+        if not doc_content:
             return "No RFP/SOW/PWS/RFI document selected for compliance.", None, None, None, 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}]")
+        logging.info(f"[COMPLIANCE] Comparing proposal [{selected_proposal_filename}] to RFP [{selected_filename}]")
         prompt = read_prompt_file('compliance')
-        prompt += f"\n---\nRFP/SOW/PWS/RFI ({selected_filename}):\n{rfp_text}\n"
+        prompt += f"\n---\nRFP/SOW/PWS/RFI ({selected_filename}):\n{doc_content}\n"
         prompt += "---\nGenerated Proposal Document:\n"
-        prompt += f"{proposal_text}\n"
+        prompt += f"{proposal_content}\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,18 +315,16 @@ def process_document(sess_data, action, selected_filename=None, chat_input=None,
             return result, None, None, None, None
 
     elif action == 'virtual_board':
-        if not selected_proposal_filename or selected_proposal_filename not in sess_data["proposals"]:
+        if not proposal_content:
             return "No proposal document selected for evaluation board.", None, None, None, None
-        if not selected_filename or selected_filename not in sess_data["uploaded_documents"]:
+        if not doc_content:
             return "No RFP/SOW/PWS/RFI document selected for evaluation board.", None, None, None, 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}]")
+        logging.info(f"[VIRTUAL_BOARD] Evaluating proposal [{selected_proposal_filename}] against RFP [{selected_filename}]")
         prompt = read_prompt_file('virtual_board')
-        prompt += f"\n---\nRFP/SOW/PWS/RFI ({selected_filename}):\n{rfp_text}\n"
+        prompt += f"\n---\nRFP/SOW/PWS/RFI ({selected_filename}):\n{doc_content}\n"
         prompt += "---\nProposal Document:\n"
-        prompt += f"{proposal_text}\n"
+        prompt += f"{proposal_content}\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)
@@ -342,7 +340,7 @@ def process_document(sess_data, action, selected_filename=None, chat_input=None,
             logging.warning("No RFP/SOW/PWS/RFI document selected for proposal action.")
             return "No RFP/SOW/PWS/RFI document selected.", None, None, None, None
         rfp_filename = selected_filename
-        rfp_fileid = sess_data["uploaded_documents_fileid"].get(selected_filename)
+        rfp_fileid = doc_fileid
         if not rfp_fileid and rfp_filename in sess_data["uploaded_documents_bytes"]:
             try:
                 fileid = upload_to_gemini_file(sess_data["uploaded_documents_bytes"][rfp_filename], rfp_filename)
@@ -356,6 +354,7 @@ def process_document(sess_data, action, selected_filename=None, chat_input=None,
         if chat_input:
             prompt += f"\nUser additional instructions: {chat_input}\n"
         prompt += f"\n---\nRFP/SOW/PWS/RFI ({rfp_filename}):\n{doc_content}\n"
+        logging.info(f"[PROPOSAL] Sending document {rfp_filename} to Gemini for proposal generation.")
         result = gemini_generate_content(prompt, file_id=rfp_fileid, chat_input=chat_input)
         sess_data["generated_response"] = result
         if result and not result.startswith("Error"):
@@ -368,20 +367,20 @@ def process_document(sess_data, action, selected_filename=None, chat_input=None,
             return result, None, None, None, None
 
     elif action == 'recover':
-        if not selected_proposal_filename or selected_proposal_filename not in sess_data["proposals"]:
+        if not proposal_content:
             logging.error("No proposal document selected for recovery.")
             return "No proposal document selected for recovery.", None, None, None, None
-        if not selected_filename or selected_filename not in sess_data["uploaded_documents"]:
+        if not doc_content:
             logging.error("No compliance check or shredded requirements document selected for recovery.")
             return "No compliance check or shredded requirements document selected for recovery.", None, None, None, None
 
-        findings_content = sess_data["uploaded_documents"][selected_filename]
-        proposal_text = sess_data["proposals"][selected_proposal_filename]
+        findings_content = doc_content
         prompt = read_prompt_file('recover')
         if chat_input:
             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"
+        prompt += f"\n---\nOriginal Proposal Document ({selected_proposal_filename}):\n{proposal_content}\n"
+        logging.info(f"[RECOVER] Recovering proposal {selected_proposal_filename} using findings from {selected_filename}.")
         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]
@@ -396,15 +395,15 @@ def process_document(sess_data, action, selected_filename=None, chat_input=None,
             return result, None, None, None, None
 
     elif action == 'loe':
-        if not selected_proposal_filename or selected_proposal_filename not in sess_data["proposals"]:
+        if not proposal_content:
             logging.warning("No proposal document selected for LOE estimation.")
             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 = read_prompt_file('loe')
         if chat_input:
             prompt += f"\nUser additional instructions: {chat_input}\n"
-        prompt += f"\n---\nProposal Document ({selected_proposal_filename}):\n{proposal_text}\n"
+        prompt += f"\n---\nProposal Document ({selected_proposal_filename}):\n{proposal_content}\n"
+        logging.info(f"[LOE] Calculating LOE for proposal {selected_proposal_filename}.")
         result = gemini_generate_content(prompt, file_id=None, chat_input=chat_input)
         if result and not result.startswith("Error"):
             loe_xlsx_name = f"{proposal_base_name}_loe.xlsx"
@@ -838,20 +837,19 @@ def master_callback(
             elif triggered_id == "compliance-action-btn":
                 action_name = "compliance"
                 result, generated_filename, generated_xlsx_bytes, _, _ = process_document(
-                    sess_data, action_name, doc_value, chat_input, uploaded_rfp_decoded_bytes, proposal_value
+                    sess_data, 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 == "board-action-btn":
                 action_name = "virtual_board"
                 result, generated_filename, generated_xlsx_bytes, _, _ = process_document(
-                    sess_data, action_name, doc_value, chat_input, uploaded_rfp_decoded_bytes, proposal_value
+                    sess_data, 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 == "proposal-action-btn":
                 action_name = "proposal"
-                selected_bytes = sess_data["uploaded_documents_bytes"].get(doc_value, None)
                 result, _, _, generated_filename, generated_docx_bytes = process_document(
-                    sess_data, action_name, doc_value, chat_input, selected_bytes, None
+                    sess_data, action_name, doc_value, chat_input, None, None
                 )
                 output_data_upload = dcc.Markdown(result, style={"whiteSpace": "pre-wrap", "wordWrap": "break-word"})
             elif triggered_id == "recover-action-btn":