Update app.py

app.py

@@ -81,10 +81,14 @@ def get_vectorstore_from_url(url):
         model_name=model, encode_kwargs=encode_kwargs, model_kwargs={"device": "cpu"}
     )
     # load from disk
-    vector_store = Chroma(persist_directory="/home/user/.cache/chroma_db", embedding_function=embeddings)
+    #vector_store = Chroma(persist_directory="/home/user/.cache/chroma_db", embedding_function=embeddings)
  
     #vectorstore = FAISS.from_texts(texts=text_chunks, embedding=embeddings)
     vector_store = Chroma.from_documents(document_chunks, embeddings, persist_directory="/home/user/.cache/chroma_db")
+    
+    all_documents = vector_store.get()['documents']
+    total_records = len(all_documents)
+    print("Total records in the collection: ", total_records)
 
     return vector_store