Update app.py

app.py

@@ -97,7 +97,9 @@ if persist_directory :
                 # https://docs.streamlit.io/library/api-reference/chat/st.chat_message
                 # https://discuss.streamlit.io/t/streamlit-chat-avatars-not-working-on-cloud/46713/2
                 # thumbs, adventurer, big-smile, micah, bottts
-                message(st.session_state["user_history"][i], is_user=True, key=str(i) + '_user', avatar_style="adventurer")
+                # message(st.session_state["user_history"][i], is_user=True, key=str(i) + '_user', avatar_style="adventurer")
                 # message(st.session_state["ai_history"][i], key=str(i), avatar_style="KaggleX.jpg")
-                message(st.session_state["ai_history"][i], key=str(i), avatar_style='bottts')
+                # message(st.session_state["ai_history"][i], key=str(i), avatar_style='bottts')
+                st.chat_message(st.session_state["user_history"][i], is_user=True, key=str(i) + '_user', AvatarStyle="adventurer")
+                st.chat_message(st.session_state["ai_history"][i], key=str(i), AvatarStyle='bottts')