|
import random |
|
|
|
import nodes |
|
import server |
|
from enum import Enum |
|
from . import prompt_support |
|
from aiohttp import web |
|
from . import backend_support |
|
|
|
|
|
max_seed = 2**32 - 1 |
|
|
|
|
|
@server.PromptServer.instance.routes.get("/inspire/prompt_builder") |
|
def prompt_builder(request): |
|
result = {"presets": []} |
|
|
|
if "category" in request.rel_url.query: |
|
category = request.rel_url.query["category"] |
|
if category in prompt_support.prompt_builder_preset: |
|
result['presets'] = prompt_support.prompt_builder_preset[category] |
|
|
|
return web.json_response(result) |
|
|
|
|
|
@server.PromptServer.instance.routes.get("/inspire/cache/remove") |
|
def cache_remove(request): |
|
if "key" in request.rel_url.query: |
|
key = request.rel_url.query["key"] |
|
del backend_support.cache[key] |
|
|
|
return web.Response(status=200) |
|
|
|
|
|
@server.PromptServer.instance.routes.get("/inspire/cache/clear") |
|
def cache_clear(request): |
|
backend_support.cache.clear() |
|
return web.Response(status=200) |
|
|
|
|
|
@server.PromptServer.instance.routes.get("/inspire/cache/list") |
|
def cache_refresh(request): |
|
return web.Response(text=backend_support.ShowCachedInfo.get_data(), status=200) |
|
|
|
|
|
@server.PromptServer.instance.routes.post("/inspire/cache/settings") |
|
async def set_cache_settings(request): |
|
data = await request.text() |
|
try: |
|
backend_support.ShowCachedInfo.set_cache_settings(data) |
|
return web.Response(text='OK', status=200) |
|
except Exception as e: |
|
return web.Response(text=f"{e}", status=500) |
|
|
|
|
|
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 > max_seed: |
|
self.base_value = 0 |
|
elif self.action == SGmode.DECR: |
|
self.base_value -= 1 |
|
if self.base_value < 0: |
|
self.base_value = max_seed |
|
elif self.action == SGmode.RAND: |
|
self.base_value = random.randint(0, max_seed) |
|
|
|
return seed |
|
|
|
|
|
def control_seed(v): |
|
action = v['inputs']['action'] |
|
value = v['inputs']['value'] |
|
|
|
if action == 'increment' or action == 'increment for each node': |
|
value += 1 |
|
if value > max_seed: |
|
value = 0 |
|
elif action == 'decrement' or action == 'decrement for each node': |
|
value -= 1 |
|
if value < 0: |
|
value = max_seed |
|
elif action == 'randomize' or action == 'randomize for each node': |
|
value = random.randint(0, max_seed) |
|
|
|
v['inputs']['value'] = value |
|
|
|
return value |
|
|
|
|
|
def prompt_seed_update(json_data): |
|
try: |
|
widget_idx_map = json_data['extra_data']['extra_pnginfo']['workflow']['widget_idx_map'] |
|
except Exception: |
|
return False, None |
|
|
|
value = None |
|
mode = None |
|
node = None |
|
action = None |
|
|
|
for k, v in json_data['prompt'].items(): |
|
if 'class_type' not in v: |
|
continue |
|
|
|
cls = v['class_type'] |
|
if cls == 'GlobalSeed //Inspire': |
|
mode = v['inputs']['mode'] |
|
action = v['inputs']['action'] |
|
value = v['inputs']['value'] |
|
node = k, v |
|
|
|
|
|
if mode is not None and mode: |
|
value = control_seed(node[1]) |
|
|
|
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 widget_idx_map or ('seed' not in widget_idx_map[k] and 'noise_seed' not in widget_idx_map[k]): |
|
continue |
|
|
|
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)) |
|
|
|
|
|
if mode is not None and not mode: |
|
control_seed(node[1]) |
|
|
|
return value is not None, mode |
|
|
|
|
|
def workflow_seed_update(json_data, mode): |
|
nodes = json_data['extra_data']['extra_pnginfo']['workflow']['nodes'] |
|
widget_idx_map = json_data['extra_data']['extra_pnginfo']['workflow']['widget_idx_map'] |
|
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'] == 'GlobalSeed //Inspire': |
|
if mode is True: |
|
node['widgets_values'][3] = node['widgets_values'][0] |
|
node['widgets_values'][0] = prompt[node_id]['inputs']['value'] |
|
node['widgets_values'][2] = 'fixed' |
|
|
|
value = prompt[node_id]['inputs']['value'] |
|
|
|
elif node_id in widget_idx_map: |
|
widget_idx = None |
|
seed = None |
|
if 'noise_seed' in prompt[node_id]['inputs']: |
|
seed = prompt[node_id]['inputs']['noise_seed'] |
|
widget_idx = widget_idx_map[node_id].get('noise_seed') |
|
elif 'seed' in prompt[node_id]['inputs']: |
|
seed = prompt[node_id]['inputs']['seed'] |
|
widget_idx = widget_idx_map[node_id].get('seed') |
|
|
|
if widget_idx is not None: |
|
node['widgets_values'][widget_idx] = seed |
|
updated_seed_map[node_id] = seed |
|
|
|
server.PromptServer.instance.send_sync("inspire-global-seed", {"value": value, "seed_map": updated_seed_map}) |
|
|
|
|
|
def prompt_sampler_update(json_data): |
|
try: |
|
widget_idx_map = json_data['extra_data']['extra_pnginfo']['workflow']['widget_idx_map'] |
|
except Exception: |
|
return None |
|
|
|
nodes = json_data['extra_data']['extra_pnginfo']['workflow']['nodes'] |
|
prompt = json_data['prompt'] |
|
|
|
sampler_name = None |
|
scheduler = None |
|
|
|
for v in prompt.values(): |
|
cls = v.get('class_type') |
|
if cls == 'GlobalSampler //Inspire': |
|
sampler_name = v['inputs']['sampler_name'] |
|
scheduler = v['inputs']['scheduler'] |
|
|
|
if sampler_name is None: |
|
return |
|
|
|
for node in nodes: |
|
cls = node.get('type') |
|
if cls == 'GlobalSampler //Inspire' or cls is None: |
|
continue |
|
|
|
node_id = str(node['id']) |
|
|
|
if node_id in prompt and node_id in widget_idx_map: |
|
sampler_widget_idx = widget_idx_map[node_id].get('sampler_name') |
|
scheduler_widget_idx = widget_idx_map[node_id].get('scheduler') |
|
|
|
prompt_inputs = prompt[node_id]['inputs'] |
|
|
|
if ('sampler_name' in prompt_inputs and 'scheduler' in prompt_inputs and |
|
isinstance(prompt_inputs['sampler_name'], str) and 'scheduler' in prompt_inputs): |
|
|
|
if sampler_widget_idx is not None: |
|
prompt_inputs['sampler_name'] = sampler_name |
|
node['widgets_values'][sampler_widget_idx] = sampler_name |
|
server.PromptServer.instance.send_sync("inspire-node-feedback", {"node_id": node_id, "widget_name": 'sampler_name', "type": "text", "data": sampler_name}) |
|
|
|
if scheduler_widget_idx is not None: |
|
prompt_inputs['scheduler'] = scheduler |
|
node['widgets_values'][scheduler_widget_idx] = scheduler |
|
server.PromptServer.instance.send_sync("inspire-node-feedback", {"node_id": node_id, "widget_name": 'scheduler', "type": "text", "data": scheduler}) |
|
|
|
|
|
def workflow_loadimage_update(json_data): |
|
prompt = json_data['prompt'] |
|
|
|
for v in prompt.values(): |
|
if 'class_type' in v and v['class_type'] == 'LoadImage //Inspire': |
|
v['inputs']['image'] = "#DATA" |
|
|
|
|
|
def populate_wildcards(json_data): |
|
prompt = json_data['prompt'] |
|
|
|
if 'ImpactWildcardProcessor' in nodes.NODE_CLASS_MAPPINGS: |
|
if not hasattr(nodes.NODE_CLASS_MAPPINGS['ImpactWildcardProcessor'], 'process'): |
|
print(f"[Inspire Pack] Your Impact Pack is outdated. Please update to the latest version.") |
|
return |
|
|
|
wildcard_process = nodes.NODE_CLASS_MAPPINGS['ImpactWildcardProcessor'].process |
|
updated_widget_values = {} |
|
mbp_updated_widget_values = {} |
|
for k, v in prompt.items(): |
|
if 'class_type' in v and v['class_type'] == 'WildcardEncode //Inspire': |
|
inputs = v['inputs'] |
|
if inputs['mode'] and isinstance(inputs['populated_text'], str): |
|
if isinstance(inputs['seed'], list): |
|
try: |
|
input_node = prompt[inputs['seed'][0]] |
|
if input_node['class_type'] == 'ImpactInt': |
|
input_seed = int(input_node['inputs']['value']) |
|
if not isinstance(input_seed, int): |
|
continue |
|
if input_node['class_type'] == 'Seed (rgthree)': |
|
input_seed = int(input_node['inputs']['seed']) |
|
if not isinstance(input_seed, int): |
|
continue |
|
else: |
|
print(f"[Inspire Pack] Only `ImpactInt`, `Seed (rgthree)` and `Primitive` Node are allowed as the seed for '{v['class_type']}'. It will be ignored. ") |
|
continue |
|
except: |
|
continue |
|
else: |
|
input_seed = int(inputs['seed']) |
|
|
|
inputs['populated_text'] = wildcard_process(text=inputs['wildcard_text'], seed=input_seed) |
|
inputs['mode'] = False |
|
|
|
server.PromptServer.instance.send_sync("inspire-node-feedback", {"node_id": k, "widget_name": "populated_text", "type": "text", "data": inputs['populated_text']}) |
|
updated_widget_values[k] = inputs['populated_text'] |
|
|
|
elif 'class_type' in v and v['class_type'] == 'MakeBasicPipe //Inspire': |
|
inputs = v['inputs'] |
|
if inputs['wildcard_mode'] and (isinstance(inputs['positive_populated_text'], str) or isinstance(inputs['negative_populated_text'], str)): |
|
if isinstance(inputs['seed'], list): |
|
try: |
|
input_node = prompt[inputs['seed'][0]] |
|
if input_node['class_type'] == 'ImpactInt': |
|
input_seed = int(input_node['inputs']['value']) |
|
if not isinstance(input_seed, int): |
|
continue |
|
if input_node['class_type'] == 'Seed (rgthree)': |
|
input_seed = int(input_node['inputs']['seed']) |
|
if not isinstance(input_seed, int): |
|
continue |
|
else: |
|
print(f"[Inspire Pack] Only `ImpactInt`, `Seed (rgthree)` and `Primitive` Node are allowed as the seed for '{v['class_type']}'. It will be ignored. ") |
|
continue |
|
except: |
|
continue |
|
else: |
|
input_seed = int(inputs['seed']) |
|
|
|
if isinstance(inputs['positive_populated_text'], str): |
|
inputs['positive_populated_text'] = wildcard_process(text=inputs['positive_wildcard_text'], seed=input_seed) |
|
server.PromptServer.instance.send_sync("inspire-node-feedback", {"node_id": k, "widget_name": "positive_populated_text", "type": "text", "data": inputs['positive_populated_text']}) |
|
|
|
if isinstance(inputs['negative_populated_text'], str): |
|
inputs['negative_populated_text'] = wildcard_process(text=inputs['negative_wildcard_text'], seed=input_seed) |
|
server.PromptServer.instance.send_sync("inspire-node-feedback", {"node_id": k, "widget_name": "negative_populated_text", "type": "text", "data": inputs['negative_populated_text']}) |
|
|
|
inputs['wildcard_mode'] = False |
|
mbp_updated_widget_values[k] = inputs['positive_populated_text'], inputs['negative_populated_text'] |
|
|
|
if 'extra_data' in json_data and 'extra_pnginfo' in json_data['extra_data']: |
|
for node in json_data['extra_data']['extra_pnginfo']['workflow']['nodes']: |
|
key = str(node['id']) |
|
if key in updated_widget_values: |
|
node['widgets_values'][3] = updated_widget_values[key] |
|
node['widgets_values'][4] = False |
|
if key in mbp_updated_widget_values: |
|
node['widgets_values'][7] = mbp_updated_widget_values[key][0] |
|
node['widgets_values'][8] = mbp_updated_widget_values[key][1] |
|
node['widgets_values'][5] = False |
|
|
|
|
|
def force_reset_useless_params(json_data): |
|
prompt = json_data['prompt'] |
|
|
|
for k, v in prompt.items(): |
|
if 'class_type' in v and v['class_type'] == 'PromptBuilder //Inspire': |
|
v['inputs']['category'] = '#PLACEHOLDER' |
|
|
|
return json_data |
|
|
|
|
|
def onprompt(json_data): |
|
prompt_support.list_counter_map = {} |
|
|
|
is_changed, mode = prompt_seed_update(json_data) |
|
if is_changed: |
|
workflow_seed_update(json_data, mode) |
|
|
|
prompt_sampler_update(json_data) |
|
|
|
workflow_loadimage_update(json_data) |
|
populate_wildcards(json_data) |
|
|
|
force_reset_useless_params(json_data) |
|
|
|
return json_data |
|
|
|
|
|
server.PromptServer.instance.add_on_prompt_handler(onprompt) |
|
|
|
|
|
NODE_CLASS_MAPPINGS = {} |
|
NODE_DISPLAY_NAME_MAPPINGS = {} |
|
|
