changed avatars

app.py

@@ -53,9 +53,9 @@ Some of your dialogues in the past:
 - Blue car, kahaan hai
 - Dekho, white car :) (looking at a random white car on road)
 - Duddoo (milk) peena hai
-
 """
 
+# st.set_page_config(menu_items={"About": None})
 ipAddress = st.context.headers.get("x-forwarded-for")
 
 
@@ -93,6 +93,7 @@ def predict(prompt):
 
 
 st.title("Chat with Aariz Bhayya 👶🏻")
+# st.image("baby.png")
 
 if "messages" not in st.session_state:
     st.session_state.messages = []
@@ -101,16 +102,17 @@ if "messages" not in st.session_state:
 for message in st.session_state.messages:
     role = message["role"]
     content = message["content"]
-    with st.chat_message(role):
+    avatar = "baby.png" if role == "user" else "man.png"
+    with st.chat_message(role, avatar=avatar):
         st.markdown(content)
 
 if prompt := st.chat_input("Mummaaaaa..."):
-    with st.chat_message("user"):
+    with st.chat_message("user", avatar="man.png"):
         st.markdown(prompt)
     pprint(f"{prompt=}")
     st.session_state.messages.append({"role": "user", "content": prompt })
 
-    with st.chat_message("assistant"):
+    with st.chat_message("assistant", avatar="baby.png"):
         responseGenerator = predict(prompt)
         response = st.write_stream(responseGenerator)
     pprint(f"{response=}")