Update app.py

app.py

@@ -1,4 +1,3 @@
-import gradio as gr
 import os
 import time
 import google.generativeai as palm
@@ -22,8 +21,8 @@ examples = [
     ]
 ]
 
-user_message = []
-history = []
+# user_message = ['']
+# history = ['']
 
 
 with gr.Blocks() as demo:
@@ -32,29 +31,23 @@ with gr.Blocks() as demo:
     clear = gr.ClearButton([msg, chatbot])
 
     def user(user_message, history):
-        user_message.append(msg.value)
-        return gr.update(value="", interactive=False), history + [[user_message[-1], None]]
-
+        return gr.update(value="", interactive=False), history + user_message
     def bot(history):
-        # Initialize the GenerateMessageRequest object
-        request = generate_message_pb2.GenerateMessageRequest()
-
-        # Set the prompt field
-        request.prompt.context = context
-        request.prompt.examples = examples
-
-        # Set the messages field
-        request.prompt.messages = history
-
-        # Generate the message
-        response = client.generate_message(request)
-
-        return response.message
-
+        # chat_history.append(message)  # Initialize chat history
+        bot_message = palm.chat(
+            context=context,
+            examples=examples,
+            messages=history
+        )
+        history[-1][1] = ""
+        for character in bot_message:
+            history[-1][1] += character
+            time.sleep(0.05)
+            yield history
     response = msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
         bot, chatbot, chatbot
     )
     response.then(lambda: gr.update(interactive=True), None, [msg], queue=False)
 
-    demo.queue()
-    demo.launch()
+demo.queue()
+demo.launch()