Update app.py

app.py

@@ -91,15 +91,14 @@ def create_retriever_from_chroma(vectorstore_path="docs/chroma/", search_type='m
     contextualize_q_system_prompt = """If the question is not clear, given a chat history and the latest user question
     which maybe reference context, formulate a standalone question,
     which can be understood without the chat history. Do NOT answer the question,
-    just reformulate it if needed and otherwise return it as is."""
+    just reformulate it if needed and otherwise return it as is.""")
     
-   )
 
     contextualize_q_prompt = ChatPromptTemplate.from_messages(
     [
         ("system", contextualize_q_system_prompt),
         MessagesPlaceholder("chat_history"),
-        ("human", "{input}"),
+        ("human", "{question}"),
     ]
 )
 
@@ -148,7 +147,7 @@ def main():
 def handle_userinput(user_question,retriever):
     st.session_state.messages.append({"role": "user", "content": user_question})
     st.chat_message("user").write(user_question)
-    docs = retriever.invoke(user_question)
+    docs = retriever.invoke({"question": user_question, "chat_history": st.session_state.messages})
 
     with st.sidebar:
         st.subheader("Your documents")