install.js

const vendors = require("./vendor_revisions")
const hunyuan3d2 = vendors.hunyuan3d2
const hunyuan3d21 = vendors.hunyuan3d21

module.exports = {
  requires: {
    bundle: "ai"
  },
  run: [
    {
      when: "{{gpu !== 'nvidia'}}",
      method: "notify",
      params: {
        html: "This app requires an NVIDIA GPU on Windows or Linux."
      },
      next: null
    },
    // Optional HuggingFace login. Maestro's default models are all on
    // PUBLIC repos, so this is NOT required — but a token lifts HuggingFace's
    // anonymous rate limits (helpful for the large model downloads) and
    // unlocks any gated models you add later. Non-blocking (wait: false):
    // Pinokio stores the token at HF_TOKEN_PATH; skip it and downloads fall
    // back to anonymous (launch.py is tolerant of an absent/blocked token).
    {
      method: "hf.login",
      params: { wait: false }
    },
    {
      method: "shell.run",
      params: {
        venv: "env",
        path: "app",
        message: [
          "uv pip install -r requirements.txt --index-strategy unsafe-best-match",
          "uv pip install hf-xet pip"
        ]
      }
    },
    {
      method: "script.start",
      params: {
        uri: "torch.js",
        params: {
          venv: "env",
          path: "app",
          xformers: true
        }
      }
    },
    // Install pre-built llama.cpp CUDA kernels for GGUF models if a
    // wheel matches the current Python / PyTorch / CUDA combo. Without
    // this, mmgp prints "[GGUF][llama.cpp CUDA] kernels unavailable,
    // using fallback" at every startup. The helper script is a soft
    // no-op when no matching wheel exists (e.g. Linux, or unreleased
    // version combo) — the fallback path still works for GGUF models,
    // and the default INT8 / BF16 variants don't use these kernels at
    // all. Idempotent on re-runs.
    {
      method: "shell.run",
      params: {
        venv: "env",
        path: "app",
        message: "python scripts/install_gguf_kernels.py"
      }
    },
    // Native Hunyuan3D runtime. Dependency isolation is internal because the
    // official stack conflicts with Maestro's audio/video Python packages;
    // from the user's perspective this is part of the normal Maestro install.
    {
      when: "{{!exists('" + hunyuan3d2.path + "')}}",
      method: "shell.run",
      params: {
        message: [
          "git clone --depth 1 " + hunyuan3d2.url + " " + hunyuan3d2.path,
          "git -C " + hunyuan3d2.path + " fetch --depth 1 origin " + hunyuan3d2.revision,
          "git -C " + hunyuan3d2.path + " checkout --detach " + hunyuan3d2.revision
        ]
      }
    },
    {
      when: "{{exists('" + hunyuan3d2.path + "/.git')}}",
      method: "shell.run",
      params: {
        path: hunyuan3d2.path,
        message: [
          "git fetch --depth 1 origin " + hunyuan3d2.revision,
          "git checkout --detach " + hunyuan3d2.revision
        ]
      }
    },
    {
      when: "{{!exists('" + hunyuan3d21.path + "')}}",
      method: "shell.run",
      params: {
        message: [
          "git clone --depth 1 " + hunyuan3d21.url + " " + hunyuan3d21.path,
          "git -C " + hunyuan3d21.path + " fetch --depth 1 origin " + hunyuan3d21.revision,
          "git -C " + hunyuan3d21.path + " checkout --detach " + hunyuan3d21.revision
        ]
      }
    },
    {
      when: "{{exists('" + hunyuan3d21.path + "/.git')}}",
      method: "shell.run",
      params: {
        path: hunyuan3d21.path,
        message: [
          "git fetch --depth 1 origin " + hunyuan3d21.revision,
          "git checkout --detach " + hunyuan3d21.revision
        ]
      }
    },
    {
      method: "shell.run",
      params: {
        conda: {
          path: "app/services/hunyuan3d/env",
          python: "3.10"
        },
        message: [
          "uv pip install torch==2.7.0 torchvision==0.22.0 torchaudio==2.7.0 --index-url https://download.pytorch.org/whl/cu128",
          "uv pip install -r app/services/hunyuan3d/requirements.txt"
        ]
      }
    },
    {
      // Clean the nested environment created by the first native-3D migration,
      // where conda.path was incorrectly resolved from this extension folder.
      when: "{{exists('app/services/hunyuan3d/vendor/Hunyuan3D-2/hy3dgen/texgen/custom_rasterizer/app')}}",
      method: "fs.rm",
      params: {
        path: "app/services/hunyuan3d/vendor/Hunyuan3D-2/hy3dgen/texgen/custom_rasterizer/app"
      }
    },
    {
      method: "shell.run",
      params: {
        conda: { path: "../../../../../env", python: "3.10" },
        env: {
          CUDA_HOME: "{{path.resolve(path.dirname(which('nvcc')), '..')}}",
          CPATH: "{{path.resolve(path.dirname(which('nvcc')), '../targets/x86_64-linux/include')}}",
          LIBRARY_PATH: "{{path.resolve(path.dirname(which('nvcc')), '../targets/x86_64-linux/lib')}}",
          LD_LIBRARY_PATH: "{{path.resolve(path.dirname(which('nvcc')), '../targets/x86_64-linux/lib')}}"
        },
        path: "app/services/hunyuan3d/vendor/Hunyuan3D-2/hy3dgen/texgen/custom_rasterizer",
        message: "uv pip install --no-build-isolation -e ."
      }
    },
    {
      method: "shell.run",
      params: {
        conda: { path: "../../../../../env", python: "3.10" },
        path: "app/services/hunyuan3d/vendor/Hunyuan3D-2/hy3dgen/texgen/differentiable_renderer",
        message: "uv pip install --no-build-isolation -e ."
      }
    },
    {
      method: "shell.run",
      params: {
        conda: { path: "../../../../env", python: "3.10" },
        env: {
          CUDA_HOME: "{{path.resolve(path.dirname(which('nvcc')), '..')}}",
          CPATH: "{{path.resolve(path.dirname(which('nvcc')), '../targets/x86_64-linux/include')}}",
          LIBRARY_PATH: "{{path.resolve(path.dirname(which('nvcc')), '../targets/x86_64-linux/lib')}}",
          LD_LIBRARY_PATH: "{{path.resolve(path.dirname(which('nvcc')), '../targets/x86_64-linux/lib')}}"
        },
        path: "app/services/hunyuan3d/vendor/Hunyuan3D-2.1/hy3dpaint/custom_rasterizer",
        message: "uv pip install --no-build-isolation -e ."
      }
    },
    {
      method: "shell.run",
      params: {
        conda: { path: "../../../../env", python: "3.10" },
        shell: "{{which('bash')}}",
        path: "app/services/hunyuan3d/vendor/Hunyuan3D-2.1/hy3dpaint/DifferentiableRenderer",
        message: "bash compile_mesh_painter.sh"
      }
    },
    {
      when: "{{!exists('app/services/hunyuan3d/vendor/Hunyuan3D-2.1/hy3dpaint/ckpt/RealESRGAN_x4plus.pth')}}",
      method: "fs.download",
      params: {
        url: "https://github.com/xinntao/Real-ESRGAN/releases/download/v0.1.0/RealESRGAN_x4plus.pth",
        path: "app/services/hunyuan3d/vendor/Hunyuan3D-2.1/hy3dpaint/ckpt/RealESRGAN_x4plus.pth"
      }
    },
    {
      method: "fs.write",
      params: {
        path: "app/services/hunyuan3d/env/.maestro_hunyuan3d_v1.installed",
        text: "ok"
      }
    },
    {
      method: "fs.write",
      params: {
        path: hunyuan3d2.marker,
        text: "repository=" + hunyuan3d2.url + "\nrevision=" + hunyuan3d2.revision + "\n"
      }
    },
    {
      method: "fs.write",
      params: {
        path: hunyuan3d21.marker,
        text: "repository=" + hunyuan3d21.url + "\nrevision=" + hunyuan3d21.revision + "\n"
      }
    },
    // MiniMax H3 runs in an isolated ComfyUI checkout because its quantized
    // ConvRot stack must not change Maestro/WanGP's Python dependencies.
    // Mirrors system/examples/comfy/install.js: clone, isolated env, then uv.
    {
      when: "{{!exists('app/services/minimax_h3/vendor/ComfyUI/main.py')}}",
      method: "shell.run",
      params: {
        message: [
          "git clone --depth 1 --branch minimax_h3 https://github.com/kijai/ComfyUI app/services/minimax_h3/vendor/ComfyUI",
          "git -C app/services/minimax_h3/vendor/ComfyUI fetch --depth 1 origin e2ab36d933356bc8cd6ecb39c655fe8be75af4e5",
          "git -C app/services/minimax_h3/vendor/ComfyUI checkout e2ab36d933356bc8cd6ecb39c655fe8be75af4e5"
        ]
      }
    },
    {
      method: "shell.run",
      params: {
        conda: { path: "app/services/minimax_h3/env", python: "3.11" },
        message: [
          "uv pip install torch==2.10.0 torchvision==0.25.0 torchaudio==2.10.0 --index-url https://download.pytorch.org/whl/cu130",
          "uv pip install -r app/services/minimax_h3/vendor/ComfyUI/requirements.txt"
        ]
      }
    },
    {
      method: "fs.write",
      params: {
        path: "app/services/minimax_h3/env/.maestro_minimax_h3_v2.installed",
        text: "e2ab36d933356bc8cd6ecb39c655fe8be75af4e5 torch-2.10.0-cu130"
      }
    },
    // Fetch the seed-vc voice-conversion component (GPL-3.0). It lives in
    // its own repository and is cloned into place at install time instead
    // of being tracked in this repo, so the GPL-licensed tree keeps its own
    // license and distribution channel. Pinned to a tag for reproducible
    // installs — bump the tag here AND in update.js when shipping a new
    // component version.
    {
      when: "{{!exists('app/postprocessing/seedvc/__init__.py')}}",
      method: "shell.run",
      params: {
        message: "git clone --depth 1 --branch v1.0.0 https://github.com/Blizaine/maestro-seedvc app/postprocessing/seedvc"
      }
    },
    {
      when: "{{exists('ui/package.json')}}",
      method: "shell.run",
      params: {
        path: "ui",
        message: [
          "npm ci",
          "npm run build"
        ]
      }
    },
    // SAM 3.1 segmentation service (used by experimental Inpaint mode)
    // is intentionally NOT installed here. It adds ~5+ minutes to a
    // fresh install (separate Python 3.12 conda env, torch wheels,
    // SAM 3 source, etc.) but is only needed for the inpaint feature
    // which most users won't touch — and which is gated behind the
    // experimental flag in Settings → Services anyway. Users who want
    // it can run "Install Inpaint Support" from the Pinokio menu when
    // they're ready, which fires sam_install.js.
    {
      method: 'input',
      params: {
        title: 'Installation completed',
        description: 'Click "Start" to get started'
      }
    }
  ]
}