Update app.py

app.py

@@ -91,10 +91,12 @@ def handle_userinput(user_question,vectorstore):
     config={
         "configurable": {"session_id": "k"}
     }, 
-)["answer"]
+    )["answer"]
+
+    
     st.session_state.chat_history = response['chat_history']
 
-    for i, message in enumerate(st.session_state.chat_history):
+    for i, message in enumerate(msgs):
         if i % 2 == 0:
             st.write(user_template.replace(
                 "{{MSG}}", message.content), unsafe_allow_html=True)
@@ -150,6 +152,16 @@ def create_conversational_rag_chain(vectorstore):
     rag_chain = create_retrieval_chain(ha_retriever, question_answer_chain)
     msgs = StreamlitChatMessageHistory(key="special_app_key")
 
+    conversational_rag_chain = RunnableWithMessageHistory(
+    rag_chain,
+    lambda session_id: msgs,
+    input_messages_key="input",
+    history_messages_key="chat_history",
+    output_messages_key="answer",
+    return_source_documents = True
+)
+
+
     return rag_chain