{ pkgs, inputs, ... }:

{
  imports = [
    inputs.comfy-ui.nixosModules.default
  ];

  services = {
    comfyui = {
      enable = true;
      gpuSupport = "cuda";
      enableManager = true;
      port = 8188;
      listenAddress = "0.0.0.0";
      dataDir = "/var/lib/comfyui";
      user = "comfyui";
      group = "comfyui";
      createUser = true;
    };

    ollama = {
      enable = true;
      package = pkgs.unstable.ollama-cuda;
      host = "0.0.0.0";
      port = 11434;
      user = "ollama";
      group = "ollama";
      home = "/var/lib/ollama";
    };

    open-webui.enable = true;
  };

  environment.systemPackages = with pkgs; [
    comfy-ui-cuda
    unstable.ollama-cuda
  ];
}