# Auto-generated using compose2nix v0.3.2-pre.
{ pkgs, lib, ... }:

{
  hardware.nvidia-container-toolkit.enable = true;
  # Runtime
  virtualisation.docker = {
    enable = true;
    # enableNvidia = true;
    enableOnBoot = true;
    autoPrune.enable = true;
    daemon.settings.features.cdi = true;
    rootless.enable = true;
    rootless.daemon.settings.features.cdi = true;
  };
  virtualisation.oci-containers.backend = "docker";

  # Containers
  virtualisation.oci-containers.containers."zonos_container" = {
    image = "compose2nix/zonos_container";
    environment = {
      "GRADIO_SHARE" = "False";
      "NVIDIA_VISIBLE_DEVICES" = "0";
    };
    cmd = [
      "python3"
      "gradio_interface.py"
    ];
    log-driver = "journald";
    extraOptions = [
      "--device=nvidia.com/gpu=0"
      "--network=host"
    ];
  };
  systemd.services."docker-zonos_container" = {
    serviceConfig = {
      Restart = lib.mkOverride 90 "no";
    };
    partOf = [
      "docker-compose-zonos-root.target"
    ];
    wantedBy = [
      "docker-compose-zonos-root.target"
    ];
  };

  # Builds
  systemd.services."docker-build-zonos_container" = {
    path = [
      pkgs.docker
      pkgs.git
    ];
    serviceConfig = {
      Type = "oneshot";
      TimeoutSec = 900;
    };
    script = ''
      cd /home/jordan/projects/jordan/Zonos
      docker build -t compose2nix/zonos_container .
    '';
  };

  # Root service
  # When started, this will automatically create all resources and start
  # the containers. When stopped, this will teardown all resources.
  systemd.targets."docker-compose-zonos-root" = {
    unitConfig = {
      Description = "Root target generated by compose2nix.";
    };
    wantedBy = [ "multi-user.target" ];
  };
}