Update app.py via AI Editor

app.py

@@ -46,6 +46,7 @@ def decode_document(decoded_bytes):
 def anthropic_stream_generate(prompt):
     stream_result = []
     try:
+        logging.info("Connecting to Anthropic Claude for streaming completion...")
         with anthropic_client.messages.create(
             model=CLAUDE3_SONNET_MODEL,
             max_tokens=CLAUDE3_MAX_OUTPUT_TOKENS,
@@ -57,6 +58,7 @@ def anthropic_stream_generate(prompt):
                     piece = event.delta.text
                     stream_result.append(piece)
                     logging.debug(f"Streaming piece: {piece}")
+        logging.info("Received response from Anthropic Claude streaming.")
         return ''.join(stream_result)
     except Exception as e:
         logging.error("Error during anthropic streaming request: %s", e)
@@ -179,8 +181,10 @@ def process_document(action, selected_filename=None, chat_input=None, selected_p
         rfp_filename = selected_filename
         generated_docname = selected_generated
         if not (selected_filename and selected_filename in uploaded_documents):
+            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]
@@ -208,11 +212,13 @@ def process_document(action, selected_filename=None, chat_input=None, selected_p
             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.")
+        logging.info(f"Sending proposal prompt to anthropic. RFP: {rfp_filename}, Generated Doc: {generated_docname}")
         result_holder = {"text": None, "docx_bytes": None, "docx_name": None}
         def thread_proposal():
             try:
+                logging.info("Connecting to Anthropic Claude for proposal streaming...")
                 result = anthropic_stream_generate(prompt)
+                logging.info("Received proposal results from Anthropic Claude.")
                 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
@@ -514,22 +520,7 @@ def master_callback(
         'recover-action-btn', 'board-action-btn', 'loe-action-btn'
     ]
 
-    if triggered_id == 'proposal-action-btn':
-        output_data_upload = html.Div("Generating proposal from selected documents...", style={"wordWrap": "break-word"})
-        # This is a progress placeholder, real work will happen in the next callback (Dash pattern)
-        return (
-            uploaded_doc_list,
-            doc_options,
-            doc_value,
-            proposal_options,
-            proposal_value,
-            uploaded_proposal_list,
-            generated_doc_list,
-            generated_doc_options,
-            generated_doc_value,
-            output_data_upload
-        )
-
+    # Fix: When "Proposal" is clicked, trigger proposal streaming with correct dropdown doc selections
     if triggered_id in action_buttons:
         result = ""
         generated_docx_bytes = None
@@ -544,12 +535,16 @@ def master_callback(
                 new_selected_generated = generated_docx_name
 
         elif triggered_id == 'proposal-action-btn':
+            # Use the currently selected doc and generated doc from dropdowns
+            rfp_doc = selected_filename
+            gen_doc = selected_generated_dropdown_state
+            logging.info(f"Starting proposal streaming with RFP: {rfp_doc}, Generated Doc: {gen_doc}")
             result, generated_docx_bytes, generated_docx_name = process_document(
                 'proposal',
-                selected_filename,
+                rfp_doc,
                 chat_input,
                 None,
-                selected_generated_dropdown_state
+                gen_doc
             )
             if generated_docx_bytes and generated_docx_name:
                 generated_documents[generated_docx_name] = generated_docx_bytes