Update app.py

app.py

@@ -33,20 +33,20 @@ with gr.Blocks() as demo:
 
     def user(user_message, history):
         return gr.update(value="", interactive=False), history + [[user_message,None]]
-    def bot(history):
+    def bot(user_message, history):
         # chat_history.append(message)  # Initialize chat history
         history[-1][1] = ""
         bot_message = palm.chat(
             context=context,
             examples=examples,
-            messages=history[0][1]
+            messages=user_message
         )
         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
+        bot, [msg, chatbot], [msg, chatbot]
     )
     response.then(lambda: gr.update(interactive=True), None, [msg], queue=False)