Update app.py

app.py

@@ -57,17 +57,17 @@ def init_chromadb():
     vectorstore = Chroma.from_documents(texts, embeddings, persist_directory=DB_DIR)
     #vectorstore.persist()
     
-    print(vectorstore)
+    #print(vectorstore)
     #vectorstore = None
 
-    db = vectorstore
-    db.get()
-    print(len(db.get()["ids"]))
+    #db = vectorstore
+    #db.get()
+    #print(len(db.get()["ids"]))
 
     # Print the list of source files
-    for x in range(len(db.get()["ids"])):
+    for x in range(len(vectorstore.get()["ids"])):
         # print(db.get()["metadatas"][x])
-        doc = db.get()["metadatas"][x]
+        doc = vectorstore.get()["metadatas"][x]
         source = doc["source"]
         print(source)