File size: 1,798 Bytes
478171f |
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 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 |
# +
import streamlit as st
from pandasai import SmartDataframe # requires 1.x numpy
from pandasai.llm import OpenAI
import matplotlib.pyplot as plt
import os
# +
import pandas as pd
import seaborn as sns
columns = ['year', 'month', 'decimal_date', 'average', 'smooth', 'std_days', 'uncertainty_monthly_mean', 'empty']
df = pd.read_csv("https://gml.noaa.gov/webdata/ccgg/trends/co2/co2_mm_mlo.txt", sep="\s+", comment="#", names= columns)
df.to_csv("data.csv")
st.session_state.df = df
st.session_state.prompt_history = []
# -
st.set_page_config(
page_title="Chat tool",
page_icon="🦜",
)
st.title("CO2 Explorer")
# +
# st.dataframe(df)
# -
llm = OpenAI(api_token=st.secrets["OPENAI_API_KEY"])
pandas_ai = SmartDataframe("data.csv",
config={
"llm": llm,
"save_charts": False,
"verbose": False
}
)
# +
chatbox = st.container()
with chatbox:
if question := st.chat_input("Plot the average CO2 concentration over time (decimal_date) using seaborn", key="chain"):
st.chat_message("user").write(question)
with st.chat_message("assistant"):
x = pandas_ai.chat(question)
if os.path.isfile(x):
im = plt.imread(x)
st.image(im)
os.remove(x)
if x is not None:
st.write(x)
st.session_state.prompt_history.append(question)
if st.session_state.df is not None:
st.subheader("Current dataframe:")
st.write(st.session_state.df)
st.subheader("Prompt history:")
st.write(st.session_state.prompt_history)
if st.button("Clear"):
st.session_state.prompt_history = []
st.session_state.df = None
|