Update app.py

app.py

@@ -52,7 +52,7 @@ additional_notes = "Please note: Perform a comprehensive analysis including visu
 agent = CodeAgent(
     tools=[],
     model=model,
-    additional_authorized_imports=["numpy", "pandas", "matplotlib.pyplot", "seaborn", "os"]
+    additional_authorized_imports=["numpy", "pandas", "matplotlib.pyplot", "seaborn", "os", "json"]
 )
 
 
@@ -64,6 +64,9 @@ def run_agent(_):
     import os
     from PIL import Image
 
+    os.makedirs("figures", exist_ok=True)  # Add this just before loading images
+
+
     if df_global is None:
         return "Please upload a file first.", []