Update app.py

app.py

@@ -118,9 +118,12 @@ if prompt := st.chat_input("What are you looking to learn?"):
     with st.chat_message("user"):
         st.markdown(prompt)
 
+    # Retrieve relevant context from the vector store based on user input
+    context = retriever.retrieve(prompt)
+
     # Assistant response generator with streaming effect
     with st.chat_message("assistant"):
-        response = qa_chain({"question": prompt})
+        response = qa_chain({"question": prompt, "context": context})
         response_text = response["answer"]
 
         # Simulate streaming response