Update app.py

app.py

@@ -92,27 +92,27 @@ def run_agent(_):
         additional_args={"source_file": temp_file.name}
     )
 
-    if isinstance(result, str):
+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, 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(_):