Update app.py

app.py

@@ -5,6 +5,7 @@ import gradio as gr
 from multiprocessing import freeze_support
 import importlib
 import inspect
+import json
 
 # === Fix path to include src/txagent
 sys.path.insert(0, os.path.join(os.path.dirname(__file__), "src"))
@@ -55,7 +56,6 @@ def create_ui(agent):
 
         # === Core handler (streaming generator)
         def handle_chat(message, history, temperature, max_new_tokens, max_tokens, multi_agent, conversation, max_round):
-            # Must yield a list of {"role": ..., "content": ...} dicts
             generator = agent.run_gradio_chat(
                 message=message,
                 history=history,
@@ -66,15 +66,30 @@ def create_ui(agent):
                 conversation=conversation,
                 max_round=max_round
             )
+
             for update in generator:
-                # Convert to list of dicts if not already
-                formatted = [
-                    {"role": m["role"], "content": m["content"]}
-                    if isinstance(m, dict)
-                    else {"role": m.role, "content": m.content}
-                    for m in update
-                ]
-                yield formatted
+                formatted_messages = []
+
+                for m in update:
+                    role = m.get("role")
+                    content = m.get("content", "")
+
+                    # 🧠 Pretty-print tool calls
+                    if "[TOOL_CALLS]" in content:
+                        try:
+                            thought, tools = content.split("[TOOL_CALLS]", 1)
+                            tools_json = json.loads(tools.strip())
+                            pretty_tools = json.dumps(tools_json, indent=2)
+
+                            # Style it like a ChatGPT output
+                            content = f"**🧠 Assistant reasoning:**\n\n{thought.strip()}\n\n" \
+                                      f"**🔧 Tool Calls:**\n\n```json\n{pretty_tools}\n```"
+                        except Exception as e:
+                            content = f"{content}\n\n⚠️ Failed to format tool calls: {e}"
+
+                    formatted_messages.append({"role": role, "content": content})
+
+                yield formatted_messages
 
         # === Trigger handlers
         send_button.click(