Update Gradio_UI.py

Gradio_UI.py

@@ -31,10 +31,12 @@ def pull_messages_from_step(
     """Extract ChatMessage objects from agent steps with proper nesting"""
     import gradio as gr
 
-    yield gr.ChatMessage(role="assistant", content=f"***Thinking***")
+    
     if isinstance(step_log, ActionStep):
         # Output the step number
         step_number = f"Step {step_log.step_number}" if step_log.step_number is not None else ""
+        if step_log.step_number == 1:
+            yield gr.ChatMessage(role="assistant", content=f"***Thinking***")
         yield gr.ChatMessage(role="assistant", content=f"**{step_number}**")
 
         # First yield the thought/reasoning from the LLM