Update app.py

app.py

@@ -283,7 +283,21 @@ def advanced_visualization(req: VisualizationRequest):
     return {
         "success": True,
         "analysis": analysis,
+        # Backward compatible keys for current client (`agents/mcp_agent.py`)
+        # - `visualization`: one main image (base64)
+        # - `visualizations_multi`: optional additional images (list[base64])
+        "visualization": (
+            images.get("main")
+            or images.get("main_bar")
+            or next(iter(images.values()))
+        ),
+        "visualizations_multi": [
+            b64 for name, b64 in images.items()
+            if name not in {"main", "main_bar"}
+        ],
+        # Keep full mapping for debugging/inspection
         "visualizations": images,
+        "script": script,
         "logs": {
             "stdout": result.get("stdout"),
             "stderr": result.get("stderr"),