import os
import runpy
import sys

from huggingface_hub import snapshot_download

MODELS = {
    "Qwen/Qwen-Image-Edit-2511": "Base model (supplies the text encoder, VAE, tokenizer and scheduler; its transformer is replaced by the tuned one)",
    "Phr00t/Qwen-Image-Edit-Rapid-AIO": "Tuned accelerated transformer weights for 4-step image editing (replaces the base transformer)",
    "black-forest-labs/FLUX.2-klein-4B": "Image refinement model (option 1 of 3, one is used at a time)",
    "black-forest-labs/FLUX.2-klein-9B": "Image refinement model (option 2 of 3, one is used at a time)",
    "black-forest-labs/FLUX.2-klein-9b-kv": "Image refinement model (option 3 of 3, one is used at a time)",
    "nateraw/vit-age-classifier": "Age verification model (classifies the age bracket of detected face crops)",
    "LiquidAI/LFM2.5-VL-450M": "Age verification VLM (primary, first-pass check on every image)",
    "OpenGVLab/InternVL3-1B-hf": "Age verification VLM (fallback, loaded only when the primary cannot be loaded)",
}

NON_HUB_COMPONENTS = {
    "YuNet face detector (OpenCV Zoo, face_detection_yunet_2023mar.onnx)": "Face detection for age verification (primary detector)",
    "MediaPipe BlazeFace short-range face detector (Google)": "Face detection for age verification (secondary detector)",
    "OpenCV Haar cascade frontal face detector (OpenCV built-in)": "Face detection for age verification (last-resort fallback)",
}


def _require_env(name):
    value = (os.environ.get(name) or "").strip()
    if not value:
        raise RuntimeError(f"Missing required Space secret: {name}")
    return value


code_folder = _require_env("CODE_DIR")
snapshot = snapshot_download(
    repo_id=_require_env("CODE_REPO"),
    token=_require_env("HF_TOKEN"),
    allow_patterns=[f"{code_folder}/*"],
)
code_dir = os.path.join(snapshot, code_folder)
sys.path.insert(0, code_dir)
runpy.run_path(os.path.join(code_dir, "app.py"), run_name="__main__")