Update app.py

app.py

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