Update app.py

app.py

@@ -115,13 +115,14 @@ def analyze_data(csv_file, additional_notes=""):
         "source_file": csv_file.name if csv_file else None
     })
     
-    agent = CodeAgent(tools=[], model=model, additional_authorized_imports=["numpy", "pandas", "matplotlib.pyplot", "seaborn", "sklearn"])
+    agent = CodeAgent(tools=[], model=model, additional_authorized_imports=["numpy", "pandas", "matplotlib.pyplot", "seaborn", "sklearn", "json"])
     analysis_result = agent.run("""
         You are an expert data analyst. Perform comprehensive analysis including:
         1. Basic statistics and data quality checks
         2. 3 insightful analytical questions about relationships in the data
         3. Visualization of key patterns and correlations
         4. Actionable real-world insights derived from findings.
+        Don't write to any files. Just return all results in the interface.
         Generate publication-quality visualizations and save to './figures/'.
         Return the analysis results as a python dictionary that can be parsed by ast.literal_eval().
         The dictionary should have the following structure: