import os
from pathlib import Path

class Config:
    BASE_DIR = Path(__file__).parent
    CHECKPOINT_DIR = BASE_DIR / "checkpoints"
    UPLOAD_DIR = BASE_DIR / "uploads"

    MEDSAM2_CHECKPOINT_URL = os.getenv(
        'MEDSAM2_URL',
        'https://huggingface.co/wanglab/MedSAM2/resolve/main/MedSAM2_latest.pt'
    )

    MAX_RAM_GB = float(os.getenv('MAX_RAM_GB', '4.0'))
    MAX_CHECKPOINT_SIZE_MB = int(os.getenv('MAX_CHECKPOINT_MB', '1000'))

    DEBUG = os.getenv('DEBUG', 'False').lower() == 'true'

    @classmethod
    def ensure_directories(cls):
        cls.CHECKPOINT_DIR.mkdir(exist_ok=True)
        cls.UPLOAD_DIR.mkdir(exist_ok=True)