Update app.py

app.py

@@ -86,22 +86,16 @@ def main(vectorstore):
 def handle_userinput(user_question,vectorstore):
     Rag_chain = create_conversational_rag_chain(vectorstore)
     msgs = StreamlitChatMessageHistory(key="special_app_key")
-    response = Rag_chain.invoke(
-    {"input": user_question},
-    
-    config={
-        "configurable": {"session_id": "k"}
-    }, 
-    )["answer"]
-    
+    response = st.session_state.conversation({'question': user_question})
     st.session_state.chat_history = response['chat_history']
 
-    # Display messages in chat history
     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(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}}", message.content), unsafe_allow_html=True)
 
     if 'retrieved_documents' in response:
         st.subheader("Retrieved Documents")
@@ -167,7 +161,7 @@ def create_conversational_rag_chain(vectorstore):
         return_source_documents=True
     )
 
-    return conversational_rag_chain
+    return rag_chain