nodes/yolain-ComfyUI-Easy-Use
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
Files
ac7db29df99afd03336505dba6d020d2d285615a
yolain-ComfyUI-Easy-Use/web/js/easy/easyDynamicWidgets.js
T
yolain ac7db29df9 add:easy sv3dLoader
2024-03-31 18:19:16 +08:00

1071 lines
39 KiB
JavaScript
Raw Blame History

import { app } from "/scripts/app.js";
import { api } from "/scripts/api.js";
import { ComfyWidgets } from "/scripts/widgets.js";
let origProps = {};
const seedNodes = ["easy seed", "easy latentNoisy", "easy wildcards", "easy preSampling", "easy preSamplingAdvanced", "easy preSamplingNoiseIn", "easy preSamplingSdTurbo", "easy preSamplingCascade", "easy preSamplingDynamicCFG", "easy preSamplingLayerDiffusion", "easy fullkSampler", "easy fullCascadeKSampler"]
const loaderNodes = ["easy fullLoader", "easy a1111Loader", "easy comfyLoader"]
const findWidgetByName = (node, name) => node.widgets.find((w) => w.name === name);
const doesInputWithNameExist = (node, name) => node.inputs ? node.inputs.some((input) => input.name === name) : false;
function updateNodeHeight(node) {node.setSize([node.size[0], node.computeSize()[1]]);}
function toggleWidget(node, widget, show = false, suffix = "") {
if (!widget || doesInputWithNameExist(node, widget.name)) return;
if (!origProps[widget.name]) {
origProps[widget.name] = { origType: widget.type, origComputeSize: widget.computeSize };
}
const origSize = node.size;
widget.type = show ? origProps[widget.name].origType : "easyHidden" + suffix;
widget.computeSize = show ? origProps[widget.name].origComputeSize : () => [0, -4];
widget.linkedWidgets?.forEach(w => toggleWidget(node, w, ":" + widget.name, show));
const height = show ? Math.max(node.computeSize()[1], origSize[1]) : node.size[1];
node.setSize([node.size[0], height]);
}
function toggleInput(node, name, show = false) {
if(!show){
}
}
function widgetLogic(node, widget) {
if (widget.name === 'lora_name') {
if (widget.value === "None") {
toggleWidget(node, findWidgetByName(node, 'lora_model_strength'))
toggleWidget(node, findWidgetByName(node, 'lora_clip_strength'))
} else {
toggleWidget(node, findWidgetByName(node, 'lora_model_strength'), true)
toggleWidget(node, findWidgetByName(node, 'lora_clip_strength'), true)
}
}
if (widget.name === 'rescale') {
let rescale_after_model = findWidgetByName(node, 'rescale_after_model').value
if (widget.value === 'by percentage' && rescale_after_model) {
toggleWidget(node, findWidgetByName(node, 'width'))
toggleWidget(node, findWidgetByName(node, 'height'))
toggleWidget(node, findWidgetByName(node, 'longer_side'))
toggleWidget(node, findWidgetByName(node, 'percent'), true)
} else if (widget.value === 'to Width/Height' && rescale_after_model) {
toggleWidget(node, findWidgetByName(node, 'width'), true)
toggleWidget(node, findWidgetByName(node, 'height'), true)
toggleWidget(node, findWidgetByName(node, 'percent'))
toggleWidget(node, findWidgetByName(node, 'longer_side'))
} else if (rescale_after_model) {
toggleWidget(node, findWidgetByName(node, 'longer_side'), true)
toggleWidget(node, findWidgetByName(node, 'width'))
toggleWidget(node, findWidgetByName(node, 'height'))
toggleWidget(node, findWidgetByName(node, 'percent'))
}
updateNodeHeight(node)
}
if (widget.name === 'upscale_method') {
if (widget.value === "None") {
toggleWidget(node, findWidgetByName(node, 'factor'))
toggleWidget(node, findWidgetByName(node, 'crop'))
} else {
toggleWidget(node, findWidgetByName(node, 'factor'), true)
toggleWidget(node, findWidgetByName(node, 'crop'), true)
}
updateNodeHeight(node)
}
if (widget.name === 'image_output') {
if (widget.value === 'Sender' || widget.value === 'Sender/Save'){
toggleWidget(node, findWidgetByName(node, 'link_id'), true)
}else {
toggleWidget(node, findWidgetByName(node, 'link_id'))
}
if (widget.value === 'Hide' || widget.value === 'Preview' || widget.value === 'Sender') {
toggleWidget(node, findWidgetByName(node, 'save_prefix'))
toggleWidget(node, findWidgetByName(node, 'output_path'))
toggleWidget(node, findWidgetByName(node, 'embed_workflow'))
toggleWidget(node, findWidgetByName(node, 'number_padding'))
toggleWidget(node, findWidgetByName(node, 'overwrite_existing'))
} else if (widget.value === 'Save' || widget.value === 'Hide/Save' || widget.value === 'Sender/Save') {
toggleWidget(node, findWidgetByName(node, 'save_prefix'), true)
toggleWidget(node, findWidgetByName(node, 'output_path'), true)
toggleWidget(node, findWidgetByName(node, 'embed_workflow'), true)
toggleWidget(node, findWidgetByName(node, 'number_padding'), true)
toggleWidget(node, findWidgetByName(node, 'overwrite_existing'), true)
}
if(widget.value === 'Hide' || widget.value === 'Hide/Save'){
toggleWidget(node, findWidgetByName(node, 'decode_vae_name'))
}else{
toggleWidget(node, findWidgetByName(node, 'decode_vae_name'), true)
}
}
if (widget.name === 'add_noise') {
if (widget.value === "disable") {
toggleWidget(node, findWidgetByName(node, 'seed'))
toggleWidget(node, findWidgetByName(node, 'control_before_generate'))
toggleWidget(node, findWidgetByName(node, 'control_after_generate'))
} else {
toggleWidget(node, findWidgetByName(node, 'seed'), true)
toggleWidget(node, findWidgetByName(node, 'control_before_generate'), true)
toggleWidget(node, findWidgetByName(node, 'control_after_generate'), true)
}
updateNodeHeight(node)
}
if (widget.name === 'num_loras') {
let number_to_show = widget.value + 1
for (let i = 0; i < number_to_show; i++) {
toggleWidget(node, findWidgetByName(node, 'lora_'+i+'_name'), true)
if (findWidgetByName(node, 'mode').value === "simple") {
toggleWidget(node, findWidgetByName(node, 'lora_'+i+'_strength'), true)
toggleWidget(node, findWidgetByName(node, 'lora_'+i+'_model_strength'))
toggleWidget(node, findWidgetByName(node, 'lora_'+i+'_clip_strength'))
} else {
toggleWidget(node, findWidgetByName(node, 'lora_'+i+'_strength'))
toggleWidget(node, findWidgetByName(node, 'lora_'+i+'_model_strength'), true)
toggleWidget(node, findWidgetByName(node, 'lora_'+i+'_clip_strength'), true)
}
}
for (let i = number_to_show; i < 21; i++) {
toggleWidget(node, findWidgetByName(node, 'lora_'+i+'_name'))
toggleWidget(node, findWidgetByName(node, 'lora_'+i+'_strength'))
toggleWidget(node, findWidgetByName(node, 'lora_'+i+'_model_strength'))
toggleWidget(node, findWidgetByName(node, 'lora_'+i+'_clip_strength'))
}
updateNodeHeight(node)
}
if (widget.name === 'mode') {
let number_to_show = findWidgetByName(node, 'num_loras').value + 1
for (let i = 0; i < number_to_show; i++) {
if (widget.value === "simple") {
toggleWidget(node, findWidgetByName(node, 'lora_'+i+'_strength'), true)
toggleWidget(node, findWidgetByName(node, 'lora_'+i+'_model_strength'))
toggleWidget(node, findWidgetByName(node, 'lora_'+i+'_clip_strength'))
} else {
toggleWidget(node, findWidgetByName(node, 'lora_'+i+'_strength'))
toggleWidget(node, findWidgetByName(node, 'lora_'+i+'_model_strength'), true)
toggleWidget(node, findWidgetByName(node, 'lora_'+i+'_clip_strength'), true)}
}
updateNodeHeight(node)
}
if (widget.name === 'resolution') {
if (widget.value === "自定义 x 自定义") {
toggleWidget(node, findWidgetByName(node, 'empty_latent_width'), true)
toggleWidget(node, findWidgetByName(node, 'empty_latent_height'), true)
} else {
toggleWidget(node, findWidgetByName(node, 'empty_latent_width'), false)
toggleWidget(node, findWidgetByName(node, 'empty_latent_height'), false)
}
updateNodeHeight(node)
}
if (widget.name === 'downscale_mode') {
const widget_names = ['block_number', 'downscale_factor', 'start_percent', 'end_percent', 'downscale_after_skip', 'downscale_method', 'upscale_method']
if (widget.value === "None") widget_names.map(name=> toggleWidget(node, findWidgetByName(node, name)))
else if(widget.value == 'Auto') widget_names.map(name =>toggleWidget(node, findWidgetByName(node, name),name == 'block_number' ? true : false))
else widget_names.map(name=> toggleWidget(node, findWidgetByName(node, name), true))
updateNodeHeight(node)
}
if (widget.name == 'range_mode'){
if(widget.value == 'step'){
toggleWidget(node, findWidgetByName(node, 'step'), true)
toggleWidget(node, findWidgetByName(node, 'num_steps'))
}else if(widget.value == 'num_steps'){
toggleWidget(node, findWidgetByName(node, 'step'))
toggleWidget(node, findWidgetByName(node, 'num_steps'), true)
}
updateNodeHeight(node)
}
if (widget.name === 'toggle') {
widget.type = 'toggle'
widget.options = {on: 'Enabled', off: 'Disabled'}
}
if(widget.name == 'text_combine_mode'){
if(widget.value == 'replace'){
toggleWidget(node, findWidgetByName(node, 'replace_text'), true)
}else{
toggleWidget(node, findWidgetByName(node, 'replace_text'))
}
updateNodeHeight(node)
}
if (widget.name === 'conditioning_mode') {
if (["replace", "concat", "combine"].includes(widget.value)) {
toggleWidget(node, findWidgetByName(node, 'average_strength'))
toggleWidget(node, findWidgetByName(node, 'old_cond_start'))
toggleWidget(node, findWidgetByName(node, 'old_cond_end'))
toggleWidget(node, findWidgetByName(node, 'new_cond_start'))
toggleWidget(node, findWidgetByName(node, 'new_cond_end'))
} else if(widget.value == 'average'){
toggleWidget(node, findWidgetByName(node, 'average_strength'), true)
toggleWidget(node, findWidgetByName(node, 'old_cond_start'))
toggleWidget(node, findWidgetByName(node, 'old_cond_end'))
toggleWidget(node, findWidgetByName(node, 'new_cond_start'))
toggleWidget(node, findWidgetByName(node, 'new_cond_end'))
}else if(widget.value == 'timestep'){
toggleWidget(node, findWidgetByName(node, 'average_strength'))
toggleWidget(node, findWidgetByName(node, 'old_cond_start'), true)
toggleWidget(node, findWidgetByName(node, 'old_cond_end'), true)
toggleWidget(node, findWidgetByName(node, 'new_cond_start'), true)
toggleWidget(node, findWidgetByName(node, 'new_cond_end'), true)
}
}
if (widget.name === 'preset') {
const normol_presets = [
'LIGHT - SD1.5 only (low strength)',
'STANDARD (medium strength)',
'VIT-G (medium strength)',
'PLUS (high strength)', 'PLUS FACE (portraits)',
'FULL FACE - SD1.5 only (portraits stronger)',
]
const faceid_presets = [
'FACEID',
'FACEID PLUS - SD1.5 only',
'FACEID PLUS V2',
'FACEID PORTRAIT (style transfer)'
]
if(normol_presets.includes(widget.value)){
toggleWidget(node, findWidgetByName(node, 'lora_strength'))
toggleWidget(node, findWidgetByName(node, 'provider'))
toggleWidget(node, findWidgetByName(node, 'weight_faceidv2'))
toggleWidget(node, findWidgetByName(node, 'use_tiled'), true)
let use_tiled = findWidgetByName(node, 'use_tiled')
if(use_tiled && use_tiled.value){
toggleWidget(node, findWidgetByName(node, 'sharpening'), true)
}else {
toggleWidget(node, findWidgetByName(node, 'sharpening'))
}
}
else if(faceid_presets.includes(widget.value)){
if(widget.value == 'FACEID PLUS V2'){
toggleWidget(node, findWidgetByName(node, 'weight_faceidv2'), true)
}else{
toggleWidget(node, findWidgetByName(node, 'weight_faceidv2'))
}
if(widget.value == 'FACEID PORTRAIT (style transfer)'){
toggleWidget(node, findWidgetByName(node, 'lora_strength'), false)
}
else{
toggleWidget(node, findWidgetByName(node, 'lora_strength'), true)
}
toggleWidget(node, findWidgetByName(node, 'provider'), true)
toggleWidget(node, findWidgetByName(node, 'use_tiled'))
toggleWidget(node, findWidgetByName(node, 'sharpening'))
}
updateNodeHeight(node)
}
if (widget.name === 'use_tiled') {
if(widget.value)
toggleWidget(node, findWidgetByName(node, 'sharpening'), true)
else
toggleWidget(node, findWidgetByName(node, 'sharpening'))
updateNodeHeight(node)
}
if (widget.name === 'num_embeds') {
let number_to_show = widget.value + 1
for (let i = 0; i < number_to_show; i++) {
toggleInput(node, 'image'+i, true)
toggleInput(node, 'mask'+i, true)
toggleWidget(node, findWidgetByName(node, 'weight'+i), true)
}
for (let i = number_to_show; i < 6; i++) {
toggleInput(node, 'image'+i)
toggleInput(node, 'mask'+i)
toggleWidget(node, findWidgetByName(node, 'weight'+i))
}
updateNodeHeight(node)
}
}
function widgetLogic2(node, widget) {
if (widget.name === 'sampler_name') {
const widget_names = ['eta','s_noise','upscale_ratio','start_step','end_step','upscale_n_step','unsharp_kernel_size','unsharp_sigma','unsharp_strength']
if (["euler_ancestral", "dpmpp_2s_ancestral", "dpmpp_2m_sde", "lcm"].includes(widget.value)) {
widget_names.map(name=> toggleWidget(node, findWidgetByName(node, name)), true)
} else {
widget_names.map(name=> toggleWidget(node, findWidgetByName(node, name)))
}
updateNodeHeight(node)
}
}
function widgetLogic3(node, widget){
if (widget.name === 'target_parameter') {
if (node.comfyClass == 'easy XYInputs: Steps'){
switch (widget.value){
case "steps":
toggleWidget(node, findWidgetByName(node, 'first_step'), true)
toggleWidget(node, findWidgetByName(node, 'last_step'), true)
toggleWidget(node, findWidgetByName(node, 'first_start_step'))
toggleWidget(node, findWidgetByName(node, 'last_start_step'))
toggleWidget(node, findWidgetByName(node, 'first_end_step'))
toggleWidget(node, findWidgetByName(node, 'last_end_step'))
break
case "start_at_step":
toggleWidget(node, findWidgetByName(node, 'first_step'))
toggleWidget(node, findWidgetByName(node, 'last_step'))
toggleWidget(node, findWidgetByName(node, 'first_start_step'), true)
toggleWidget(node, findWidgetByName(node, 'last_start_step'), true)
toggleWidget(node, findWidgetByName(node, 'first_end_step'))
toggleWidget(node, findWidgetByName(node, 'last_end_step'))
break
case "end_at_step":
toggleWidget(node, findWidgetByName(node, 'first_step'))
toggleWidget(node, findWidgetByName(node, 'last_step'))
toggleWidget(node, findWidgetByName(node, 'first_start_step'))
toggleWidget(node, findWidgetByName(node, 'last_start_step'))
toggleWidget(node, findWidgetByName(node, 'first_end_step'),true)
toggleWidget(node, findWidgetByName(node, 'last_end_step'),true)
break
}
}
if (node.comfyClass == 'easy XYInputs: Sampler/Scheduler'){
let number_to_show = findWidgetByName(node, 'input_count').value + 1
for (let i = 0; i < number_to_show; i++) {
switch (widget.value) {
case "sampler":
toggleWidget(node, findWidgetByName(node, 'sampler_'+i), true)
toggleWidget(node, findWidgetByName(node, 'scheduler_'+i))
break
case "scheduler":
toggleWidget(node, findWidgetByName(node, 'scheduler_'+i), true)
toggleWidget(node, findWidgetByName(node, 'sampler_'+i))
break
default:
toggleWidget(node, findWidgetByName(node, 'sampler_'+i), true)
toggleWidget(node, findWidgetByName(node, 'scheduler_'+i), true)
break
}
}
updateNodeHeight(node)
}
if (node.comfyClass == 'easy XYInputs: ControlNet'){
switch (widget.value){
case "strength":
toggleWidget(node, findWidgetByName(node, 'first_strength'), true)
toggleWidget(node, findWidgetByName(node, 'last_strength'), true)
toggleWidget(node, findWidgetByName(node, 'strength'))
toggleWidget(node, findWidgetByName(node, 'start_percent'), true)
toggleWidget(node, findWidgetByName(node, 'end_percent'), true)
toggleWidget(node, findWidgetByName(node, 'first_start_percent'))
toggleWidget(node, findWidgetByName(node, 'last_start_percent'))
toggleWidget(node, findWidgetByName(node, 'first_end_percent'))
toggleWidget(node, findWidgetByName(node, 'last_end_percent'))
break
case "start_percent":
toggleWidget(node, findWidgetByName(node, 'first_strength'))
toggleWidget(node, findWidgetByName(node, 'last_strength'))
toggleWidget(node, findWidgetByName(node, 'strength'), true)
toggleWidget(node, findWidgetByName(node, 'start_percent'))
toggleWidget(node, findWidgetByName(node, 'end_percent'), true)
toggleWidget(node, findWidgetByName(node, 'first_start_percent'), true)
toggleWidget(node, findWidgetByName(node, 'last_start_percent'), true)
toggleWidget(node, findWidgetByName(node, 'first_end_percent'))
toggleWidget(node, findWidgetByName(node, 'last_end_percent'))
break
case "end_percent":
toggleWidget(node, findWidgetByName(node, 'first_strength'))
toggleWidget(node, findWidgetByName(node, 'last_strength'))
toggleWidget(node, findWidgetByName(node, 'strength'), true)
toggleWidget(node, findWidgetByName(node, 'start_percent'), true)
toggleWidget(node, findWidgetByName(node, 'end_percent'))
toggleWidget(node, findWidgetByName(node, 'first_start_percent'))
toggleWidget(node, findWidgetByName(node, 'last_start_percent'))
toggleWidget(node, findWidgetByName(node, 'first_end_percent'), true)
toggleWidget(node, findWidgetByName(node, 'last_end_percent'), true)
break
}
updateNodeHeight(node)
}
}
if (node.comfyClass == 'easy XYInputs: PromptSR'){
let number_to_show = findWidgetByName(node, 'replace_count').value + 1
for (let i = 0; i < number_to_show; i++) {
toggleWidget(node, findWidgetByName(node, 'replace_'+i), true)
}
for (let i = number_to_show; i < 31; i++) {
toggleWidget(node, findWidgetByName(node, 'replace_'+i))
}
updateNodeHeight(node)
}
if(widget.name == 'input_count'){
let number_to_show = widget.value + 1
for (let i = 0; i < number_to_show; i++) {
if (findWidgetByName(node, 'target_parameter').value === "sampler") {
toggleWidget(node, findWidgetByName(node, 'sampler_'+i), true)
toggleWidget(node, findWidgetByName(node, 'scheduler_'+i))
}
else if (findWidgetByName(node, 'target_parameter').value === "scheduler") {
toggleWidget(node, findWidgetByName(node, 'scheduler_'+i), true)
toggleWidget(node, findWidgetByName(node, 'sampler_'+i))
} else {
toggleWidget(node, findWidgetByName(node, 'sampler_'+i), true)
toggleWidget(node, findWidgetByName(node, 'scheduler_'+i), true)
}
}
for (let i = number_to_show; i < 31; i++) {
toggleWidget(node, findWidgetByName(node, 'sampler_'+i))
toggleWidget(node, findWidgetByName(node, 'scheduler_'+i))
}
updateNodeHeight(node)
}
if (widget.name === 'lora_count') {
let number_to_show = widget.value + 1
const isWeight = findWidgetByName(node, 'input_mode').value.indexOf("Weights") == -1
for (let i = 0; i < number_to_show; i++) {
toggleWidget(node, findWidgetByName(node, 'lora_name_'+i), true)
if (isWeight) {
toggleWidget(node, findWidgetByName(node, 'lora_name_'+i), true)
toggleWidget(node, findWidgetByName(node, 'model_str_'+i))
toggleWidget(node, findWidgetByName(node, 'clip_str_'+i))
} else {
toggleWidget(node, findWidgetByName(node, 'lora_name_'+i), true)
toggleWidget(node, findWidgetByName(node, 'model_str_'+i),true)
toggleWidget(node, findWidgetByName(node, 'clip_str_'+i), true)
}
}
for (let i = number_to_show; i < 11; i++) {
toggleWidget(node, findWidgetByName(node, 'lora_name_'+i))
toggleWidget(node, findWidgetByName(node, 'model_str_'+i))
toggleWidget(node, findWidgetByName(node, 'clip_str_'+i))
}
updateNodeHeight(node)
}
if (widget.name === 'ckpt_count') {
let number_to_show = widget.value + 1
const hasClipSkip = findWidgetByName(node, 'input_mode').value.indexOf("ClipSkip") != -1
const hasVae = findWidgetByName(node, 'input_mode').value.indexOf("VAE") != -1
for (let i = 0; i < number_to_show; i++) {
toggleWidget(node, findWidgetByName(node, 'ckpt_name_'+i), true)
if (hasClipSkip && hasVae) {
toggleWidget(node, findWidgetByName(node, 'clip_skip_'+i), true)
toggleWidget(node, findWidgetByName(node, 'vae_name_'+i), true)
} else if (hasVae){
toggleWidget(node, findWidgetByName(node, 'clip_skip_' + i))
toggleWidget(node, findWidgetByName(node, 'vae_name_' + i), true)
}else{
toggleWidget(node, findWidgetByName(node, 'clip_skip_' + i))
toggleWidget(node, findWidgetByName(node, 'vae_name_' + i))
}
}
for (let i = number_to_show; i < 11; i++) {
toggleWidget(node, findWidgetByName(node, 'ckpt_name_'+i))
toggleWidget(node, findWidgetByName(node, 'clip_skip_'+i))
toggleWidget(node, findWidgetByName(node, 'vae_name_'+i))
}
updateNodeHeight(node)
}
if (widget.name === 'input_mode') {
if(node.comfyClass == 'easy XYInputs: Lora'){
let number_to_show = findWidgetByName(node, 'lora_count').value + 1
const hasWeight = widget.value.indexOf("Weights") != -1
for (let i = 0; i < number_to_show; i++) {
toggleWidget(node, findWidgetByName(node, 'lora_name_'+i), true)
if (hasWeight) {
toggleWidget(node, findWidgetByName(node, 'model_str_'+i), true)
toggleWidget(node, findWidgetByName(node, 'clip_str_'+i), true)
} else {
toggleWidget(node, findWidgetByName(node, 'model_str_' + i))
toggleWidget(node, findWidgetByName(node, 'clip_str_' + i))
}
}
if(hasWeight){
toggleWidget(node, findWidgetByName(node, 'model_strength'))
toggleWidget(node, findWidgetByName(node, 'clip_strength'))
}else{
toggleWidget(node, findWidgetByName(node, 'model_strength'), true)
toggleWidget(node, findWidgetByName(node, 'clip_strength'),true)
}
}
else if(node.comfyClass == 'easy XYInputs: Checkpoint'){
let number_to_show = findWidgetByName(node, 'ckpt_count').value + 1
const hasClipSkip = widget.value.indexOf("ClipSkip") != -1
const hasVae = widget.value.indexOf("VAE") != -1
for (let i = 0; i < number_to_show; i++) {
toggleWidget(node, findWidgetByName(node, 'ckpt_name_'+i), true)
if (hasClipSkip && hasVae) {
toggleWidget(node, findWidgetByName(node, 'clip_skip_'+i), true)
toggleWidget(node, findWidgetByName(node, 'vae_name_'+i), true)
} else if (hasClipSkip){
toggleWidget(node, findWidgetByName(node, 'clip_skip_' + i), true)
toggleWidget(node, findWidgetByName(node, 'vae_name_' + i))
}else{
toggleWidget(node, findWidgetByName(node, 'clip_skip_' + i))
toggleWidget(node, findWidgetByName(node, 'vae_name_' + i))
}
}
}
updateNodeHeight(node)
}
// if(widget.name == 'replace_count'){
// let number_to_show = widget.value + 1
// for (let i = 0; i < number_to_show; i++) {
// toggleWidget(node, findWidgetByName(node, 'replace_'+i), true)
// }
// for (let i = number_to_show; i < 31; i++) {
// toggleWidget(node, findWidgetByName(node, 'replace_'+i))
// }
// updateNodeHeight(node)
// }
}
app.registerExtension({
name: "comfy.easyUse.dynamicWidgets",
nodeCreated(node) {
switch (node.comfyClass){
case "easy fullLoader":
case "easy a1111Loader":
case "easy comfyLoader":
case "easy cascadeLoader":
case "easy svdLoader":
case "easy dynamiCrafterLoader":
case "easy loraStack":
case "easy latentNoisy":
case "easy preSamplingAdvanced":
case "easy preSamplingNoiseIn":
case "easy preSamplingSdTurbo":
case "easy preSamplingCascade":
case "easy preSamplingLayerDiffusion":
case "easy fullkSampler":
case "easy kSampler":
case "easy kSamplerSDTurbo":
case "easy kSamplerTiled":
case "easy kSamplerLayerDiffusion":
case "easy kSamplerInpainting":
case "easy kSamplerDownscaleUnet":
case "easy fullCascadeKSampler":
case "easy cascadeKSampler":
case "easy hiresFix":
case "easy detailerFix":
case "easy imageRemBg":
case "easy XYInputs: Steps":
case "easy XYInputs: Sampler/Scheduler":
case 'easy XYInputs: Checkpoint':
case "easy XYInputs: Lora":
case "easy XYInputs: PromptSR":
case "easy XYInputs: ControlNet":
case "easy rangeInt":
case "easy rangeFloat":
case 'easy latentCompositeMaskedWithCond':
case 'easy pipeEdit':
case 'easy ipadapterApply':
case 'easy ipadapterApplyADV':
case 'easy ipadapterApplyEncoder':
getSetters(node)
break
case "easy wildcards":
const wildcard_text_widget_index = node.widgets.findIndex((w) => w.name == 'text');
const wildcard_text_widget = node.widgets[wildcard_text_widget_index];
// lora selector, wildcard selector
let combo_id = 1;
Object.defineProperty(node.widgets[combo_id], "value", {
set: (value) => {
const stackTrace = new Error().stack;
if(stackTrace.includes('inner_value_change')) {
if(value != "Select the LoRA to add to the text") {
let lora_name = value;
if (lora_name.endsWith('.safetensors')) {
lora_name = lora_name.slice(0, -12);
}
wildcard_text_widget.value += `<lora:${lora_name}>`;
}
}
},
get: () => { return "Select the LoRA to add to the text"; }
});
Object.defineProperty(node.widgets[combo_id+1], "value", {
set: (value) => {
const stackTrace = new Error().stack;
if(stackTrace.includes('inner_value_change')) {
if(value != "Select the Wildcard to add to the text") {
if(wildcard_text_widget.value != '')
wildcard_text_widget.value += ', '
wildcard_text_widget.value += value;
}
}
},
get: () => { return "Select the Wildcard to add to the text"; }
});
// Preventing validation errors from occurring in any situation.
node.widgets[combo_id].serializeValue = () => { return "Select the LoRA to add to the text"; }
node.widgets[combo_id+1].serializeValue = () => { return "Select the Wildcard to add to the text"; }
break
case "easy detailerFix":
const textarea_widget_index = node.widgets.findIndex((w) => w.type === "customtext");
if(textarea_widget_index == -1) return
node.widgets[textarea_widget_index].dynamicPrompts = false
node.widgets[textarea_widget_index].inputEl.placeholder = "wildcard spec: if kept empty, this option will be ignored";
node.widgets[textarea_widget_index].serializeValue = () => {return node.widgets[textarea_widget_index].value};
break
case "easy XYInputs: ModelMergeBlocks":
let preset_i = 3;
let vector_i = 4;
let file_i = 5;
node._value = "Preset";
let valuesWidget = node.widgets[vector_i]
Object.defineProperty(node.widgets[preset_i], "value", {
set: (value) => {
const stackTrace = new Error().stack;
if(stackTrace.includes('inner_value_change')) {
if(value != "Preset") {
if(!value.startsWith('@') && valuesWidget.value)
valuesWidget.value += "\n";
if(value.startsWith('@')) {
let spec = value.split(':')[1];
var n;
var sub_n = null;
var block = null;
if(isNaN(spec)) {
let sub_spec = spec.split(',');
if(sub_spec.length != 3) {
valuesWidget = '!! SPEC ERROR !!';
node._value = '';
return;
}
n = parseInt(sub_spec[0].trim());
sub_n = parseInt(sub_spec[1].trim());
block = parseInt(sub_spec[2].trim());
}
else {
n = parseInt(spec.trim());
}
valuesWidget.value = "";
if(sub_n == null) {
for(let i=1; i<=n; i++) {
var temp = "1,1";
for(let j=1; j<=n; j++) {
if(temp!='')
temp += ',';
if(j==i)
temp += '1';
else
temp += '0';
}
temp += ',1; ';
valuesWidget.value += `B${i}:${temp}\n`;
}
}
else {
for(let i=1; i<=sub_n; i++) {
var temp = "";
for(let j=1; j<=n; j++) {
if(temp!='')
temp += ',';
if(block!=j)
temp += '0';
else {
temp += ' ';
for(let k=1; k<=sub_n; k++) {
if(k==i)
temp += '1 ';
else
temp += '0 ';
}
}
}
valuesWidget.value += `B${block}.SUB${i}:${temp}\n`;
}
}
}
else {
valuesWidget.value += `${value}; `;
}
// if(node.widgets_values) {
// valuesWidget.value = node.widgets[preset_i].value+ `; `;
// }
}
}
node._value = value;
},
get: () => {
return node._value;
}
});
const cb = node.callback;
valuesWidget.callback = function () {
if (cb) {
return cb.apply(this, arguments);
}
};
// upload .csv
async function uploadFile(file) {
try {
const body = new FormData();
body.append("csv", file);
const resp = await api.fetchApi("/easyuse/upload/csv", {
method: "POST",
body,
});
if (resp.status === 200) {
const data = await resp.json();
node.widgets[vector_i].value = data
} else {
alert(resp.status + " - " + resp.statusText);
}
} catch (error) {
alert(error);
}
}
const fileInput = document.createElement("input");
Object.assign(fileInput, {
type: "file",
accept: "text/csv",
style: "show: none",
onchange: async (event) => {
if (fileInput.files.length) {
await uploadFile(fileInput.files[0], true);
event.target.value = ''
}
},
});
document.body.append(fileInput);
const name = "choose .csv file into values"
let uploadWidget = node.addWidget("button", name, "csv", () => {
fileInput.click();
});
uploadWidget.label = name;
uploadWidget.serialize = false;
break
}
},
async beforeRegisterNodeDef(nodeType, nodeData, app) {
function addText(arr_text) {
var text = '';
for (let i = 0; i < arr_text.length; i++) {
text += arr_text[i];
}
return text
}
if (["easy showSpentTime"].includes(nodeData.name)) {
function populate(text) {
if (this.widgets) {
const pos = this.widgets.findIndex((w) => w.name === "spent_time");
if (pos !== -1 && this.widgets[pos]) {
const w = this.widgets[pos]
w.value = text;
}
}
}
// When the node is executed we will be sent the input text, show this in the widget
const onExecuted = nodeType.prototype.onExecuted;
nodeType.prototype.onExecuted = function (message) {
onExecuted?.apply(this, arguments);
const text = addText(message.text)
populate.call(this, text);
};
}
if (["easy showLoaderSettingsNames"].includes(nodeData.name)) {
function populate(text) {
if (this.widgets) {
const pos = this.widgets.findIndex((w) => w.name === "names");
if (pos !== -1 && this.widgets[pos]) {
const w = this.widgets[pos]
w.value = text;
}
}
}
// When the node is executed we will be sent the input text, show this in the widget
const onExecuted = nodeType.prototype.onExecuted;
nodeType.prototype.onExecuted = function (message) {
onExecuted?.apply(this, arguments);
const text = addText(message.text)
populate.call(this, text);
};
}
if (loaderNodes.includes(nodeData.name)) {
function populate(text, type = 'positive') {
if (this.widgets) {
const pos = this.widgets.findIndex((w) => w.name === type + "_prompt");
const className = "comfy-multiline-input wildcard_" + type + '_' + this.id.toString()
if (pos == -1 && text) {
const inputEl = document.createElement("textarea");
inputEl.className = className;
inputEl.placeholder = "Wildcard Prompt (" + type + ")"
const widget = this.addDOMWidget(type + "_prompt", "customtext", inputEl, {
getValue() {
return inputEl.value;
},
setValue(v) {
inputEl.value = v;
},
serialize: false,
});
widget.inputEl = inputEl;
widget.inputEl.readOnly = true
inputEl.addEventListener("input", () => {
widget.callback?.(widget.value);
});
widget.value = text;
} else if (this.widgets[pos]) {
if (text) {
const w = this.widgets[pos]
w.value = text;
} else {
this.widgets.splice(pos, 1);
const element = document.getElementsByClassName(className)
if (element && element[0]) element[0].remove()
}
}
}
}
const onExecuted = nodeType.prototype.onExecuted;
nodeType.prototype.onExecuted = function (message) {
onExecuted?.apply(this, arguments);
const positive = addText(message.positive)
const negative = addText(message.negative)
populate.call(this, positive, "positive");
populate.call(this, negative, "negative");
};
}
if(["easy sv3dLoader"].includes(nodeData.name)){
function changeSchedulerText(mode, batch_size, inputEl) {
console.log(mode)
switch (mode){
case 'azimuth':
inputEl.readOnly = true
inputEl.style.opacity = 0.6
return `0:(0.0,0.0)` + (batch_size > 1 ? `\n${batch_size-1}:(360.0,0.0)` : '')
case 'elevation':
inputEl.readOnly = true
inputEl.style.opacity = 0.6
return `0:(-90.0,0.0)` + (batch_size > 1 ? `\n${batch_size-1}:(90.0,0.0)` : '')
case 'custom':
inputEl.readOnly = false
inputEl.style.opacity = 1
return `0:(0.0,0.0)\n9:(180.0,0.0)\n20:(360.0,0.0)`
}
}
const onNodeCreated = nodeType.prototype.onNodeCreated;
nodeType.prototype.onNodeCreated = async function () {
onNodeCreated ? onNodeCreated.apply(this, []) : undefined;
const easing_mode_widget = this.widgets.find(w => w.name == 'easing_mode')
const batch_size = this.widgets.find(w => w.name == 'batch_size')
const scheduler = this.widgets.find(w => w.name == 'scheduler')
setTimeout(_=>{
if(!scheduler.value) scheduler.value = changeSchedulerText(easing_mode_widget.value, batch_size.value, scheduler.inputEl)
},1)
easing_mode_widget.callback = value=>{
scheduler.value = changeSchedulerText(value, batch_size.value, scheduler.inputEl)
}
batch_size.callback = value =>{
scheduler.value = changeSchedulerText(easing_mode_widget.value, value, scheduler.inputEl)
}
}
}
if (seedNodes.includes(nodeData.name)) {
const onNodeCreated = nodeType.prototype.onNodeCreated;
nodeType.prototype.onNodeCreated = async function () {
onNodeCreated ? onNodeCreated.apply(this, []) : undefined;
// const values = ["randomize", "fixed", "increment", "decrement"]
// const seed_widget = this.widgets.find(w => w.name == 'seed_num')
// const seed_control = this.addWidget("combo", "control_before_generate", values[0], () => {
// }, {
// values,
// serialize: false
// })
// seed_widget.linkedWidgets = [seed_control]
const seed_widget = this.widgets.find(w => ['seed_num','seed'].includes(w.name))
const seed_control = this.widgets.find(w=> ['control_before_generate','control_after_generate'].includes(w.name))
if(nodeData.name == 'easy seed'){
this.addWidget("button", "🎲 Manual Random Seed", null, _=>{
if(seed_control.value != 'fixed'){
seed_control.value = 'fixed'
}
seed_widget.value = Math.floor(Math.random() * 1125899906842624)
})
}
}
const onAdded = nodeType.prototype.onAdded;
nodeType.prototype.onAdded = async function () {
onAdded ? onAdded.apply(this, []) : undefined;
const seed_widget = this.widgets.find(w => ['seed_num','seed'].includes(w.name))
const seed_control = this.widgets.find(w=> ['control_before_generate','control_after_generate'].includes(w.name))
if(seed_control.name == 'control_before_generate' && seed_widget.value === 0){
seed_widget.value = Math.floor(Math.random() * 1125899906842624)
}
}
}
if (nodeData.name == 'easy imageInsetCrop') {
function setWidgetStep(a) {
const measurementWidget = a.widgets[0]
for (let i = 1; i <= 4; i++) {
if (measurementWidget.value === 'Pixels') {
a.widgets[i].options.step = 80;
a.widgets[i].options.max = 8192;
} else {
a.widgets[i].options.step = 10;
a.widgets[i].options.max = 99;
}
}
}
nodeType.prototype.onAdded = async function (graph) {
const measurementWidget = this.widgets[0];
let callback = measurementWidget.callback;
measurementWidget.callback = (...args) => {
setWidgetStep(this);
callback && callback.apply(measurementWidget, [...args]);
};
setTimeout(_=>{
setWidgetStep(this);
},1)
}
}
if(['easy showAnything', 'easy showTensorShape'].includes(nodeData.name)){
function populate(text) {
if (this.widgets) {
const pos = this.widgets.findIndex((w) => w.name === "text");
if (pos !== -1) {
for (let i = pos; i < this.widgets.length; i++) {
this.widgets[i].onRemove?.();
}
this.widgets.length = pos;
}
}
for (const list of text) {
const w = ComfyWidgets["STRING"](this, "text", ["STRING", { multiline: true }], app).widget;
w.inputEl.readOnly = true;
w.inputEl.style.opacity = 0.6;
w.value = list;
}
requestAnimationFrame(() => {
const sz = this.computeSize();
if (sz[0] < this.size[0]) {
sz[0] = this.size[0];
}
if (sz[1] < this.size[1]) {
sz[1] = this.size[1];
}
this.onResize?.(sz);
app.graph.setDirtyCanvas(true, false);
});
}
// When the node is executed we will be sent the input text, display this in the widget
const onExecuted = nodeType.prototype.onExecuted;
nodeType.prototype.onExecuted = function (message) {
onExecuted?.apply(this, arguments);
populate.call(this, message.text);
};
const onConfigure = nodeType.prototype.onConfigure;
nodeType.prototype.onConfigure = function () {
onConfigure?.apply(this, arguments);
if (this.widgets_values?.length) {
populate.call(this, this.widgets_values);
}
};
}
if(nodeData.name == 'easy convertAnything'){
const onNodeCreated = nodeType.prototype.onNodeCreated;
const changeType = async function (type) {
const body = new FormData();
body.append("type", type);
const response = await api.fetchApi("/easyuse/convert", { method:'POST',body});
}
nodeType.prototype.onNodeCreated = async function () {
onNodeCreated ? onNodeCreated.apply(this, []) : undefined;
setTimeout(_=>{
const type_control = this.widgets[this.widgets.findIndex((w) => w.name === "output_type")]
let _this = this
changeType(type_control.value)
type_control.callback = async() => {
_this.outputs[0].type = (type_control.value).toUpperCase()
_this.outputs[0].name = type_control.value
_this.outputs[0].label = type_control.value
changeType(type_control.value)
}
},300)
}
}
}
});
const getSetWidgets = ['rescale_after_model', 'rescale',
'lora_name', 'lora1_name', 'lora2_name', 'lora3_name',
'refiner_lora1_name', 'refiner_lora2_name', 'upscale_method',
'image_output', 'add_noise', 'info', 'sampler_name',
'ckpt_B_name', 'ckpt_C_name', 'save_model', 'refiner_ckpt_name',
'num_loras', 'mode', 'toggle', 'resolution', 'target_parameter',
'input_count', 'replace_count', 'downscale_mode', 'range_mode','text_combine_mode', 'input_mode',
'lora_count','ckpt_count', 'conditioning_mode', 'preset', 'use_tiled', 'use_batch', 'num_embeds',
"easing_mode"
]
function getSetters(node) {
if (node.widgets)
for (const w of node.widgets) {
if (getSetWidgets.includes(w.name)) {
if(node.comfyClass.indexOf("easy XYInputs:") != -1) widgetLogic3(node, w)
else if(w.name == 'sampler_name' && node.comfyClass == 'easy preSamplingSdTurbo') widgetLogic2(node, w);
else widgetLogic(node, w);
let widgetValue = w.value;
// Define getters and setters for widget values
Object.defineProperty(w, 'value', {
get() {
return widgetValue;
},
set(newVal) {
if (newVal !== widgetValue) {
widgetValue = newVal;
if(node.comfyClass.indexOf("easy XYInputs:") != -1) widgetLogic3(node, w)
else if(w.name == 'sampler_name' && node.comfyClass == 'easy preSamplingSdTurbo') widgetLogic2(node, w);
else widgetLogic(node, w);
}
}
});
}
}
}
Reference in New Issue View Git Blame Copy Permalink