Update app.py

app.py

@@ -108,19 +108,27 @@ def handle_userinput(user_question):
 
     for i, message in enumerate(st.session_state.chat_history):
         if i % 2 == 0:
-            # Display user messages
-            st.write(user_template.replace("{{MSG}}", message["content"]), unsafe_allow_html=True)
+            st.write(user_template.replace("{{MSG}}", message.content), unsafe_allow_html=True
+            # st.write(f"<div style='color: gray;'>User: {message['content']}</div>", unsafe_allow_html=True)
         else:
-            # Display assistant messages
-            st.write(bot_template.replace("{{MSG}}", message["content"]), unsafe_allow_html=True)
-            
+            st.write(bot_template.replace("{{MSG}}", message.content), unsafe_allow_html=True
+            # st.write(f"<div style='color: black;'>Bot: {message['content']}</div>", unsafe_allow_html=True)
+
     # for i, message in enumerate(st.session_state.chat_history):
     #     if i % 2 == 0:
-    #         st.write(user_template.replace("{{MSG}}", message.content), unsafe_allow_html=True
-    #         # st.write(f"<div style='color: gray;'>User: {message['content']}</div>", unsafe_allow_html=True)
+    #         # Display user messages
+    #         st.write(user_template.replace("{{MSG}}", message["content"]), unsafe_allow_html=True)
     #     else:
-    #         st.write(bot_template.replace("{{MSG}}", message.content), unsafe_allow_html=True
-    #         # st.write(f"<div style='color: black;'>Bot: {message['content']}</div>", unsafe_allow_html=True)
+    #         # Display assistant messages
+    #         st.write(bot_template.replace("{{MSG}}", message["content"]), unsafe_allow_html=True)
+            
+    for i, message in enumerate(st.session_state.chat_history):
+        if i % 2 == 0:
+            st.write(user_template.replace("{{MSG}}", message.content), unsafe_allow_html=True
+            # st.write(f"<div style='color: gray;'>User: {message['content']}</div>", unsafe_allow_html=True)
+        else:
+            st.write(bot_template.replace("{{MSG}}", message.content), unsafe_allow_html=True
+            # st.write(f"<div style='color: black;'>Bot: {message['content']}</div>", unsafe_allow_html=True)
 
 
 def main():