Update app.py

app.py

@@ -35,13 +35,14 @@ def create_vector_embedding():
         st.session_state.loader = PyPDFDirectoryLoader("documents")
         st.session_state.docs = st.session_state.loader.load()
         st.session_state.text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=200)
-        st.session_state.final_documents = st.session_state.text_splitter.split_documents()
+        st.session_state.final_documents = st.session_state.text_splitter.split_documents(st.session_state.docs[:50])
         st.session_state.vectors = FAISS.from_documents(st.session_state.final_documents, st.session_state.embeddings)
 
 if "vectors" not in st.session_state:
     st.write("The vector store database is not yet ready")
     if st.button("Create"):
-        create_vector_embedding()
+        with st.spinner("Working..."):
+            create_vector_embedding()
         st.write("Done")
 
 if "vectors" in st.session_state: