Update app.py

app.py

@@ -62,13 +62,13 @@ def get_conversation_chain(vectorstore, model_name):
 
     llm = ChatOpenAI()
     
-    memory = ConversationBufferMemory(memory_key='chat_history', return_messages=True)
+    memory = ConversationBufferMemory(memory_key='chat_history', input_key='question', output_key='answer', return_messages=True)
     
     conversation_chain = ConversationalRetrievalChain.from_llm(llm=llm,
                                                                # condense_question_prompt=CONDENSE_QUESTION_PROMPT,
                                                                retriever=vectorstore.as_retriever(),
                                                                memory=memory,
-                                                               #return_source_documents=True
+                                                               return_source_documents=True
                                                                )
 
     return conversation_chain
@@ -79,15 +79,17 @@ def handle_userinput(user_question):
 
     st.session_state.chat_history = response['chat_history']
 
-    #st.session_state.retrieved_text = response['source_documents'][0]
+    st.session_state.retrieved_text = response['source_documents'][0]
 
-    for i, message in enumerate(st.session_state.chat_history):
+    for i, (message, text) in enumerate(zip(st.session_state.chat_history, st.session_state.retrieved_text):
         if i % 2 == 0:
             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(bot_template.replace(
+                "{{MSG}}", text.content), unsafe_allow_html=True)