Update app.py

app.py

@@ -129,18 +129,18 @@ def analyze_data(csv_file, additional_notes=""):
     
     agent = CodeAgent(tools=[], model=model, additional_authorized_imports=["numpy", "pandas", "matplotlib.pyplot", "seaborn", "sklearn", "json"])
     analysis_result = agent.run("""
-        You are a helpful data analysis agent. Just return insight information and visualization.
-        Load the data that is passed.do not create your own.
-        Automatically detect numeric columns and names.
-        2. 5 data visualizations
-        3. at least 5 insights from data
-        5. Generate publication-quality visualizations and save to './figures/'.
-        Do not use 'open()' or write to files. Just return variables and plots.
+        You are a data analysis agent. Just return insight information and visualization and follow following instruction strictly. 
+        1. Load the data from provide source_file.Do not create your own.
+        2. Detect numeric and categorical columns.
+        3. Generate at least 5 visualizations and 5 insights.
+        4. Generate publication-quality visualizations and Save all plots to `./figures/` as PNGs using matplotlib or seaborn.
+        5. Do not use 'open()' or write to files. Just return variables and plots.
+        6. Return insights in this exact Python dictionary structure:
         The dictionary should have the following structure:
         {
             'observations': {
-                'observation_1_key': 'observation_1_value',
-                'observation_2_key': 'observation_2_value',
+                'observation_1_key': 'Brief, clear observation.',
+                'observation_2_key': 'Another brief point.',
                 ...
             },
             'insights': {
@@ -149,6 +149,9 @@ def analyze_data(csv_file, additional_notes=""):
                 ...
             }
         }
+    DO NOT use open(), DO NOT print. Only return this dictionary object as your final output in a code block.
+
+    Be concise. Do not include text explanations outside the code block.
     """, additional_args={"additional_notes": additional_notes, "source_file": csv_file})
     
     execution_time = time.time() - start_time