Update app.py via AI Editor

app.py

@@ -6,7 +6,6 @@ from dash import dcc, html, Input, Output, State, callback_context
 import dash_bootstrap_components as dbc
 import pandas as pd
 import anthropic
-from threading import Thread
 import logging
 from docx import Document
 
@@ -63,6 +62,19 @@ def anthropic_stream_generate(prompt):
         logging.error("Error during anthropic streaming request: %s", e)
         return f"Error during streaming: {e}"
 
+def anthropic_sync_generate(prompt):
+    try:
+        response = anthropic_client.messages.create(
+            model=CLAUDE3_SONNET_MODEL,
+            max_tokens=CLAUDE3_MAX_OUTPUT_TOKENS,
+            messages=[{"role": "user", "content": prompt}]
+        )
+        logging.info("Synchronous anthropic generation complete.")
+        return response.content[0].text if hasattr(response, "content") else ""
+    except Exception as e:
+        logging.error("Error during anthropic synchronous request: %s", e)
+        return f"Error during synchronous call: {e}"
+
 def save_shredded_as_docx(shredded_text, rfp_filename):
     doc = Document()
     doc.add_heading(f"Shredded Requirements for {rfp_filename}", 0)
@@ -73,7 +85,7 @@ def save_shredded_as_docx(shredded_text, rfp_filename):
     memf.seek(0)
     return memf.read()
 
-def process_document(action, selected_filename=None, chat_input=None, selected_proposal=None):
+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}")
 
@@ -127,6 +139,7 @@ def process_document(action, selected_filename=None, chat_input=None, selected_p
                 logging.error("Error in thread_shred: %s", e)
                 result_holder["text"] = shredded_document
         shredded_document = "Shredding in progress..."
+        from threading import Thread
         t = Thread(target=thread_shred)
         t.start()
         t.join()
@@ -159,15 +172,52 @@ def process_document(action, selected_filename=None, chat_input=None, selected_p
                 logging.error("Error in thread_generate: %s", e)
                 result_holder["text"] = generated_response
         generated_response = "Generating response..."
+        from threading import Thread
         t = Thread(target=thread_generate)
         t.start()
         t.join()
         return result_holder["text"], None, None
 
     elif action == 'proposal':
-        if not doc_content:
-            return "No proposal document uploaded.", None, None
-        return dcc.Markdown(doc_content, style={"whiteSpace": "pre-wrap", "wordWrap": "break-word"}), None, None
+        rfp_content = None
+        generated_doc_content = None
+        rfp_filename = selected_filename
+        generated_docname = selected_generated
+        if not (selected_filename and selected_filename in uploaded_documents):
+            return "No RFP/SOW/PWS/RFI document selected.", None, None
+        if not (selected_generated and selected_generated in generated_documents):
+            return "No generated document selected.", None, None
+        rfp_content = uploaded_documents[selected_filename]
+        gen_bytes = generated_documents[selected_generated]
+        try:
+            # Try to read as docx, fallback to decode
+            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}")
+
+        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 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"
+        )
+        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 synchronous call.")
+        result = anthropic_sync_generate(prompt)
+        return result, None, None
+
     elif action == 'compliance':
         return "Compliance checking not implemented yet.", None, None
     elif action == 'recover':
@@ -374,10 +424,8 @@ def master_callback(
     ctx = callback_context
     triggered_id = ctx.triggered[0]['prop_id'].split('.')[0] if ctx.triggered else None
 
-    # Upload or delete document or proposal or generated document
     upload_triggered = False
 
-    # Handle upload document
     if triggered_id == 'upload-document' and rfp_content is not None and rfp_filename:
         content_type, content_string = rfp_content.split(',')
         decoded = base64.b64decode(content_string)
@@ -389,7 +437,6 @@ def master_callback(
             logging.error(f"Failed to decode uploaded document: {rfp_filename}")
         upload_triggered = True
 
-    # Handle upload proposal
     if triggered_id == 'upload-proposal' and proposal_content is not None and proposal_filename:
         content_type, content_string = proposal_content.split(',')
         decoded = base64.b64decode(content_string)
@@ -401,7 +448,6 @@ def master_callback(
             logging.error(f"Failed to decode uploaded proposal: {proposal_filename}")
         upload_triggered = True
 
-    # Handle delete document
     if triggered_id and isinstance(ctx.inputs_list[2], list):
         for i, n_click in enumerate(rfp_delete_clicks):
             if n_click:
@@ -415,7 +461,6 @@ def master_callback(
                     upload_triggered = True
                     break
 
-    # Handle delete proposal
     if triggered_id and isinstance(ctx.inputs_list[6], list):
         for i, n_click in enumerate(proposal_delete_clicks):
             if n_click:
@@ -429,7 +474,6 @@ def master_callback(
                     upload_triggered = True
                     break
 
-    # Handle delete generated document
     if triggered_id and isinstance(ctx.inputs_list[10], list):
         for i, n_click in enumerate(generated_delete_clicks):
             if n_click:
@@ -454,7 +498,6 @@ def master_callback(
     uploaded_proposal_list = get_uploaded_proposal_list(uploaded_proposals)
     generated_doc_list = get_generated_doc_list(generated_documents)
 
-    # Default output
     output_data_upload = html.Div("No action taken yet.", style={"wordWrap": "break-word"})
 
     action_buttons = [
@@ -462,7 +505,6 @@ def master_callback(
         'recover-action-btn', 'board-action-btn', 'loe-action-btn'
     ]
 
-    # Handle action buttons
     if triggered_id in action_buttons:
         result = ""
         generated_docx_bytes = None
@@ -476,7 +518,7 @@ def master_callback(
                 logging.info(f"Generated docx saved: {generated_docx_name}")
                 new_selected_generated = generated_docx_name
         elif triggered_id == 'generate-action-btn':
-            result, _, _ = process_document('generate', selected_filename, chat_input)
+            result, _, _ = process_document('proposal', selected_filename, chat_input, selected_proposal_dropdown, selected_generated_dropdown_state)
         elif triggered_id == 'compliance-action-btn':
             result, _, _ = process_document('compliance', selected_filename, chat_input)
         elif triggered_id == 'recover-action-btn':
@@ -499,7 +541,6 @@ def master_callback(
         generated_doc_value = new_selected_generated if new_selected_generated in generated_documents else (next(iter(generated_documents), None) if generated_documents else None)
         generated_doc_list = get_generated_doc_list(generated_documents)
 
-    # Handle select-generated-dropdown for download
     elif triggered_id == 'select-generated-dropdown':
         sel_gen = selected_generated_dropdown
         if not sel_gen or sel_gen not in generated_documents: