import os
import sys
from torchvision.datasets.utils import download_url

path = os.path.join(os.path.dirname(__file__))
comfy_path = os.path.join(path, '..', '..', '..')

sys.path.append(comfy_path)

import folder_paths
model_path = folder_paths.models_dir
ultralytics_bbox_path = os.path.join(model_path, "ultralytics", "bbox")

if not os.path.exists(os.path.join(path, '..', '..', 'skip_download_model')):
    if not os.path.exists(ultralytics_bbox_path):
        os.makedirs(ultralytics_bbox_path)


    if not os.path.exists(os.path.join(ultralytics_bbox_path, "face_yolov8m.pt")):
        download_url("https://huggingface.co/Bingsu/adetailer/resolve/main/face_yolov8m.pt",
                     ultralytics_bbox_path)