// - models/checkpoints/sdxl_lightning_2step.safetensors

// - models/loras/sdxl-PaperCutouts-Dreambooth.safetensors

// - models/loras/SDXL_Yarn_Art_Style.safetensors

// to be available on the ComfyUI server

​

// See the following links for more information about

// the LoRAs used, and their training data

// - https://huggingface.co/Norod78/sdxl-PaperCutouts-Dreambooth

// - https://huggingface.co/Norod78/SDXL-YarnArtStyle-LoRA

​

let workflow;

let comfy;

let resImg;

let loraSelect;

​

function preload() {

  workflow = loadJSON("workflow_api.json");

}

​

function setup() {

  createCanvas(512, 512);

  pixelDensity(2); // SDXL operates on 1024x1024

​

  comfy = new ComfyUiP5Helper("https://gpu1.gohai.xyz:8188/");

  console.log("workflow is", workflow);

​

  loraSelect = createSelect();

  loraSelect.option("yarn");

  loraSelect.option("paper");

  loraSelect.changed(changeLora);

​

  let button = createButton("new seed");

  button.mousePressed(requestImage);

}

​

function requestImage() {

  workflow[3].inputs.seed = random(9999999);

  comfy.run(workflow, gotImage);

}

​

function changeLora() {

  if (loraSelect.value() == "paper") {

    workflow[6].inputs.text = workflow[6].inputs.text.replace(

      "Yarn art style",

      "PaperCutout style"

    );

    workflow[15].inputs.lora_name = "sdxl-PaperCutouts-Dreambooth.safetensors";

  } else {

    workflow[6].inputs.text = workflow[6].inputs.text.replace(

      "PaperCutout style",

      "Yarn art style"

    );

    workflow[15].inputs.lora_name = "SDXL_Yarn_Art_Style.safetensors";

  }

​

  comfy.run(workflow, gotImage);

}

​

function gotImage(data, err) {

  // data is an array of outputs from running the workflow

  console.log("gotImage", data);

​

  // we can load them like so

  if (data.length > 0) {

    resImg = loadImage(data[0].src);

  }

​

  // we could automatically run again if we wanted

  //requestImage();

}

​

function draw() {

  background(255);

  // if we have an image, put it onto the canvas

  if (resImg) {

    image(resImg, 0, 0, width, height);

  }

}

​