Update app.py

app.py

@@ -95,10 +95,11 @@ bot.write("Hello human! Wanna practice naming some words?")
 
 #for showing history of messages
 for message in st.session_state.messages:
-  with st.chat_message(message['role']):
-    if message['role'] == 'user':
-      st.markdown(message['content'], avatar='simon.jpg')
-    else:
+  if message['role'] == 'user':
+    with st.chat_message(message['role'], avatar='simon.jpg'):
+      st.markdown(message['content'])
+  else:
+    with st.chat_message(message['role']):
       st.markdown(message['content'])
 
 def get_text():