Update app.py

app.py

@@ -11,9 +11,9 @@ def respond(user_message, history, chat_state):
     if chat_state is None:
         # Start a new chat with an initial greeting
         chat_state = model.start_chat(
-            history=[
-                {"author": "user", "parts": ["Hello"]},
-                {"author": "assistant", "parts": ["Great to meet you. What would you like to know?"]},
+            messages=[
+                {"author": "user", "content": "Hello"},
+                {"author": "assistant", "content": "Great to meet you. What would you like to know?"},
             ]
         )
         # Initialize history if it's empty
@@ -21,11 +21,12 @@ def respond(user_message, history, chat_state):
             history = [["Hello", "Great to meet you. What would you like to know?"]]
     else:
         # Continue the conversation
-        chat_state.messages.append({"author": "user", "parts": [user_message]})
-        # Generate a response
-        response = chat_state.generate_message()
-        # Append the user's message and model's response to the history
-        history.append([user_message, response.text])
+        pass  # No need to manually append messages; send_message handles it
+
+    # Send the user's message to the model
+    response = chat_state.send_message(user_message)
+    # Append the user's message and model's response to the history
+    history.append([user_message, response.text])
     return history, chat_state, ''
 
 with gr.Blocks() as demo: