dotfiles/modules/nixos/services/comfyui/default.nix

{
  config,
  pkgs,
  flake,
  ...
}:
let
  inherit (flake.config.machines.devices) ceres;
  inherit (flake.config.services.instances) comfyui web;
  inherit (flake.config.people) user0;
  service = comfyui;
  localhost = web.localhost.address1;
in
{

  services.comfyui = {
    enable = true;
    package = pkgs.comfyui-nvidia;
    host = localhost;
    user = service.name;
    acceleration = "cuda";
    models = builtins.attrValues {
      inherit (pkgs.nixified-ai)
        models
        ;
    };
    customNodes = builtins.attrValues {
      inherit (comfyui.pkgs)
        comfyui-gguf
        comfyui-impact-pack
        ;
    };
  };
  fileSystems."/var/lib/${service.name}" = {
    device = service.paths.path0;
    fsType = "none";
    options = [
      "bind"
    ];
    depends = [
      ceres.storage0.mount
    ];
  };

  systemd.tmpfiles.rules = [
    "Z ${service.paths.path0} 755 ${service.name} ${service.name} -"
  ];

  networking = {
    firewall = {
      allowedTCPPorts = [
        service.ports.port0
      ];
    };
  };
}