name chroma

src/visual_qa.py

@@ -26,6 +26,21 @@ def initialize_collection(collection_name="clip_image_embeddings"):
 
     return collection
 
+def initialize_collection(collection_name="clip_image_embeddings"):
+    client = chromadb.PersistentClient(path="./chroma_db")
+
+    # Extract just the collection names
+    existing_collections = [col.name for col in client.list_collections()]
+
+    if collection_name in existing_collections:
+        collection = client.get_collection(name=collection_name)
+        print(f"Using existing collection: {collection_name}")
+    else:
+        collection = client.create_collection(name=collection_name)
+        print(f"Created new collection: {collection_name}")
+
+    return collection
+
 def main_create_image_collection(df_emb, collection_name="clip_image_embeddings"):
     documents = []
     embeddings_all = df_emb["fashion_clip_image"].tolist()