Update app.py

app.py

@@ -159,7 +159,11 @@ st.write("📂 **Vector Store Created:**", st.session_state.vector_created)
 query = st.text_input("🔍 Ask a question about the document:")
 if query:
     with st.spinner("🔄 Retrieving relevant context..."):
-        retriever = st.session_state.vector_store.as_retriever(search_type="similarity", search_kwargs={"k": 5})
+        if st.session_state.vector_store is None:
+            st.error("❌ Vector store is not initialized. Ensure document processing and chunking are completed.")
+        else:
+            retriever = st.session_state.vector_store.as_retriever(search_type="similarity", search_kwargs={"k": 5})
+
         contexts = retriever.invoke(query)
         # Debugging: Check what was retrieved
         st.write("Retrieved Contexts:", contexts)