-
Notifications
You must be signed in to change notification settings - Fork 1.4k
/
save_chart.py
35 lines (28 loc) · 1007 Bytes
/
save_chart.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
"""Example of using PandasAI to generate and save a chart from a Pandas DataFrame"""
import os
import pandas as pd
from data.sample_dataframe import dataframe
from pandasai import Agent
from pandasai.helpers import path
df = pd.DataFrame(dataframe)
try:
user_defined_path = path.find_project_root()
except ValueError:
user_defined_path = os.getcwd()
# By default, unless you choose a different LLM, it will use BambooLLM.
# You can get your free API key signing up at https://pandabi.ai (you can also configure it in your .env file)
os.environ["PANDASAI_API_KEY"] = "your-api-key"
user_defined_path = os.path.join(user_defined_path, "exports", "charts")
agent = Agent(
df,
config={
"save_charts_path": user_defined_path,
"save_charts": True,
"verbose": True,
},
)
response = agent.chat(
"Plot the histogram of countries showing for each the gpd,"
" using different colors for each bar",
)
# Output: check out $pwd/exports/charts/{hashid}/chart.png