Edit Prompt

app.py

@@ -49,21 +49,33 @@ def get_table_schema(table):
     ddl_create = ddl_create.replace(old_path, full_path)
     return ddl_create, full_path
 
+
 def get_visualization(question, tool):
     agent = ReactCodeAgent(tools=[tool], llm_engine=llm_engine, add_base_tools=True,
                            additional_authorized_imports=['matplotlib.pyplot',
                                                  'pandas', 'plotly.express',
-                                                 'seaborn'], max_iterations=20)
+                                                 'seaborn'], max_iterations=10)
     fig = agent.run(
+        
         task=f'''
-    Use seaborn. Always 
-    Question: {question}
-    Always use the right colors.
-    If the question is about showing n number of rows return empty figure.
-    In the end you have to return a final fig using the `final_answer` tool
+    Here are the steps you should follow while writing code for Visualization:
+    1. Select the most effective visualization type for the data and purpose.
+    2. Ensure clear and appropriate labels, colors, and design elements, keeping visual elements legible and uncluttered.
+    3. Follow best practices, avoiding unnecessary visual distractions (chartjunk).
+    4. Ensure the code is error-free, with correct fields, transformations, and aesthetics.
+    5. Use descriptive and accurate x and y axis labels that reflect the data.
+    6. Ensure units of measurement are clearly indicated on axes (e.g., %, $, cm).
+    7. Ensure that categorical data is plotted on one axis and numerical data on the other, with appropriate labels that clearly represent the data being visualized.
+    8. When plotting categorical data, arrange categories in a meaningful order (e.g., by size, time, or frequency) rather than randomly.
+    9. Ensure that the categorical data are plotted on the x-axis, and the frequencies (numerical data) are plotted on the y-axis.
+    9. Use seaborn
+    10. In the end you have to return a final fig using the `final_answer` tool.
+
+    Here is the task:
+    task: {question}
     ''',
     )
-    
+
     return fig