Update app.py

app.py

@@ -12,8 +12,8 @@ def respond(user_message, history, chat_state):
         # Start a new chat with an initial greeting
         chat_state = model.start_chat(
             history=[
-                {"author": "user", "content": "Hello"},
-                {"author": "assistant", "content": "Great to meet you. What would you like to know?"},
+                {"author": "user", "parts": ["Hello"]},
+                {"author": "assistant", "parts": ["Great to meet you. What would you like to know?"]},
             ]
         )
         # Initialize history if it's empty
@@ -21,11 +21,11 @@ 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", "content": user_message})
+        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])
+        history.append([user_message, response.text])
     return history, chat_state, ''
 
 with gr.Blocks() as demo: