Update app.py

app.py

@@ -92,27 +92,28 @@ def run_agent(_):
         additional_args={"source_file": temp_file.name}
     )
 
-if isinstance(result, str):
-    try:
-        result = json.loads(result)
-    except json.JSONDecodeError:
-        insights = "Failed to parse result from agent."
-        return insights, []
-
-if isinstance(result, dict):
-    insights = result.get("insights", "No insights generated.")
-    image_paths = result.get("figures", [])
-else:
-    insights = "Error: Unexpected result format from agent."
-    image_paths = []
-
-images = []
-for path in image_paths:
-    try:
-        images.append(Image.open(path))
-    except Exception as e:
-        print(f"Error loading image {path}: {e}")
-return insights, images
+    if isinstance(result, str):
+        try:
+            result = json.loads(result)
+        except json.JSONDecodeError:
+            insights = "Failed to parse result from agent."
+            return insights, []
+
+    if isinstance(result, dict):
+        insights = result.get("insights", "No insights generated.")
+        image_paths = result.get("figures", [])
+    else:
+        insights = "Error: Unexpected result format from agent."
+        image_paths = []
+
+    images = []
+    for path in image_paths:
+        try:
+            images.append(Image.open(path))
+        except Exception as e:
+            print(f"Error loading image {path}: {e}")
+    return insights, images
+
 
     
 def train_model(_):