Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import os
 import time
-import google.generativeai as palm
+from google.generativeai import palm
 
 palm.configure(api_key=os.environ.get("palm_key"))
 
@@ -25,10 +25,6 @@ examples = [
 user_input = []
 
 
-import gradio as gr
-import random
-import time
-
 with gr.Blocks() as demo:
     chatbot = gr.Chatbot()
     msg = gr.Textbox()
@@ -36,7 +32,7 @@ with gr.Blocks() as demo:
 
     def respond(messages):
         global user_input
-        user_input.append(messages[0])
+        user_input.append([messages[0]])  # Wrap user input in a list
         bot_message = palm.chat(
             **defaults,
             context=context,
@@ -44,8 +40,8 @@ with gr.Blocks() as demo:
             messages=user_input
         )
         time.sleep(2)
-        return messages
+        return bot_message['message']
 
-    msg.submit(respond, msg,chatbot)
+    msg.submit(respond, [msg, chatbot])
 
 demo.launch()