Update app.py

app.py

@@ -118,9 +118,11 @@ def handle_userinput(user_question):
     # Display the chat history
     for message in st.session_state.chat_history:
         if message["role"] == "user":
-            st.write(f"<div style='color: white;background-red: lightgray; padding: 0 1.5rem; border-radius: 50%;'>User: {message['content']}</div>", unsafe_allow_html=True)
+            st.write(user_template.replace("{{MSG}}", message['content']), unsafe_allow_html=True)
+            # st.write(f"<div style='color: white;background-red: lightgray; padding: 0 1.5rem; border-radius: 50%;'>User: {message['content']}</div>", unsafe_allow_html=True)
         else:
-            st.write(f"<div style='color: white;background-color: blue; padding: 0 1.5rem; border-radius: 50%;'>Bot: {message['content']}</div>", unsafe_allow_html=True)
+            st.write(bot_template.replace("{{MSG}}", message['content']), unsafe_allow_html=True)
+            # st.write(f"<div style='color: white;background-color: blue; padding: 0 1.5rem; border-radius: 50%;'>Bot: {message['content']}</div>", unsafe_allow_html=True)
 
     # for i, message in enumerate(st.session_state.chat_history):
     #     if i % 2 == 0: