Update app.py

app.py

@@ -180,7 +180,7 @@ def transcribe_audio(audio_path):
 # --- CORE TOOLS ---
 def chat_stream(message, history, model, temperature, max_tokens):
     if not message.strip():
-        yield history
+        yield history or []
         return
     
     history = history or []
@@ -188,25 +188,36 @@ def chat_stream(message, history, model, temperature, max_tokens):
     # 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 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"
+    
+    # Handle history - could be list of dicts with role/content
+    for msg in history[-10:]:
+        try:
+            if isinstance(msg, dict):
+                role = msg.get("role", "")
+                content = msg.get("content", "")
+                if role == "user":
+                    conversation += f"<|im_start|>user\n{content}<|im_end|>\n"
+                elif role == "assistant":
+                    conversation += f"<|im_start|>assistant\n{content}<|im_end|>\n"
+        except Exception as e:
+            print(f"Error parsing history: {e}")
+            continue
     
     prompt = f"<|im_start|>system\n{system}<|im_end|>\n{conversation}<|im_start|>user\n{message}<|im_end|>\n<|im_start|>assistant\n"
     
-    # Add user message to history
-    history = history + [{"role": "user", "content": message}]
+    # Build new history with user message
+    new_history = list(history) + [{"role": "user", "content": message}]
     
     response = ""
     for chunk in generate_stream(prompt, model, max_tokens, temperature):
         response += chunk
         # Yield history with partial assistant response
-        yield history + [{"role": "assistant", "content": response}]
+        yield new_history + [{"role": "assistant", "content": response}]
     
     # Final yield with complete response
-    yield history + [{"role": "assistant", "content": response}]
+    if not response:
+        response = "❌ No response generated. Please try again."
+    yield new_history + [{"role": "assistant", "content": response}]
 
 def generate_code(description, language, model, max_tokens):
     if not description.strip():