MinaNasser commited on
Commit
e3a4ee9
·
1 Parent(s): a597565
Files changed (2) hide show
  1. .dockerignore +1 -1
  2. helpers/configs.py +26 -15
.dockerignore CHANGED
@@ -10,7 +10,7 @@ dist/
10
  build/
11
  chroma_data/
12
  assets/*.pt
13
- .env
14
  *.log
15
  .git/
16
  .gitignore
 
10
  build/
11
  chroma_data/
12
  assets/*.pt
13
+ #.env
14
  *.log
15
  .git/
16
  .gitignore
helpers/configs.py CHANGED
@@ -1,22 +1,33 @@
1
  from pydantic_settings import BaseSettings
 
2
 
3
  class Settings(BaseSettings):
4
- APP_NAME: str
5
- APP_VERSION: str
6
- APP_VARIENT: str
 
7
 
8
- host:str
9
- port:int
 
10
 
11
- DETECTION_MODEL:str = "mtcnn" # Options: mtcnn, yoloface
12
-
13
- YOLOFACE_MODEL_PATH: str = "assets/yolov12n-face.pt"
14
- CHROMA_DB_PATH:str = "./chroma_data"
15
- COLLECTION_NAME:str = "face_embeddings_collection"
16
- SIMILARITY_THRESHOLD:float = 0.7
17
- MAX_RESULTS:int = 1
18
- class Config:
19
- env_file = ".env"
 
 
 
 
 
 
 
 
20
 
21
- def get_settings(): ## this makes any got by "get_settings().APP_NAME"
22
  return Settings()
 
1
  from pydantic_settings import BaseSettings
2
+ from typing import Optional
3
 
4
  class Settings(BaseSettings):
5
+ # Application settings with defaults
6
+ APP_NAME: str = "FaceRecognitionAPI"
7
+ APP_VERSION: str = "1.0.0"
8
+ APP_VARIENT: str = "v1"
9
 
10
+ # Server settings with defaults
11
+ host: str = "0.0.0.0"
12
+ port: int = 7860
13
 
14
+ # Detection settings with defaults
15
+ DETECTION_MODEL: str = "mtcnn" # Options: mtcnn, yoloface
16
+ YOLOFACE_MODEL_PATH: str = "assets/yolov11n-face.pt"
17
+
18
+ # Database settings with defaults
19
+ CHROMA_DB_PATH: str = "./chroma_data"
20
+ COLLECTION_NAME: str = "face_embeddings_collection"
21
+
22
+ # Recognition settings with defaults
23
+ SIMILARITY_THRESHOLD: float = 0.7
24
+ MAX_RESULTS: int = 1
25
+
26
+ class Config:
27
+ env_file = ".env"
28
+ env_file_encoding = 'utf-8'
29
+ case_sensitive = False
30
+ extra = "ignore" # Ignore extra environment variables
31
 
32
+ def get_settings():
33
  return Settings()