Update app.py

app.py

@@ -42,15 +42,18 @@ question_examples = [
 
 # === Extract tool name and format output
 def extract_tool_name_and_clean_content(msg):
-    content_raw = msg.get("content") if isinstance(msg, dict) else getattr(msg, "content", "")
-    
+    tool_name = "Tool Result"
+    content = msg.get("content") if isinstance(msg, dict) else getattr(msg, "content", "")
+
+    # Attempt to load JSON from tool response
     try:
-        parsed = json.loads(content_raw)
-        tool_name = parsed.get("tool_name", "Tool Result")
-        content = parsed.get("content", "")
-    except Exception:
-        tool_name = "Tool Result"
-        content = content_raw
+        parsed = json.loads(content)
+        if isinstance(parsed, dict):
+            tool_name = parsed.get("tool_name", tool_name)
+            content = parsed.get("content", content)
+    except Exception as e:
+        # fallback if content is not JSON
+        pass
 
     if isinstance(content, (dict, list)):
         content = json.dumps(content, indent=2)