Update app.py

app.py

@@ -1,7 +1,7 @@
 import os
 import streamlit as st
 import pandas as pd
-from pandasai import SmartDataframe
+from pandasai import Agent
 from pandasai.responses.streamlit_response import StreamlitResponse
 from langchain_groq import ChatGroq
 
@@ -44,14 +44,14 @@ url =  (
 
 # Load data only if URL changes
 if st.session_state['pre_url'] != url:
-    df = SmartDataframe(
+    agent = Agent(
         pd.read_csv(url, index_col=0),
         config={
             "llm": llm,
             "response_parser": StreamlitResponse
         }
     )
-    st.session_state['df'] = df
+    st.session_state['agent'] = agent
     st.session_state['pre_url'] = url
 
 # Step 2: Request for analysis or chart
@@ -61,8 +61,8 @@ request = st.text_input(
 
 # Execute chat and display results
 if st.button("Analyze"):
-    if st.session_state['df'] is not None:
-        file = st.session_state['df'].chat(request)
+    if st.session_state['agent'] is not None:
+        file = st.session_state['agent'].chat(request)
         if os.path.exists(file):
             st.image(file)
         else: