Update app.py

app.py

@@ -101,6 +101,8 @@ def get_conversation_chain(vectorstore, tokenH):
 
 
 def handle_userinput(user_question):
+    if "chat_history" not in st.session_state:
+        st.session_state.chat_history = []
     response = st.session_state.conversation(user_question)
 
     st.session_state.chat_history = [{"role": "user", "content": user_question}]
@@ -108,11 +110,9 @@ def handle_userinput(user_question):
 
     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)
+            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)
+            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: