feat: try palm2

app.py

@@ -1,8 +1,12 @@
 import streamlit as st
+import google.generativeai as palm
 import random
 import time
+import os 
 
-st.title("Simple chat")
+palm.configure(api_key=os.environ.get('API_Key'))
+
+st.title("PaLM2 Chat")
 
 # Initialize chat history
 if "messages" not in st.session_state:
@@ -25,13 +29,7 @@ if prompt := st.chat_input("What is up?"):
     with st.chat_message("assistant"):
         message_placeholder = st.empty()
         full_response = ""
-        assistant_response = random.choice(
-            [
-                "Hello there! How can I assist you today?",
-                "Hi, human! Is there anything I can help you with?",
-                "Do you need help?",
-            ]
-        )
+        assistant_response = palm.chat(messages=[prompt]).last
         # Simulate stream of response with milliseconds delay
         for chunk in assistant_response.split():
             full_response += chunk + " "