Update main.py

main.py

@@ -1,4 +1,5 @@
 import google.generativeai as palm
+from langchain_google_genai import ChatGoogleGenerativeAI
 import pandas as pd
 import os
 import io
@@ -52,7 +53,8 @@ def bot():
     #df['date'] = pd.to_datetime(df['date'], format="%b %d", errors='coerce')
     #df['Profit'] = df['Profit'].apply(lambda x: "R{:.1f}".format((x)))
     #df['Revenue'] = df['Revenue'].apply(lambda x: "R{:.1f}".format((x)))
-    llm = GooglePalm(temperature=0, google_api_key=os.environ['PALM'])
+    llm = ChatGoogleGenerativeAI(model='gemini-1.5-pro', temperature=0.1)
+    #llm = GooglePalm(temperature=0, google_api_key=os.environ['PALM'])
     
     agent = create_pandas_dataframe_agent(llm, df, agent="structured_chat-zero-shot-react-description", verbose=True)
     response = agent.run(user_question)