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"