Update app2.py

app2.py

@@ -1,6 +1,7 @@
 import os
 import streamlit as st
 from langchain_community.utilities import GoogleSearchAPIWrapper
+from langchain.llms import GooglePalm
 from langchain.chains import ConversationChain
 from langchain.memory import ConversationBufferMemory
 import subprocess
@@ -80,15 +81,12 @@ if st.session_state.GOOGLE_API_KEY:
             return f"An unexpected error occurred while running the command."
 
     def get_agent_response(message: str, system_prompt: str):
-        # Access API key from session state
-        llm = GoogleSearchAPIWrapper(google_api_key=st.session_state.GOOGLE_API_KEY)  
-        memory = ConversationBufferMemory()
-        conversation = ConversationChain(llm=llm, memory=memory)
-        
-        # Combine system prompt and user message
-        full_prompt = f"{system_prompt}\n{message}" 
-        response = conversation.run(full_prompt)
-        return response
+      llm = GooglePalm(google_api_key=st.session_state.GOOGLE_API_KEY)  # Use GooglePalm
+      memory = ConversationBufferMemory()
+      conversation = ConversationChain(llm=llm, memory=memory) 
+      full_prompt = f"{system_prompt}\n{message}"
+      response = conversation.run(full_prompt)
+      return response
 
     def display_agent_info(agent_name: str):
         agent = agents[agent_name]