Update app.py

app.py

@@ -113,7 +113,7 @@ def respond(msg, chat_history, temperature, max_new_tokens, max_tokens, multi_ag
 
     message = msg.strip()
     chat_history.append({"role": "user", "content": message})
-    formatted_history = [(m["role"], m["content"]) for m in chat_history]
+    formatted_history = [(m["role"], m["content"]) for m in chat_history if "role" in m and "content" in m]
 
     try:
         response_generator = agent.run_gradio_chat(
@@ -129,17 +129,21 @@ def respond(msg, chat_history, temperature, max_new_tokens, max_tokens, multi_ag
             call_agent_level=None,
             sub_agent_task=None
         )
+
         collected = ""
         for chunk in response_generator:
-            if chunk is None:
-                continue
-            elif isinstance(chunk, dict):
-                collected += chunk.get("content", "")
-            else:
+            if isinstance(chunk, dict) and "content" in chunk:
+                collected += chunk["content"]
+            elif isinstance(chunk, str):
+                collected += chunk
+            elif chunk is not None:
                 collected += str(chunk)
-        chat_history.append({"role": "assistant", "content": collected})
+
+        chat_history.append({"role": "assistant", "content": collected or "⚠️ No content returned."})
+
     except Exception as e:
-        chat_history.append({"role": "assistant", "content": f"Error: {e}"})
+        chat_history.append({"role": "assistant", "content": f"❌ Error: {str(e)}"})
+
     return chat_history
 
 def create_demo(agent):