sorawm/configs.py

from pathlib import Path

ROOT = Path(__file__).parent.parent


RESOURCES_DIR = Path("/tmp/resources")
WATER_MARK_TEMPLATE_IMAGE_PATH = RESOURCES_DIR / "watermark_template.png"

WATER_MARK_DETECT_YOLO_WEIGHTS = RESOURCES_DIR / "best.pt"

WATER_MARK_DETECT_YOLO_WEIGHTS_HASH_JSON = RESOURCES_DIR / "model_version.json"



OUTPUT_DIR = Path("/tmp/output")
OUTPUT_DIR.mkdir(exist_ok=True, parents=True)


DEFAULT_WATERMARK_REMOVE_MODEL = "lama"

WORKING_DIR = Path("/tmp/working_dir")
WORKING_DIR.mkdir(exist_ok=True, parents=True)


LOGS_PATH = Path("/tmp/logs")
LOGS_PATH.mkdir(exist_ok=True, parents=True)

DATA_PATH = Path("/tmp/data")
DATA_PATH.mkdir(exist_ok=True, parents=True)

SQLITE_PATH = DATA_PATH / "db.sqlite3"