Fix syntax

pipeline.py

@@ -89,6 +89,12 @@ def stream_chat(
     final_use_rag = use_rag and has_rag_index and not disable_agentic_reasoning
     final_use_web_search = use_web_search and not disable_agentic_reasoning
     
+    # Initialize updated_history early to avoid UnboundLocalError
+    updated_history = history + [
+        {"role": "user", "content": original_message},
+        {"role": "assistant", "content": ""}
+    ]
+    
     plan = None
     if not disable_agentic_reasoning:
         reasoning_stage_start = time.time()
@@ -119,9 +125,8 @@ def stream_chat(
         pipeline_diagnostics["strategy_decisions"].append("Agentic reasoning disabled by user")
     
     # Update thoughts after reasoning stage
-    if show_thoughts and thought_handler:
-        thoughts_text = thought_handler.get_thoughts()
-        yield updated_history, thoughts_text
+    thoughts_text = thought_handler.get_thoughts() if (show_thoughts and thought_handler) else ""
+    yield updated_history, thoughts_text
     
     if disable_agentic_reasoning:
         logger.info("[MAC] Agentic reasoning disabled - using MedSwin alone")
@@ -138,9 +143,8 @@ def stream_chat(
         logger.info(f"[GEMINI SUPERVISOR] Created {len(breakdown.get('sub_topics', []))} sub-topics")
     
     # Update thoughts after breakdown
-    if show_thoughts and thought_handler:
-        thoughts_text = thought_handler.get_thoughts()
-        yield updated_history, thoughts_text
+    thoughts_text = thought_handler.get_thoughts() if (show_thoughts and thought_handler) else ""
+    yield updated_history, thoughts_text
     
     search_contexts = []
     web_urls = []
@@ -259,11 +263,8 @@ def stream_chat(
     logger.info(f"[MEDSWIN] Executing {len(breakdown.get('sub_topics', []))} tasks sequentially...")
     medswin_answers = []
     
-    updated_history = history + [
-        {"role": "user", "content": original_message},
-        {"role": "assistant", "content": ""}
-    ]
-    thoughts_text = thought_handler.get_thoughts() if thought_handler else ""
+    # Update thoughts before starting MedSwin tasks
+    thoughts_text = thought_handler.get_thoughts() if (show_thoughts and thought_handler) else ""
     yield updated_history, thoughts_text
     
     medswin_stage_start = time.time()
@@ -302,7 +303,7 @@ def stream_chat(
             
             partial_final = "\n\n".join(medswin_answers)
             updated_history[-1]["content"] = partial_final
-            thoughts_text = thought_handler.get_thoughts() if thought_handler else ""
+            thoughts_text = thought_handler.get_thoughts() if (show_thoughts and thought_handler) else ""
             yield updated_history, thoughts_text
     
         except Exception as e:
@@ -409,9 +410,8 @@ def stream_chat(
             record_stage("followup_search", followup_stage_start)
             
             # Update thoughts after followup search
-            if show_thoughts and thought_handler:
-                thoughts_text = thought_handler.get_thoughts()
-                yield updated_history, thoughts_text
+            thoughts_text = thought_handler.get_thoughts() if (show_thoughts and thought_handler) else ""
+            yield updated_history, thoughts_text
     
     citations_text = ""
     
@@ -434,7 +434,7 @@ def stream_chat(
     final_answer_with_metadata = final_answer + citations_text + speaker_icon
     
     updated_history[-1]["content"] = final_answer_with_metadata
-    thoughts_text = thought_handler.get_thoughts() if thought_handler else ""
+    thoughts_text = thought_handler.get_thoughts() if (show_thoughts and thought_handler) else ""
     # Always yield thoughts_text, even if empty, to ensure UI updates
     yield updated_history, thoughts_text