Updated Prompt

app.py

@@ -66,8 +66,8 @@ def get_visualization(question, tool):
     fig = agent.run(
         
         task=f'''
-    THINK STEP BY STEP
-    Here are the steps you should follow while writing code for Visualization:
+    
+    You are an expert in writing Visualization code. Here are the steps you should follow while writing code for Visualization:
     1. Select the most appropriate chart type for data. Use bar charts for categorical comparisons, line charts for trends over time, scatter plots for relationships between variables, pie charts for proportions, histograms for distribution analysis, and box plots for visualizing data spread and outliers.
     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).
@@ -77,8 +77,9 @@ def get_visualization(question, tool):
     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.
-    10. Use seaborn
+    10. Do not write functions.
     11. In the end you have to return a dict which contain final fig as fig key, Generated SQL as sql key, Data as a dataframe with data key using the `final_answer` tool.
+    DO NOT PRINT THE FINAL ANSWER OR FINAL OUTPUT
     Here is the task:
     task: {question}
     ''',