and here it goes

app.py

@@ -1,70 +1,51 @@
-# +
+import os
 import streamlit as st
-from pandasai import SmartDataframe # requires 1.x numpy
-from pandasai.llm import OpenAI
+import pandas as pd
 import matplotlib.pyplot as plt
+from pandasai.llm.openai import OpenAI
+from pandasai import Agent
+from pandasai.responses.streamlit_response import StreamlitResponse
 
-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 = []
-
-
-# -
+llm = OpenAI(api_token=st.secrets["OPENAI_API_KEY"])
+df1 = pd.read_csv("data.csv")
 
-st.set_page_config(
-    page_title="Chat tool",
-    page_icon="🦜",
+agent = Agent(
+    [df1],
+    config={"verbose": True, "response_parser": StreamlitResponse, "llm": llm},
 )
-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
-                      }
-            )
 
+st.title("Pandas-AI demo using climate data")
 
-# +
-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)
+'''
+A simple demo based on Mauna Loa CO2 Records.  Ask questions of the data below:
+'''
 
-            if x is not None:
-                st.write(x)
-            st.session_state.prompt_history.append(question)
+## Dataset
+st.dataframe(df1)
 
-    if st.session_state.df is not None:
-        st.subheader("Current dataframe:")
-        st.write(st.session_state.df)
+## Chatbot
 
-    st.subheader("Prompt history:")
-    st.write(st.session_state.prompt_history)
 
+with st.container():
+    prompt = st.chat_input("Plot CO2 over time")
+    if prompt:
+        with st.spinner():
+            resp = agent.chat(prompt)
+            if os.path.isfile('exports/charts/temp_chart.png'):
+                im = plt.imread('exports/charts/temp_chart.png')
+                st.image(im)
+                os.remove('exports/charts/temp_chart.png')
+            st.write(resp)
+        
+        
+st.divider()
 
-if st.button("Clear"):
-    st.session_state.prompt_history = []
-    st.session_state.df = None
+'''
 
+[pandas-ai docs](https://docs.pandas-ai.com/intro)
 
+Copyright 2024 [Carl Boettiger](https://carlboettiger.info), UC Berkeley.  
+License: BSD2
+Data from: <https://gml.noaa.gov/webdata/ccgg/trends/co2/co2_mm_mlo.txt>
+'''