Update src/database.py

src/database.py

@@ -1,5 +1,6 @@
 from typing import Any, Dict, List
 import chromadb
+import os
 from deepface import DeepFace
 from uuid import uuid4
 from chromadb.utils.distance_functions import cosine
@@ -59,7 +60,11 @@ class DeepFaceEmebeddingFunction(EmbeddingFunction[Documents]):
 
 class Database:
     def __init__(self, db_path: str):
-        self.client = chromadb.PersistentClient(path=db_path)
+        self.client = chromadb.CloudClient(
+          api_key=os.getenv("api_key"),
+          tenant=os.getenv("tenant"),
+          database=os.getenv("database")
+        )
         self.embedding_func=DeepFaceEmebeddingFunction(model_name=EMBEDDING_MODEL, detector_backend=DETECTOR_BACKEND)
         self.collection = self.client.get_or_create_collection(
             name="face-database",