jaidaken
f09734b0ee
Some checks failed
Python Linting / Run Ruff (push) Has been cancelled
Python Linting / Run Pylint (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.10, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.11, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.12, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-unix-nightly (12.1, , linux, 3.11, [self-hosted Linux], nightly) (push) Has been cancelled
Execution Tests / test (macos-latest) (push) Has been cancelled
Execution Tests / test (ubuntu-latest) (push) Has been cancelled
Execution Tests / test (windows-latest) (push) Has been cancelled
Test server launches without errors / test (push) Has been cancelled
Unit Tests / test (macos-latest) (push) Has been cancelled
Unit Tests / test (ubuntu-latest) (push) Has been cancelled
Unit Tests / test (windows-2022) (push) Has been cancelled
Includes 30 custom nodes committed directly, 7 Civitai-exclusive loras stored via Git LFS, and a setup script that installs all dependencies and downloads HuggingFace-hosted models on vast.ai. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
1274 lines
48 KiB
JavaScript
1274 lines
48 KiB
JavaScript
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);
|
|
}
|
|
}
|
|
});
|
|
}
|
|
}
|
|
} |