Update app.py

app.py

@@ -61,6 +61,8 @@ agent = CodeAgent(
 visual_output = gr.Gallery(label="Generated Visualizations", columns=3, height=600, object_fit="contain")
 
 def run_agent(_):
+    import os
+    from PIL import Image
 
     if df_global is None:
         return "Please upload a file first.", []
@@ -106,7 +108,19 @@ def run_agent(_):
         insights = "\n".join(raw_insights) if isinstance(raw_insights, list) else str(raw_insights)
 
         image_paths = result.get("figures", [])
-        print("🔍 Image paths
+        print("🔍 Image paths received:", image_paths)
+
+        for path in image_paths:
+            if os.path.exists(path):
+                try:
+                    images.append(Image.open(path))
+                except Exception as e:
+                    print(f"⚠️ Error loading {path}: {e}")
+            else:
+                print(f"❌ File not found: {path}")
+
+    return insights, images
+