early_env.py

# early_env.py — import this BEFORE numpy/torch/cv2/etc.
import os, re

def _clean_int_env(name: str, default: str | None = None):
    val = os.environ.get(name)
    if val is None:
        if default is not None:
            os.environ[name] = default
        return
    if not re.fullmatch(r"\d+", str(val).strip()):
        if default is None:
            os.environ.pop(name, None)
        else:
            os.environ[name] = default

# Keep thread env sane for Spaces (prevents libgomp errors)
_clean_int_env("OMP_NUM_THREADS", "2")
_clean_int_env("MKL_NUM_THREADS", "2")
_clean_int_env("OPENBLAS_NUM_THREADS", "2")
_clean_int_env("NUMEXPR_NUM_THREADS", "2")

# Configure PyTorch threading EARLY (before any parallel work)
try:
    import torch
    if hasattr(torch, "set_num_interop_threads"):
        torch.set_num_interop_threads(2)
    if hasattr(torch, "set_num_threads"):
        torch.set_num_threads(2)
except Exception:
    pass