"""Post-install script for ComfyUI-InfiniDepth.

Installs git-based dependencies that cannot go into requirements.txt.
"""

import subprocess
import sys


def pip_install(*args):
    subprocess.check_call([sys.executable, "-m", "pip", "install", *args])


try:
    import moge  # noqa: F401
except ImportError:
    print("[InfiniDepth] Installing MoGe …")
    pip_install("git+https://github.com/microsoft/MoGe.git")

try:
    import gsplat  # noqa: F401
except ImportError:
    print("[InfiniDepth] Installing gsplat …")
    pip_install("git+https://github.com/nerfstudio-project/gsplat.git")