Add support for charts (sinaptik-ai#24)

Author: gventuri

README.md

@@ -8,6 +8,7 @@
 Pandas AI is a Python library that adds generative artificial intelligence capabilities to Pandas, the popular data analysis and manipulation tool. It is designed to be used in conjunction with Pandas, and is not a replacement for it.
 
 <!-- Add images/pandas-ai.png -->
+
 ![PandasAI](images/pandas-ai.png?raw=true)
 
 ## Demo
@@ -68,6 +69,17 @@ The above code will return the following:
 14516000
 ```
 
+You can also ask PandasAI to draw a graph:
+
+```python
+pandas_ai.run(
+    df,
+    "Plot the histogram of countries showing for each the gpd, using different colors for each bar",
+)
+```
+
+![Chart](images/histogram-chart.png?raw=true)
+
 You can find more examples in the [examples](examples) directory.
 
 ## Environment Variables

examples/from_dataframe.py

@@ -1,9 +1,9 @@
 """Example of using PandasAI with a Pandas DataFrame"""
 
+from data.sample_dataframe import dataframe
 import pandas as pd
 from pandasai import PandasAI
 from pandasai.llm.openai import OpenAI
-from .data.sample_dataframe import dataframe
 
 df = pd.DataFrame(dataframe)
 

examples/show_chart.py

@@ -0,0 +1,17 @@
+"""Example of using PandasAI to generate a chart from a Pandas DataFrame"""
+
+from data.sample_dataframe import dataframe
+import pandas as pd
+from pandasai import PandasAI
+from pandasai.llm.openai import OpenAI
+
+df = pd.DataFrame(dataframe)
+
+llm = OpenAI()
+pandas_ai = PandasAI(llm)
+response = pandas_ai.run(
+    df,
+    "Plot the histogram of countries showing for each the gpd, using different colors for each bar",
+)
+print(response)
+# Output: check out images/histogram-chart.png

examples/with_privacy_enforced.py

@@ -1,9 +1,9 @@
 """Example of using PandasAI with a Pandas DataFrame"""
 
+from data.sample_dataframe import dataframe
 import pandas as pd
 from pandasai import PandasAI
 from pandasai.llm.openai import OpenAI
-from .data.sample_dataframe import dataframe
 
 df = pd.DataFrame(dataframe)
 

images/histogram-chart.png

@@ -13,6 +13,7 @@ python-dotenv = "^1.0.0"
 pandas = "^2.0.1"
 astor = "^0.8.1"
 openai = "^0.27.5"
+matplotlib = "^3.7.1"
 
 
 [tool.poetry.group.dev.dependencies]