221 lines
8.4 KiB
Python
221 lines
8.4 KiB
Python
import random
|
|
import server
|
|
from enum import Enum
|
|
|
|
|
|
class SGmode(Enum):
|
|
FIX = 1
|
|
INCR = 2
|
|
DECR = 3
|
|
RAND = 4
|
|
|
|
|
|
class SeedGenerator:
|
|
def __init__(self, base_value, action):
|
|
self.base_value = base_value
|
|
|
|
if action == "fixed" or action == "increment" or action == "decrement" or action == "randomize":
|
|
self.action = SGmode.FIX
|
|
elif action == 'increment for each node':
|
|
self.action = SGmode.INCR
|
|
elif action == 'decrement for each node':
|
|
self.action = SGmode.DECR
|
|
elif action == 'randomize for each node':
|
|
self.action = SGmode.RAND
|
|
|
|
def next(self):
|
|
seed = self.base_value
|
|
|
|
if self.action == SGmode.INCR:
|
|
self.base_value += 1
|
|
if self.base_value > 1125899906842624:
|
|
self.base_value = 0
|
|
elif self.action == SGmode.DECR:
|
|
self.base_value -= 1
|
|
if self.base_value < 0:
|
|
self.base_value = 1125899906842624
|
|
elif self.action == SGmode.RAND:
|
|
self.base_value = random.randint(0, 1125899906842624)
|
|
|
|
return seed
|
|
|
|
|
|
def control_seed(v, action, seed_is_global):
|
|
action = v['inputs']['action'] if seed_is_global else action
|
|
value = v['inputs']['value'] if seed_is_global else v['inputs']['seed_num']
|
|
|
|
if action == 'increment' or action == 'increment for each node':
|
|
value = value + 1
|
|
if value > 1125899906842624:
|
|
value = 0
|
|
elif action == 'decrement' or action == 'decrement for each node':
|
|
value = value - 1
|
|
if value < 0:
|
|
value = 1125899906842624
|
|
elif action == 'randomize' or action == 'randomize for each node':
|
|
value = random.randint(0, 1125899906842624)
|
|
if seed_is_global:
|
|
v['inputs']['value'] = value
|
|
|
|
return value
|
|
|
|
|
|
def prompt_seed_update(json_data):
|
|
try:
|
|
seed_widget_map = json_data['extra_data']['extra_pnginfo']['workflow']['seed_widgets']
|
|
except:
|
|
return None
|
|
|
|
workflow = json_data['extra_data']['extra_pnginfo']['workflow']
|
|
seed_widget_map = workflow['seed_widgets']
|
|
value = None
|
|
mode = None
|
|
node = None
|
|
action = None
|
|
seed_is_global = False
|
|
|
|
for k, v in json_data['prompt'].items():
|
|
if 'class_type' not in v:
|
|
continue
|
|
|
|
cls = v['class_type']
|
|
|
|
if cls == 'easy globalSeed':
|
|
mode = v['inputs']['mode']
|
|
action = v['inputs']['action']
|
|
value = v['inputs']['value']
|
|
node = k, v
|
|
seed_is_global = True
|
|
|
|
# control before generated
|
|
if mode is not None and mode and seed_is_global:
|
|
value = control_seed(node[1], action, seed_is_global)
|
|
|
|
if seed_is_global:
|
|
if value is not None:
|
|
seed_generator = SeedGenerator(value, action)
|
|
|
|
for k, v in json_data['prompt'].items():
|
|
for k2, v2 in v['inputs'].items():
|
|
if isinstance(v2, str) and '$GlobalSeed.value$' in v2:
|
|
v['inputs'][k2] = v2.replace('$GlobalSeed.value$', str(value))
|
|
|
|
if k not in seed_widget_map:
|
|
continue
|
|
|
|
if 'seed_num' in v['inputs']:
|
|
if isinstance(v['inputs']['seed_num'], int):
|
|
v['inputs']['seed_num'] = seed_generator.next()
|
|
|
|
if 'seed' in v['inputs']:
|
|
if isinstance(v['inputs']['seed'], int):
|
|
v['inputs']['seed'] = seed_generator.next()
|
|
|
|
if 'noise_seed' in v['inputs']:
|
|
if isinstance(v['inputs']['noise_seed'], int):
|
|
v['inputs']['noise_seed'] = seed_generator.next()
|
|
|
|
for k2, v2 in v['inputs'].items():
|
|
if isinstance(v2, str) and '$GlobalSeed.value$' in v2:
|
|
v['inputs'][k2] = v2.replace('$GlobalSeed.value$', str(value))
|
|
# control after generated
|
|
if mode is not None and not mode:
|
|
control_seed(node[1], action, seed_is_global)
|
|
else:
|
|
prompts = json_data['prompt'].items()
|
|
for k, v in prompts:
|
|
if 'class_type' not in v:
|
|
continue
|
|
cls = v['class_type']
|
|
if cls in ["easy wildcards","easy preSampling","easy preSamplingAdvanced","easy preSamplingSdTurbo","easy preSamplingDynamicCFG","easy preSamplingCascade","easy fullCascadeKSampler","easy fullkSampler","easy seed","easy latentNoisy"]:
|
|
extra_data = next((x for x in workflow["nodes"] if str(x["id"]) == k), None)
|
|
if extra_data is not None:
|
|
inputs = extra_data.get('inputs')
|
|
widgets_value = extra_data.get('widgets_values')
|
|
widgets_length = len(widgets_value)
|
|
if "disable" in widgets_value:
|
|
break
|
|
if inputs is not None and inputs != []:
|
|
seed_num_input = next((x for x in inputs if x['name'] == 'seed_num' and x['type'] == 'INT'), None)
|
|
if seed_num_input is not None:
|
|
action = 'fixed'
|
|
else:
|
|
action = widgets_value[widgets_length - 1]
|
|
else:
|
|
control_index = widgets_length - 2 if cls == 'easy seed' else widgets_length - 1
|
|
action = widgets_value[control_index]
|
|
|
|
# print(action)
|
|
node = k, v
|
|
value = control_seed(node[1], action, False)
|
|
|
|
if k not in seed_widget_map:
|
|
continue
|
|
|
|
if 'seed_num' in v['inputs']:
|
|
if isinstance(v['inputs']['seed_num'], int):
|
|
v['inputs']['seed_num'] = value
|
|
|
|
# 修改和seed节点连接的节点 (没有作用,不生效)
|
|
# if cls == 'easy seed':
|
|
# outputs = extra_data.get('outputs')
|
|
# if outputs and outputs[0] and 'links' in outputs[0]:
|
|
# for id in outputs[0]['links']:
|
|
# for x in workflow["nodes"]:
|
|
# if "inputs" in x and x['inputs'] != []:
|
|
# x_seed_num = next((i for i in x['inputs'] if i['name'] == 'seed_num' and i['type'] == 'INT'), None)
|
|
# if x_seed_num is not None and "link" in x_seed_num and id == x_seed_num['link']:
|
|
# widgets_values = x['widgets_values']
|
|
# if widgets_values:
|
|
# widgets_values[len(widgets_values)-1] = action
|
|
# widgets_values[len(widgets_values)-2] = value
|
|
# print(x)
|
|
|
|
|
|
return value is not None
|
|
|
|
|
|
def workflow_seed_update(json_data):
|
|
nodes = json_data['extra_data']['extra_pnginfo']['workflow']['nodes']
|
|
seed_widget_map = json_data['extra_data']['extra_pnginfo']['workflow']['seed_widgets']
|
|
prompt = json_data['prompt']
|
|
|
|
updated_seed_map = {}
|
|
value = None
|
|
|
|
for node in nodes:
|
|
node_id = str(node['id'])
|
|
if node_id in prompt:
|
|
if node['type'] == 'easy globalSeed':
|
|
value = prompt[node_id]['inputs']['value']
|
|
length = len(node['widgets_values'])
|
|
node['widgets_values'][length-1] = node['widgets_values'][0]
|
|
node['widgets_values'][0] = value
|
|
elif node_id in seed_widget_map:
|
|
widget_idx = seed_widget_map[node_id]
|
|
|
|
if 'seed_num' in prompt[node_id]['inputs']:
|
|
seed = prompt[node_id]['inputs']['seed_num']
|
|
elif 'noise_seed' in prompt[node_id]['inputs']:
|
|
seed = prompt[node_id]['inputs']['noise_seed']
|
|
else:
|
|
seed = prompt[node_id]['inputs']['seed']
|
|
|
|
node['widgets_values'][widget_idx] = seed
|
|
updated_seed_map[node_id] = seed
|
|
|
|
server.PromptServer.instance.send_sync("easyuse-global-seed", {"id": node_id, "value": value, "seed_map": updated_seed_map})
|
|
|
|
|
|
def onprompt(json_data):
|
|
is_changed = prompt_seed_update(json_data)
|
|
if is_changed:
|
|
workflow_seed_update(json_data)
|
|
|
|
return json_data
|
|
|
|
server.PromptServer.instance.add_on_prompt_handler(onprompt)
|
|
|
|
|
|
NODE_CLASS_MAPPINGS = {}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {} |