Update app.py via AI Editor

app.py

@@ -181,6 +181,7 @@ def process_document(action, selected_filename=None, chat_input=None, rfp_decode
         if chat_input:
             prompt += f"User 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)
         if result and not result.startswith("Error"):
             docx_bytes = save_shredded_as_docx(result, selected_filename)
@@ -270,41 +271,8 @@ def process_document(action, selected_filename=None, chat_input=None, rfp_decode
 
         findings_content = uploaded_documents[selected_filename]
         proposal_text = proposals[selected_proposal_filename]
-        findings_fileid = uploaded_documents_fileid.get(selected_filename)
-        proposal_fileid = proposals_fileid.get(selected_proposal_filename)
-
-        # Ensure both findings and proposal are uploaded as files in Gemini
-        if not findings_fileid:
-            findings_bytes = None
-            if selected_filename in shredded_documents:
-                findings_bytes = shredded_documents[selected_filename]
-            elif selected_filename in uploaded_documents_bytes:
-                findings_bytes = uploaded_documents_bytes[selected_filename]
-            else:
-                findings_bytes = save_compliance_as_docx(findings_content, selected_filename)
-            if findings_bytes:
-                try:
-                    findings_fileid = upload_to_gemini_file(findings_bytes, selected_filename)
-                    if findings_fileid:
-                        uploaded_documents_fileid[selected_filename] = findings_fileid
-                        logging.info(f"Findings/Compliance doc {selected_filename} uploaded to Gemini for recover.")
-                except Exception as e:
-                    logging.error(f"Failed to upload findings doc {selected_filename} for recover: {e}")
-
-        if not proposal_fileid:
-            proposal_bytes = save_proposal_as_docx(proposals[selected_proposal_filename], selected_proposal_filename)
-            if proposal_bytes:
-                try:
-                    proposal_fileid = upload_to_gemini_file(proposal_bytes, selected_proposal_filename)
-                    if proposal_fileid:
-                        proposals_fileid[selected_proposal_filename] = proposal_fileid
-                        logging.info(f"Proposal doc {selected_proposal_filename} uploaded to Gemini for recover.")
-                except Exception as e:
-                    logging.error(f"Failed to upload proposal doc {selected_proposal_filename} for recover: {e}")
-
-        logging.info(f"Recovery: fixing proposal [{selected_proposal_filename}] based on findings [{selected_filename}]")
         prompt = (
-            "You are a proposal compliance recovery expert. Use the findings and recommendations table and the original proposal response provided in the attached files. "
+            "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. "
@@ -314,14 +282,10 @@ def process_document(action, selected_filename=None, chat_input=None, rfp_decode
         if chat_input:
             prompt += f"User additional instructions: {chat_input}\n"
 
-        # Compose file_ids and pass both files to Gemini in order
-        file_ids = []
-        if findings_fileid:
-            file_ids.append(findings_fileid)
-        if proposal_fileid:
-            file_ids.append(proposal_fileid)
+        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_ids=file_ids)
+        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]
             recovered_docx_name = f"{base_name}_recovered.docx"