Spaces:
Configuration error
Configuration error
| import { app } from "../../../../scripts/app.js"; | |
| import { api } from "../../../../scripts/api.js"; | |
| import { ComfyWidgets } from "../../../../scripts/widgets.js"; | |
| import { toast} from "../common/toast.js"; | |
| import { $t } from '../common/i18n.js'; | |
| import { findWidgetByName, toggleWidget, updateNodeHeight} from "../common/utils.js"; | |
| 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", "easy fluxLoader", "easy hunyuanDiTLoader", "easy pixArtLoader"] | |
| 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 == 'Preview&Choose' || 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') { | |
| let control_before_widget = findWidgetByName(node, 'control_before_generate') | |
| let control_after_widget = findWidgetByName(node, 'control_after_generate') | |
| if (widget.value === "disable") { | |
| toggleWidget(node, findWidgetByName(node, 'seed')) | |
| if(control_before_widget){ | |
| control_before_widget.last_value = control_before_widget.value | |
| control_before_widget.value = 'fixed' | |
| toggleWidget(node, control_before_widget) | |
| } | |
| if(control_after_widget){ | |
| control_after_widget.last_value = control_after_widget.value | |
| control_after_widget.value = 'fixed' | |
| toggleWidget(node, control_after_widget) | |
| } | |
| } else { | |
| toggleWidget(node, findWidgetByName(node, 'seed'), true) | |
| if(control_before_widget){ | |
| if(control_before_widget?.last_value) control_before_widget.value = control_before_widget.last_value | |
| toggleWidget(node, control_before_widget, true) | |
| } | |
| if(control_after_widget) { | |
| if(control_after_widget?.last_value) control_after_widget.value = control_after_widget.last_value | |
| toggleWidget(node, findWidgetByName(node, control_after_widget, 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 === 'num_controlnet') { | |
| let number_to_show = widget.value + 1 | |
| for (let i = 0; i < number_to_show; i++) { | |
| toggleWidget(node, findWidgetByName(node, 'controlnet_'+i), true) | |
| toggleWidget(node, findWidgetByName(node, 'controlnet_'+i+'_strength'), true) | |
| toggleWidget(node, findWidgetByName(node, 'scale_soft_weight_'+i),true) | |
| if (findWidgetByName(node, 'mode').value === "simple") { | |
| toggleWidget(node, findWidgetByName(node, 'start_percent_'+i)) | |
| toggleWidget(node, findWidgetByName(node, 'end_percent_'+i)) | |
| } else { | |
| toggleWidget(node, findWidgetByName(node, 'start_percent_'+i),true) | |
| toggleWidget(node, findWidgetByName(node, 'end_percent_'+i), true) | |
| } | |
| } | |
| for (let i = number_to_show; i < 10; i++) { | |
| toggleWidget(node, findWidgetByName(node, 'controlnet_'+i)) | |
| toggleWidget(node, findWidgetByName(node, 'controlnet_'+i+'_strength')) | |
| toggleWidget(node, findWidgetByName(node, 'start_percent_'+i)) | |
| toggleWidget(node, findWidgetByName(node, 'end_percent_'+i)) | |
| toggleWidget(node, findWidgetByName(node, 'scale_soft_weight_'+i)) | |
| } | |
| updateNodeHeight(node) | |
| } | |
| if (widget.name === 'mode') { | |
| switch (node.comfyClass) { | |
| case 'easy loraStack': | |
| for (let i = 0; i < (findWidgetByName(node, 'num_loras').value + 1); 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) | |
| break | |
| case 'easy controlnetStack': | |
| for (let i = 0; i < (findWidgetByName(node, 'num_controlnet').value + 1); i++) { | |
| if (widget.value === "simple") { | |
| toggleWidget(node, findWidgetByName(node, 'start_percent_'+i)) | |
| toggleWidget(node, findWidgetByName(node, 'end_percent_'+i)) | |
| } else { | |
| toggleWidget(node, findWidgetByName(node, 'start_percent_' + i), true) | |
| toggleWidget(node, findWidgetByName(node, 'end_percent_' + i), true) | |
| } | |
| } | |
| updateNodeHeight(node) | |
| break | |
| case 'easy icLightApply': | |
| if (widget.value === "Foreground") { | |
| toggleWidget(node, findWidgetByName(node, 'lighting'), true) | |
| toggleWidget(node, findWidgetByName(node, 'remove_bg'), true) | |
| toggleWidget(node, findWidgetByName(node, 'source')) | |
| } else { | |
| toggleWidget(node, findWidgetByName(node, 'lighting')) | |
| toggleWidget(node, findWidgetByName(node, 'source'), true) | |
| toggleWidget(node, findWidgetByName(node, 'remove_bg')) | |
| } | |
| updateNodeHeight(node) | |
| break | |
| } | |
| } | |
| if (widget.name === 'resolution') { | |
| if(widget.value === "自定义 x 自定义"){ | |
| widget.value = 'width x height (custom)' | |
| } | |
| if (widget.value === "自定义 x 自定义" || widget.value === 'width x height (custom)') { | |
| 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) | |
| } | |
| } | |
| if (widget.name === 'ratio') { | |
| if (widget.value === "custom") { | |
| 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) | |
| } | |
| } | |
| 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 KOLORS', | |
| 'FACEID PLUS V2', | |
| 'FACEID PORTRAIT (style transfer)', | |
| 'FACEID PORTRAIT UNNORM - SDXL only (strong)' | |
| ] | |
| 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, 'weight_kolors')) | |
| 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)){ | |
| toggleWidget(node, findWidgetByName(node, 'weight_faceidv2'), ['FACEID PLUS V2','FACEID PLUS KOLORS'].includes(widget.value) ? true : false); | |
| toggleWidget(node, findWidgetByName(node, 'weight_kolors'), ['FACEID PLUS KOLORS'].includes(widget.value) ? true : false); | |
| if(['FACEID PLUS KOLORS','FACEID PORTRAIT (style transfer)','FACEID PORTRAIT UNNORM - SDXL only (strong)'].includes(widget.value)){ | |
| 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++) { | |
| toggleWidget(node, findWidgetByName(node, 'weight'+i), true) | |
| } | |
| for (let i = number_to_show; i < 6; i++) { | |
| toggleWidget(node, findWidgetByName(node, 'weight'+i)) | |
| } | |
| updateNodeHeight(node) | |
| } | |
| if (widget.name === 'guider'){ | |
| switch (widget.value){ | |
| case 'Basic': | |
| toggleWidget(node, findWidgetByName(node, 'cfg')) | |
| toggleWidget(node, findWidgetByName(node, 'cfg_negative')) | |
| break | |
| case 'CFG': | |
| toggleWidget(node, findWidgetByName(node, 'cfg'),true) | |
| toggleWidget(node, findWidgetByName(node, 'cfg_negative')) | |
| break | |
| case 'IP2P+DualCFG': | |
| case 'DualCFG': | |
| toggleWidget(node, findWidgetByName(node, 'cfg'),true) | |
| toggleWidget(node, findWidgetByName(node, 'cfg_negative'), true) | |
| break | |
| } | |
| updateNodeHeight(node) | |
| } | |
| if (widget.name === 'scheduler'){ | |
| if (['karrasADV','exponentialADV','polyExponential'].includes(widget.value)){ | |
| toggleWidget(node, findWidgetByName(node, 'sigma_max'), true) | |
| toggleWidget(node, findWidgetByName(node, 'sigma_min'), true) | |
| toggleWidget(node, findWidgetByName(node, 'denoise')) | |
| toggleWidget(node, findWidgetByName(node, 'beta_d')) | |
| toggleWidget(node, findWidgetByName(node, 'beta_min')) | |
| toggleWidget(node, findWidgetByName(node, 'eps_s')) | |
| toggleWidget(node, findWidgetByName(node, 'coeff')) | |
| if(widget.value != 'exponentialADV'){ | |
| toggleWidget(node, findWidgetByName(node, 'rho'), true) | |
| }else{ | |
| toggleWidget(node, findWidgetByName(node, 'rho')) | |
| } | |
| }else if(widget.value == 'vp'){ | |
| toggleWidget(node, findWidgetByName(node, 'sigma_max')) | |
| toggleWidget(node, findWidgetByName(node, 'sigma_min')) | |
| toggleWidget(node, findWidgetByName(node, 'denoise')) | |
| toggleWidget(node, findWidgetByName(node, 'rho')) | |
| toggleWidget(node, findWidgetByName(node, 'beta_d'),true) | |
| toggleWidget(node, findWidgetByName(node, 'beta_min'),true) | |
| toggleWidget(node, findWidgetByName(node, 'eps_s'),true) | |
| toggleWidget(node, findWidgetByName(node, 'coeff')) | |
| } | |
| else{ | |
| toggleWidget(node, findWidgetByName(node, 'denoise'),true) | |
| toggleWidget(node, findWidgetByName(node, 'sigma_max')) | |
| toggleWidget(node, findWidgetByName(node, 'sigma_min')) | |
| toggleWidget(node, findWidgetByName(node, 'beta_d')) | |
| toggleWidget(node, findWidgetByName(node, 'beta_min')) | |
| toggleWidget(node, findWidgetByName(node, 'eps_s')) | |
| toggleWidget(node, findWidgetByName(node, 'rho')) | |
| if(widget.value == 'gits') toggleWidget(node, findWidgetByName(node, 'coeff'), true) | |
| else toggleWidget(node, findWidgetByName(node, 'coeff')) | |
| } | |
| updateNodeHeight(node) | |
| } | |
| if(widget.name === 'inpaint_mode'){ | |
| switch (widget.value){ | |
| case 'normal': | |
| case 'fooocus_inpaint': | |
| toggleWidget(node, findWidgetByName(node, 'dtype')) | |
| toggleWidget(node, findWidgetByName(node, 'fitting')) | |
| toggleWidget(node, findWidgetByName(node, 'function')) | |
| toggleWidget(node, findWidgetByName(node, 'scale')) | |
| toggleWidget(node, findWidgetByName(node, 'start_at')) | |
| toggleWidget(node, findWidgetByName(node, 'end_at')) | |
| break | |
| case 'brushnet_random': | |
| case 'brushnet_segmentation': | |
| toggleWidget(node, findWidgetByName(node, 'dtype'), true) | |
| toggleWidget(node, findWidgetByName(node, 'fitting')) | |
| toggleWidget(node, findWidgetByName(node, 'function')) | |
| toggleWidget(node, findWidgetByName(node, 'scale'), true) | |
| toggleWidget(node, findWidgetByName(node, 'start_at'), true) | |
| toggleWidget(node, findWidgetByName(node, 'end_at'), true) | |
| break | |
| case 'powerpaint': | |
| toggleWidget(node, findWidgetByName(node, 'dtype'), true) | |
| toggleWidget(node, findWidgetByName(node, 'fitting'),true) | |
| toggleWidget(node, findWidgetByName(node, 'function'),true) | |
| toggleWidget(node, findWidgetByName(node, 'scale'), true) | |
| toggleWidget(node, findWidgetByName(node, 'start_at'), true) | |
| toggleWidget(node, findWidgetByName(node, 'end_at'), true) | |
| break | |
| } | |
| updateNodeHeight(node) | |
| } | |
| if(widget.name == 't5_type'){ | |
| switch (widget.value){ | |
| case 'sd3': | |
| toggleWidget(node, findWidgetByName(node, 'clip_name'), true) | |
| toggleWidget(node, findWidgetByName(node, 'padding'), true) | |
| toggleWidget(node, findWidgetByName(node, 't5_name')) | |
| toggleWidget(node, findWidgetByName(node, 'device')) | |
| toggleWidget(node, findWidgetByName(node, 'dtype')) | |
| break | |
| case 't5v11': | |
| toggleWidget(node, findWidgetByName(node, 'clip_name')) | |
| toggleWidget(node, findWidgetByName(node, 'padding')) | |
| toggleWidget(node, findWidgetByName(node, 't5_name'),true) | |
| toggleWidget(node, findWidgetByName(node, 'device'),true) | |
| toggleWidget(node, findWidgetByName(node, 'dtype'),true) | |
| } | |
| updateNodeHeight(node) | |
| } | |
| if(widget.name == 'rem_mode'){ | |
| switch (widget.value){ | |
| case 'Inspyrenet': | |
| toggleWidget(node, findWidgetByName(node, 'torchscript_jit'), true) | |
| break | |
| default: | |
| toggleWidget(node, findWidgetByName(node, 'torchscript_jit'), false) | |
| break | |
| } | |
| } | |
| } | |
| 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 fluxLoader": | |
| case "easy comfyLoader": | |
| case "easy cascadeLoader": | |
| case "easy svdLoader": | |
| case "easy dynamiCrafterLoader": | |
| case "easy hunyuanDiTLoader": | |
| case "easy pixArtLoader": | |
| case "easy kolorsLoader": | |
| case "easy loraStack": | |
| case "easy controlnetStack": | |
| case "easy latentNoisy": | |
| case "easy preSampling": | |
| case "easy preSamplingAdvanced": | |
| case "easy preSamplingNoiseIn": | |
| case "easy preSamplingCustom": | |
| 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 imageColorMatch": | |
| case "easy imageDetailTransfer": | |
| case "easy loadImageBase64": | |
| 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 icLightApply': | |
| case 'easy ipadapterApply': | |
| case 'easy ipadapterApplyADV': | |
| case 'easy ipadapterApplyFaceIDKolors': | |
| case 'easy ipadapterApplyEncoder': | |
| case 'easy applyInpaint': | |
| 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] | |
| console.log(text) | |
| 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 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'){ | |
| const randomSeedButton = this.addWidget("button", "🎲 Manual Random Seed", null, _=>{ | |
| if(seed_control.value != 'fixed') seed_control.value = 'fixed' | |
| seed_widget.value = Math.floor(Math.random() * 1125899906842624) | |
| app.queuePrompt(0, 1) | |
| },{ serialize:false}) | |
| seed_widget.linkedWidgets = [randomSeedButton, seed_control]; | |
| } | |
| } | |
| 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)) | |
| setTimeout(_=>{ | |
| if(seed_control.name == 'control_before_generate' && seed_widget.value === 0) { | |
| seed_widget.value = Math.floor(Math.random() * 1125899906842624) | |
| } | |
| },1) | |
| } | |
| } | |
| 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', 'easy imageInterrogator'].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); | |
| }; | |
| if(!['easy imageInterrogator'].includes(nodeData.name)) { | |
| 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; | |
| 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 | |
| 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 | |
| } | |
| },300) | |
| } | |
| } | |
| if (nodeData.name == 'easy promptLine') { | |
| const onAdded = nodeType.prototype.onAdded; | |
| nodeType.prototype.onAdded = async function () { | |
| onAdded ? onAdded.apply(this, []) : undefined; | |
| let prompt_widget = this.widgets.find(w => w.name == "prompt") | |
| const button = this.addWidget("button", "get values from COMBO link", '', () => { | |
| const output_link = this.outputs[1]?.links?.length>0 ? this.outputs[1]['links'][0] : null | |
| const all_nodes = app.graph._nodes | |
| const node = all_nodes.find(cate=> cate.inputs?.find(input=> input.link == output_link)) | |
| if(!output_link || !node){ | |
| toast.error($t('No COMBO link'), 3000) | |
| return | |
| } | |
| else{ | |
| const input = node.inputs.find(input=> input.link == output_link) | |
| const widget_name = input.widget.name | |
| const widgets = node.widgets | |
| const widget = widgets.find(cate=> cate.name == widget_name) | |
| let values = widget?.options.values || null | |
| if(values){ | |
| values = values.join('\n') | |
| prompt_widget.value = values | |
| } | |
| } | |
| }, { | |
| serialize: false | |
| }) | |
| } | |
| } | |
| } | |
| }); | |
| 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', 'num_controlnet', 'mode', 'toggle', 'resolution', 'ratio', '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", "guider", "scheduler", "inpaint_mode", 't5_type', 'rem_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); | |
| } | |
| } | |
| }); | |
| } | |
| } | |
| } |