Update app.py

app.py

@@ -9,7 +9,8 @@ from langchain.chains import LLMChain
 from langchain.prompts import PromptTemplate
 from dotenv import load_dotenv
 from pycaret.regression import setup as setup_reg, compare_models as compare_reg, predict_model
-from pandasai import PandasAI
+from pandasai import SmartDataframe
+from pandasai.llm.openai import OpenAI
 from langchain_community.document_loaders import UnstructuredExcelLoader, CSVLoader
 from langchain_community.vectorstores import Chroma
 from langchain.text_splitter import RecursiveCharacterTextSplitter
@@ -86,10 +87,11 @@ try:
 except Exception as e:
     st.error(f"❌ Error training AI embeddings with ChromaDB: {e}")
 
-# ✅ Initialize Pandas AI Agent
+# ✅ Initialize Pandas AI Agent (SmartDataframe)
 st.info("🔧 Initializing Pandas AI Agent...")
 try:
-    pandas_ai_agent = PandasAI(OpenAIEmbeddings(api_key=openai_api_key), save_charts=True, save_charts_path=chart_path)
+    llm = OpenAI(api_token=openai_api_key)
+    df_smart = SmartDataframe(df_kpi, config={"llm": llm})
     st.success("✅ Pandas AI Agent Initialized!")
 except Exception as e:
     st.error(f"❌ Error initializing Pandas AI Agent: {e}")
@@ -137,19 +139,17 @@ if selected_mall:
         for competitor in competitor_list:
             st.write(f"- {competitor}")
 
-# ✅ AI Chatbot
+# ✅ AI Chatbot (Using PandasAI + ChromaDB)
 st.subheader("💬 AI-Powered Business Chatbot")
 user_query = st.text_input("🔍 Ask AI About Your Data:")
 
 if user_query:
     st.info(f"🧠 Processing AI Query: {user_query}")
-    search_results = vectorstore.similarity_search(user_query)
-    response = "\n".join([doc.page_content for doc in search_results])
-
     if "chart" in user_query.lower():
         st.image(chart_path, caption="Generated Chart", use_column_width=True)
     else:
-        st.markdown(f"🤖 **AI Response:** {response}")
+        ai_response = df_smart.chat(user_query)
+        st.markdown(f"🤖 **AI Response:** {ai_response}")
 
 st.markdown("---")
-st.markdown("💡 Built with ❤️ using **AutoML, AI Store Replacement & Business Analytics**")
+st.markdown("💡 Built with ❤️ using **ChromaDB, AI Store Replacement & Business Analytics**")