import { NodeComfyui } from "comfyui";

async function run() {
  const nodeComfyui = new NodeComfyui({
    serverAddress: "http://127.0.0.1:8188",
  });
  const nodeExecutor = await nodeComfyui.prepare();
  const pipeline = nodeExecutor.pipeline;

  // i am title
  const emptyLatentImage = pipeline.EmptyLatentImage({
    width: 512,
    height: 512,
    batch_size: 1,
  });

  const checkpointLoaderSimple = pipeline.CheckpointLoaderSimple({
    ckpt_name: "v1-5-pruned-emaonly.ckpt",
  });

  const clipTextEncode = pipeline.CLIPTextEncode({
    text: "beautiful scenery nature glass bottle landscape, , purple galaxy bottle,",
    clip: checkpointLoaderSimple.clip,
  });

  const clipTextEncode1 = pipeline.CLIPTextEncode({
    text: "text, watermark",
    clip: checkpointLoaderSimple.clip,
  });

  const kSampler = pipeline.KSampler({
    seed: 156680208700286,
    steps: 20,
    cfg: 8,
    sampler_name: "euler",
    scheduler: "normal",
    denoise: 1,
    model: checkpointLoaderSimple.model,
    positive: clipTextEncode.conditioning,
    negative: clipTextEncode1.conditioning,
    latent_image: emptyLatentImage.latent,
  });

  const vaeDecode = pipeline.VAEDecode({
    samples: kSampler.latent,
    vae: checkpointLoaderSimple.vae,
  });

  const saveImage = pipeline.SaveImage({
    filename_prefix: "ComfyUI",
    images: vaeDecode.image,
  });

  const results = await nodeExecutor.execute();
  console.log(results);

  nodeComfyui.close();
}

run().catch((reason) => console.error(reason));