Update app.py

app.py

@@ -101,15 +101,20 @@ def get_conversation_chain(vectorstore, tokenH):
 
 
 def handle_userinput(user_question):
+    # Ensure chat_history is initialized
     if "chat_history" not in st.session_state:
         st.session_state.chat_history = []
+
+    # Get the response from the conversation
     response = st.session_state.conversation(user_question)
 
-    st.session_state.chat_history = [{"role": "user", "content": user_question}]
+    # Append the user's question and the assistant's response to chat history
+    st.session_state.chat_history.append({"role": "user", "content": user_question})
     st.session_state.chat_history.append({"role": "assistant", "content": response})
 
-    for i, message in enumerate(st.session_state.chat_history):
-        if i % 2 == 0:
+    # Display the chat history
+    for message in st.session_state.chat_history:
+        if message["role"] == "user":
             st.write(f"<div style='color: gray;'>User: {message['content']}</div>", unsafe_allow_html=True)
         else:
             st.write(f"<div style='color: black;'>Bot: {message['content']}</div>", unsafe_allow_html=True)