add avatar

pages/2_Context-based_chatbot.py

@@ -195,7 +195,7 @@ if 'descriptions_context' not in st.session_state:
 st.title("You name it! 🗣")
 
 # JS: would remove Simon by some neutral avatar
-with st.chat_message('user'):
+with st.chat_message('user', avatar='julian.jpg'):
   st.write("Hey assistant!")
   
 bot = st.chat_message('assistant')
@@ -204,7 +204,7 @@ bot.write("Hello human! Wanna practice naming some words?")
 #for showing history of messages_context
 for message in st.session_state.messages_context:
   if message['role'] == 'user':
-    with st.chat_message(message['role']):
+    with st.chat_message(message['role'], avatar='julian.jpg'):
       st.markdown(message['content'])
   else:
     with st.chat_message(message['role']):
@@ -214,7 +214,7 @@ for message in st.session_state.messages_context:
 prompt_context = get_text() 
 if prompt_context:
   #JS: would replace Simon by some neutral character 
-  with st.chat_message('user'):
+  with st.chat_message('user', avatar="julian.jpg"):
     st.markdown(prompt_context)
   #add to history
   st.session_state.messages_context.append({'role': 'user', 'content': prompt_context})