Update app.py

app.py

@@ -43,8 +43,8 @@ chat_container = st.container()
 # Loop over chat history and display messages
 for message in chat_history:
   # Get message role and text
-  role = message.role
-  text = message.parts[0].text
+  role = message['role']
+  text = message['text']
 
   # Display message with markdown
   chat_container.markdown(f"**{role}:** {text}")
@@ -55,20 +55,20 @@ user_input = st.text_input("You")
 # Check if user input is not empty
 if user_input:
   # Add user message to chat history
-  chat_history.append(user_input)
+  chat_history.append({'role': 'user', 'text': user_input})
 
   # Display user message with markdown
   chat_container.markdown(f"**user:** {user_input}")
 
   # Get model response with start_chat method
   with st.spinner("Thinking..."):
-    response = model.start_chat(user_input)
+    response = model.start_chat({'user_input': user_input})
 
   # Get response text from response object
-  response_text = response.parts[0].text
+  response_text = response['text']
 
   # Add response message to chat history
-  chat_history.append(response_text)
+  chat_history.append({'role': 'assistant', 'text': response_text})
 
   # Display response message with markdown
   chat_container.markdown(f"**assistant:** {response_text}")