Update app3.py

app3.py

@@ -2,7 +2,7 @@ import streamlit as st
 import pandas as pd
 import plotly.express as px
 from datasets import load_dataset
-from pandasai import Agent
+from pandasai import SmartDataframe
 from pandasai.llm.openai import OpenAI
 from langchain_community.embeddings.openai import OpenAIEmbeddings
 from langchain_community.vectorstores import FAISS
@@ -37,7 +37,7 @@ logger.debug(f"OPENAI_API_KEY: {api_key}")
 logger.debug(f"PANDASAI_API_KEY: {pandasai_api_key}")
 
 # Title of the app
-st.title("Data Analyzer")
+st.title("PandasAI and RAG Data Analyzer")
 
 # Function to load datasets into session
 def load_dataset_into_session():
@@ -99,9 +99,11 @@ load_dataset_into_session()
 if st.session_state.df is not None:
     df = st.session_state.df
     try:
-        # Initialize PandasAI Agent
-        llm = OpenAI(api_key=pandasai_api_key)
-        agent = Agent(df, llm=llm)
+        # Initialize OpenAI LLM
+        llm = OpenAI(api_token=pandasai_api_key)  # PandasAI LLM
+
+        # Create SmartDataframe for PandasAI
+        smart_df = SmartDataframe(df, config={"llm": llm})
 
         # Convert DataFrame to documents for RAG
         documents = [
@@ -114,7 +116,7 @@ if st.session_state.df is not None:
             for index, row in df.iterrows()
         ]
 
-        # Set up RAG
+        # Set up RAG 
         embeddings = OpenAIEmbeddings()
         vectorstore = FAISS.from_documents(documents, embeddings)
         retriever = vectorstore.as_retriever()
@@ -135,7 +137,7 @@ if st.session_state.df is not None:
             pandas_question = st.text_input("Ask a question about the data (PandasAI):")
             if pandas_question:
                 try:
-                    result = agent.chat(pandas_question)
+                    result = smart_df.chat(pandas_question)
                     if result:
                         st.write("PandasAI Answer:", result)
                     else:
@@ -164,7 +166,7 @@ if st.session_state.df is not None:
             )
             if viz_question:
                 try:
-                    result = agent.chat(viz_question)
+                    result = smart_df.chat(viz_question)
                     import re
                     code_pattern = r"```python\n(.*?)\n```"
                     code_match = re.search(code_pattern, result, re.DOTALL)
@@ -181,4 +183,4 @@ if st.session_state.df is not None:
     except Exception as e:
         st.error(f"An error occurred during processing: {e}")
 else:
-    st.info("Please load a dataset to start analysis.")
+    st.info("Please load a dataset to start analysis.")