Create config.py

src/core/config.py

@@ -0,0 +1,41 @@
+import os
+from dotenv import load_dotenv
+
+load_dotenv()
+
+DEFAULT_PINECONE_KEY = os.getenv("DEFAULT_PINECONE_KEY", "")
+DEFAULT_CLOUDINARY_URL = os.getenv("DEFAULT_CLOUDINARY_URL", "")
+IDX_FACES = os.getenv("IDX_FACES", "enterprise-faces")
+IDX_OBJECTS = os.getenv("IDX_OBJECTS", "enterprise-objects")
+
+MAX_CONCURRENT_INFERENCES = int(os.getenv("MAX_CONCURRENT_INFERENCES", "1"))
+MAX_FILES_PER_UPLOAD = int(os.getenv("MAX_FILES_PER_UPLOAD", "50"))
+INFERENCE_CACHE_SIZE = int(os.getenv("INFERENCE_CACHE_SIZE", "128"))
+
+SUPABASE_URL = os.getenv("SUPABASE_URL", "")
+SUPABASE_SERVICE_KEY = os.getenv("SUPABASE_SERVICE_KEY", "")
+
+MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "1024"))
+MAX_CROPS = int(os.getenv("MAX_CROPS", "10"))
+YOLO_PERSON_CLASS_ID = 0
+YOLO_MIN_CROP_PX = int(os.getenv("YOLO_MIN_CROP_PX", "50"))
+YOLO_CONF_THRESHOLD = float(os.getenv("YOLO_CONF_THRESHOLD", "0.25"))
+
+DET_SIZE_PRIMARY = (640, 640)
+DET_SCALES = [(1280, 1280), (960, 960), (640, 640)]
+IOU_DEDUP_THRESHOLD = float(os.getenv("IOU_DEDUP_THRESHOLD", "0.4"))
+MIN_FACE_SIZE = int(os.getenv("MIN_FACE_SIZE", "30"))
+MAX_FACES_PER_IMAGE = int(os.getenv("MAX_FACES_PER_IMAGE", "20"))
+FACE_QUALITY_GATE = float(os.getenv("FACE_QUALITY_GATE", "0.5"))
+
+FACE_DIM = 512
+ADAFACE_DIM = 512
+FUSED_FACE_DIM = 1024
+
+FACE_CROP_THUMB_SIZE = int(os.getenv("FACE_CROP_THUMB_SIZE", "112"))
+FACE_CROP_QUALITY = int(os.getenv("FACE_CROP_QUALITY", "85"))
+FACE_CROP_PADDING = float(os.getenv("FACE_CROP_PADDING", "0.2"))
+ADAFACE_CROP_PADDING = float(os.getenv("ADAFACE_CROP_PADDING", "0.1"))
+
+ENABLE_ADAFACE = int(os.getenv("ENABLE_ADAFACE", "1"))
+HF_TOKEN = os.getenv("HF_TOKEN", "")