Update app.py via AI Editor

app.py

@@ -72,6 +72,16 @@ def save_shredded_as_docx(shredded_text, rfp_filename):
     memf.seek(0)
     return memf.read()
 
+def save_proposal_as_docx(proposal_text, base_filename):
+    doc = Document()
+    doc.add_heading(f"Proposal Response for {base_filename}", 0)
+    for line in proposal_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, selected_proposal=None, selected_generated=None):
     global shredded_document, generated_response
     logging.info(f"Process document called with action: {action}")
@@ -144,7 +154,7 @@ def process_document(action, selected_filename=None, chat_input=None, selected_p
         if chat_input:
             prompt += f"User additional instructions: {chat_input}\n"
         prompt += f"\nFile Name: {selected_filename}\n\n{shredded_document}"
-        result_holder = {"text": None}
+        result_holder = {"text": None, "docx_bytes": None, "docx_name": None}
         def thread_generate():
             global generated_response
             generated_response = ""
@@ -153,7 +163,11 @@ def process_document(action, selected_filename=None, chat_input=None, selected_p
                 result = anthropic_stream_generate(prompt)
                 generated_response = result
                 logging.info("Proposal response generated successfully.")
+                docx_bytes = save_proposal_as_docx(result, selected_filename)
+                generated_docx_name = f"{os.path.splitext(selected_filename)[0]}_proposal.docx"
                 result_holder["text"] = result
+                result_holder["docx_bytes"] = docx_bytes
+                result_holder["docx_name"] = generated_docx_name
             except Exception as e:
                 generated_response = f"Error during generation: {e}"
                 logging.error("Error in thread_generate: %s", e)
@@ -163,7 +177,7 @@ def process_document(action, selected_filename=None, chat_input=None, selected_p
         t = Thread(target=thread_generate)
         t.start()
         t.join()
-        return result_holder["text"], None, None
+        return result_holder["text"], result_holder["docx_bytes"], result_holder["docx_name"]
 
     elif action == 'proposal':
         rfp_content = None
@@ -194,18 +208,22 @@ def process_document(action, selected_filename=None, chat_input=None, selected_p
             "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 and numbering. "
             "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, significantly limiting the use of bullets.\n"
+            "Be sure to respond in paragraph format only, no numbering, no 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{rfp_content}\n"
         prompt += f"\n---\nGenerated Document ({generated_docname}):\n{generated_doc_content}\n"
         logging.info("Sending proposal prompt to anthropic. This is a streaming call.")
-        result_holder = {"text": None}
+        result_holder = {"text": None, "docx_bytes": None, "docx_name": None}
         def thread_proposal():
             try:
                 result = anthropic_stream_generate(prompt)
+                docx_bytes = save_proposal_as_docx(result, rfp_filename)
+                generated_docx_name = f"{os.path.splitext(rfp_filename)[0]}_{os.path.splitext(generated_docname)[0]}_proposal.docx"
                 result_holder["text"] = result
+                result_holder["docx_bytes"] = docx_bytes
+                result_holder["docx_name"] = generated_docx_name
             except Exception as e:
                 logging.error("Error during streaming proposal request: %s", e)
                 result_holder["text"] = f"Error during streaming: {e}"
@@ -213,7 +231,7 @@ def process_document(action, selected_filename=None, chat_input=None, selected_p
         t = Thread(target=thread_proposal)
         t.start()
         t.join()
-        return result_holder["text"], None, None
+        return result_holder["text"], result_holder["docx_bytes"], result_holder["docx_name"]
 
     elif action == 'compliance':
         return "Compliance checking not implemented yet.", None, None
@@ -514,8 +532,14 @@ def master_callback(
                 generated_documents[generated_docx_name] = generated_docx_bytes
                 logging.info(f"Generated docx saved: {generated_docx_name}")
                 new_selected_generated = generated_docx_name
+
         elif triggered_id == 'generate-action-btn':
-            result, _, _ = process_document('proposal', selected_filename, chat_input, selected_proposal_dropdown, selected_generated_dropdown_state)
+            result, generated_docx_bytes, generated_docx_name = process_document('generate', selected_filename, chat_input)
+            if generated_docx_bytes and generated_docx_name:
+                generated_documents[generated_docx_name] = generated_docx_bytes
+                logging.info(f"Generated proposal docx saved: {generated_docx_name}")
+                new_selected_generated = generated_docx_name
+
         elif triggered_id == 'compliance-action-btn':
             result, _, _ = process_document('compliance', selected_filename, chat_input)
         elif triggered_id == 'recover-action-btn':