Update app.py

app.py

@@ -185,23 +185,28 @@ def chat_stream(message, history, model, temperature, max_tokens):
     
     history = history or []
     
-    # Build conversation context
+    # Build conversation context from history
     system = "You are Axon, an expert AI coding assistant. Be helpful, concise, and provide working code examples when appropriate."
     conversation = ""
-    for user_msg, assistant_msg in history[-5:]:  # Last 5 turns
-        conversation += f"<|im_start|>user\n{user_msg}<|im_end|>\n<|im_start|>assistant\n{assistant_msg}<|im_end|>\n"
+    for msg in history[-10:]:  # Last 10 messages
+        if msg["role"] == "user":
+            conversation += f"<|im_start|>user\n{msg['content']}<|im_end|>\n"
+        elif msg["role"] == "assistant":
+            conversation += f"<|im_start|>assistant\n{msg['content']}<|im_end|>\n"
     
     prompt = f"<|im_start|>system\n{system}<|im_end|>\n{conversation}<|im_start|>user\n{message}<|im_end|>\n<|im_start|>assistant\n"
     
-    response = ""
-    history.append([message, ""])
+    # Add user message to history
+    history = history + [{"role": "user", "content": message}]
     
+    response = ""
     for chunk in generate_stream(prompt, model, max_tokens, temperature):
         response += chunk
-        history[-1][1] = response
-        yield history
+        # Yield history with partial assistant response
+        yield history + [{"role": "assistant", "content": response}]
     
-    yield history
+    # Final yield with complete response
+    yield history + [{"role": "assistant", "content": response}]
 
 def generate_code(description, language, model, max_tokens):
     if not description.strip():
@@ -747,7 +752,7 @@ def create_ui():
         
         msg.submit(respond, [msg, chatbot, model_dropdown, temperature, max_tokens], [chatbot, msg])
         send_btn.click(respond, [msg, chatbot, model_dropdown, temperature, max_tokens], [chatbot, msg])
-        clear_btn.click(lambda: [], None, chatbot)
+        clear_btn.click(lambda: ([], ""), None, [chatbot, msg])
         transcribe_btn.click(transcribe_audio, audio_input, msg)
         
         # Core tools