Update app.py via AI Editor

app.py

@@ -132,13 +132,13 @@ def save_proposal_as_docx(proposal_text, base_filename):
     memf.seek(0)
     return memf.read()
 
-def process_document(action, selected_filename=None, chat_input=None, selected_proposal=None, selected_generated=None):
+def process_document(action, selected_filename=None, chat_input=None):
     global shredded_document, generated_response
     logging.info(f"Process document called with action: {action}")
 
     doc_content = None
     doc_fileid = None
-    if action in ["shred", "generate"]:
+    if action in ["shred", "generate", "proposal"]:
         if selected_filename and selected_filename in uploaded_documents:
             doc_content = uploaded_documents[selected_filename]
             doc_fileid = uploaded_documents_fileid.get(selected_filename)
@@ -149,8 +149,6 @@ def process_document(action, selected_filename=None, chat_input=None, selected_p
         else:
             doc_content = None
             doc_fileid = None
-    elif action == "proposal":
-        pass
 
     if action == 'shred':
         if not doc_content:
@@ -225,45 +223,21 @@ def process_document(action, selected_filename=None, chat_input=None, selected_p
         return result_holder["text"], result_holder["docx_bytes"], result_holder["docx_name"]
 
     elif action == 'proposal':
-        rfp_content = None
-        generated_doc_content = None
-        rfp_filename = selected_filename
-        generated_docname = selected_generated
-        rfp_fileid = uploaded_documents_fileid.get(selected_filename)
-        gen_fileid = None
-        if not (selected_filename and selected_filename in uploaded_documents):
+        if not doc_content:
             logging.warning("No RFP/SOW/PWS/RFI document selected for proposal action.")
             return "No RFP/SOW/PWS/RFI document selected.", None, None
-        if not (selected_generated and selected_generated in generated_documents):
-            logging.warning("No generated document selected for proposal action.")
-            return "No generated document selected.", None, None
-        rfp_content = uploaded_documents[selected_filename]
-        gen_bytes = generated_documents[selected_generated]
-        try:
-            try:
-                docx_stream = io.BytesIO(gen_bytes)
-                doc = Document(docx_stream)
-                generated_doc_content = "\n".join([para.text for para in doc.paragraphs])
-            except Exception as e:
-                try:
-                    generated_doc_content = gen_bytes.decode('utf-8')
-                except Exception:
-                    generated_doc_content = "<Unable to decode generated document.>"
-        except Exception as e:
-            generated_doc_content = "<Unable to read generated document.>"
-            logging.error(f"Failed to read generated document: {e}")
-
+        rfp_filename = selected_filename
+        rfp_fileid = uploaded_documents_fileid.get(selected_filename)
         prompt = (
-            "Respond to the following RFP/SOW/PWS/RFI provided and focus on the Generated Document provided by creating a highly detailed proposal response that follows each section and subsection header. "
+            "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.\n"
         )
         if chat_input:
             prompt += f"User additional instructions: {chat_input}\n"
-        prompt += f"\n---\nRFP/SOW/PWS/RFI ({rfp_filename}):\n"
-        prompt += f"\n---\nGenerated Document ({generated_docname}):\n{generated_doc_content}\n"
-        logging.info(f"Sending proposal prompt to Gemini. RFP: {rfp_filename}, Generated Doc: {generated_docname}")
+        prompt += f"\n---\nRFP/SOW/PWS/RFI ({rfp_filename}):\n{doc_content}\n"
+        logging.info(f"Sending proposal prompt to Gemini. RFP: {rfp_filename}")
         result_holder = {"text": None, "docx_bytes": None, "docx_name": None}
         def thread_proposal():
             try:
@@ -271,7 +245,7 @@ def process_document(action, selected_filename=None, chat_input=None, selected_p
                 response = gemini_generate_content(prompt, file_id=rfp_fileid, chat_input=chat_input)
                 logging.info("Received proposal results from Gemini.")
                 docx_bytes = save_proposal_as_docx(response, rfp_filename)
-                generated_docx_name = f"{os.path.splitext(rfp_filename)[0]}_{os.path.splitext(generated_docname)[0]}_proposal.docx"
+                generated_docx_name = f"{os.path.splitext(rfp_filename)[0]}_proposal.docx"
                 result_holder["text"] = response
                 result_holder["docx_bytes"] = docx_bytes
                 result_holder["docx_name"] = generated_docx_name
@@ -649,14 +623,11 @@ def master_callback(
 
         elif triggered_id == 'proposal-action-btn':
             rfp_doc = selected_filename
-            gen_doc = selected_generated_dropdown_state
-            logging.info(f"Starting proposal streaming with RFP: {rfp_doc}, Generated Doc: {gen_doc}")
+            logging.info(f"Starting proposal with RFP: {rfp_doc}")
             result, generated_docx_bytes, generated_docx_name = process_document(
                 'proposal',
                 rfp_doc,
-                chat_input,
-                None,
-                gen_doc
+                chat_input
             )
             if generated_docx_bytes and generated_docx_name:
                 generated_documents[generated_docx_name] = generated_docx_bytes