Update app.py

app.py

@@ -1,7 +1,7 @@
 # https://docs.streamlit.io/knowledge-base/tutorials/build-conversational-apps#build-a-chatgpt-like-app
 
 import streamlit as st
-# from streamlit_chat import message
+from streamlit_chat import message
 
 from langchain.document_loaders import CSVLoader
 from langchain_openai import OpenAIEmbeddings
@@ -97,9 +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')
-                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')
+                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')