improving the code generation template in case of plotly

Author: matteocacciola

pandasai/core/prompts/templates/generate_python_code_with_sql.tmpl

@@ -11,7 +11,8 @@ def execute_sql_query(sql_query: str) -> pd.Dataframe
 </function>
 
 For the charts, you can either use `matplotlib.pyplot` or `plotly.express` to generate the charts.
-If you use `plotly.express`, you have to save each chart by using `fig.write_json` (e.g., `fig.write_json("path/to/file.json", pretty=True)`).
+Use `plotly.express` only if the user explicitly asks for it, otherwise use `matplotlib.pyplot`.
+If `plotly.express` is used, save each chart by using `fig.write_json` (e.g., `fig.write_json("path/to/file.json", pretty=True)`).
 
 {% if last_code_generated != "" and context.memory.count() > 0 %}
 {{ last_code_generated }}

tests/unit_tests/prompts/test_sql_prompt.py

@@ -82,7 +82,8 @@ def execute_sql_query(sql_query: str) -> pd.Dataframe
 </function>
 
 For the charts, you can either use `matplotlib.pyplot` or `plotly.express` to generate the charts.
-If you use `plotly.express`, you have to save each chart by using `fig.write_json` (e.g., `fig.write_json("path/to/file.json", pretty=True)`).
+Use `plotly.express` only if the user explicitly asks for it, otherwise use `matplotlib.pyplot`.
+If `plotly.express` is used, save each chart by using `fig.write_json` (e.g., `fig.write_json("path/to/file.json", pretty=True)`).
 
 
 Update this initial code: