Update app.py

app.py

@@ -36,22 +36,25 @@ with gr.Blocks() as demo:
         return gr.update(value="", interactive=False), history + [[user_message[-1], None]]
 
     def bot(history):
-        # 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
+        # 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
 
     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()