Fixes heading

main.py

@@ -89,6 +89,7 @@ def chat(prompt, history):
     buffer = ""
     reasoning = ""
     thinking = False
+    reasoning_heading = "# Reasoning\n\n"
 
     for new_text in streamer:
         if not thinking and "<think>" in new_text:
@@ -100,12 +101,14 @@ def chat(prompt, history):
 
         if thinking:
             reasoning += new_text
-            heading = "# Reasoning\n\n"
-            yield "I'm thinking, please wait a moment...", heading + reasoning
+            yield (
+                "I'm thinking, please wait a moment...",
+                reasoning_heading + reasoning,
+            )
             continue
 
         buffer += new_text
-        yield reformat_math(buffer), reasoning
+        yield reformat_math(buffer), reasoning_heading + reasoning
 
 
 chat_bot = gr.Chatbot(