Update app.py

app.py

@@ -2,25 +2,21 @@ import os
 import streamlit as st
 import yfinance as yf
 import pandas as pd
-from langchain.agents import create_csv_agent, AgentType
+from langchain.agents import  AgentType
+from langchain_experimental.agents.agent_toolkits import create_pandas_dataframe_agent
 from langchain.chat_models import ChatOpenAI
 from htmlTemplates import css, user_template, bot_template
-from langchain.llms.base import LLM
-from typing import Optional, List
-import g4f
 
+# Set OpenAI API Key
 os.environ['OPENAI_API_KEY'] = os.getenv('OPENAI_API_KEY')
 
-print("API Key:",  os.getenv('OPENAI_API_KEY'))
-
+# Initialize LangChain ChatOpenAI agent
 llm = ChatOpenAI(
     model='gpt-3.5-turbo',
     max_tokens=500,
     temperature=0.7,
 )
 
-
-
 def init_ses_states():
     st.session_state.setdefault('chat_history', [])
 
@@ -82,9 +78,9 @@ def main():
                 try:
                     DF = pd.DataFrame(df)
                     DF.to_csv('data.csv')
-                    agent = create_csv_agent(
+                    agent = create_pandas_dataframe_agent(
                         llm,
-                        'data.csv',
+                        DF,
                         verbose=True,
                         agent_type=AgentType.OPENAI_FUNCTIONS,
                     )