File size: 6,831 Bytes
0163a2c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 |
import gradio as gr
import scripts.mergers.components as components
from scripts.mergers.mergers import smergegen, simggen
from scripts.mergers.xyplot import numanager
from scripts.mergers.pluslora import frompromptf
from modules import scripts, script_callbacks
class GenParamGetter(scripts.Script):
txt2img_gen_button = None
img2img_gen_button = None
events_assigned = False
def title(self):
return "Super Marger Generation Parameter Getter"
def show(self, is_img2img):
return scripts.AlwaysVisible
def get_wanted_params(params,wanted):
output = []
for target in wanted:
if target is None:
output.append(params[0])
continue
for param in params:
if hasattr(param,"label"):
if param.label == target:
output.append(param)
return output
def after_component(self, component: gr.components.Component, **_kwargs):
"""Find generate button"""
if component.elem_id == "txt2img_generate":
GenParamGetter.txt2img_gen_button = component
elif component.elem_id == "img2img_generate":
GenParamGetter.img2img_gen_button = component
def get_components_by_ids(root: gr.Blocks, ids: list[int]):
components: list[gr.Blocks] = []
if root._id in ids:
components.append(root)
ids = [_id for _id in ids if _id != root._id]
if hasattr(root,"children"):
for block in root.children:
components.extend(GenParamGetter.get_components_by_ids(block, ids))
return components
def compare_components_with_ids(components: list[gr.Blocks], ids: list[int]):
return len(components) == len(ids) and all(component._id == _id for component, _id in zip(components, ids))
def get_params_components(demo: gr.Blocks, app):
for _id, _is_txt2img in zip([GenParamGetter.txt2img_gen_button._id, GenParamGetter.img2img_gen_button._id], [True, False]):
dependencies: list[dict] = [x for x in demo.dependencies if x["trigger"] == "click" and _id in x["targets"]]
dependency: dict = None
cnet_dependency: dict = None
UiControlNetUnit = None
for d in dependencies:
if len(d["outputs"]) == 1:
outputs = GenParamGetter.get_components_by_ids(demo, d["outputs"])
output = outputs[0]
if (
isinstance(output, gr.State)
and type(output.value).__name__ == "UiControlNetUnit"
):
cnet_dependency = d
UiControlNetUnit = type(output.value)
elif len(d["outputs"]) == 4:
dependency = d
params = [params for params in demo.fns if GenParamGetter.compare_components_with_ids(params.inputs, dependency["inputs"])]
from pprint import pprint
if _is_txt2img:
components.paramsnames = [x.label if hasattr(x,"label") else "None" for x in params[0].inputs]
if _is_txt2img:
components.txt2img_params = params[0].inputs
else:
components.img2img_params = params[0].inputs
if not GenParamGetter.events_assigned:
with demo:
components.merge.click(
fn=smergegen,
inputs=[*components.msettings,components.esettings1,*components.genparams,*components.hiresfix,*components.lucks,components.currentmodel,components.dfalse,*components.txt2img_params],
outputs=[components.submit_result,components.currentmodel]
)
components.mergeandgen.click(
fn=smergegen,
inputs=[*components.msettings,components.esettings1,*components.genparams,*components.hiresfix,*components.lucks,components.currentmodel,components.dtrue,*components.txt2img_params],
outputs=[components.submit_result,components.currentmodel,*components.imagegal]
)
components.gen.click(
fn=simggen,
inputs=[*components.genparams,*components.hiresfix,components.currentmodel,components.id_sets,gr.Textbox(value="No id",visible=False),*components.txt2img_params],
outputs=[*components.imagegal],
)
components.s_reserve.click(
fn=numanager,
inputs=[gr.Textbox(value="reserve",visible=False),*components.xysettings,*components.msettings,*components.genparams,*components.hiresfix,*components.lucks,*components.txt2img_params],
outputs=[components.numaframe]
)
components.s_reserve1.click(
fn=numanager,
inputs=[gr.Textbox(value="reserve",visible=False),*components.xysettings,*components.msettings,*components.genparams,*components.hiresfix,*components.lucks,*components.txt2img_params],
outputs=[components.numaframe]
)
components.gengrid.click(
fn=numanager,
inputs=[gr.Textbox(value="normal",visible=False),*components.xysettings,*components.msettings,*components.genparams,*components.hiresfix,*components.lucks,*components.txt2img_params],
outputs=[components.submit_result,components.currentmodel,*components.imagegal],
)
components.s_startreserve.click(
fn=numanager,
inputs=[gr.Textbox(value=" ",visible=False),*components.xysettings,*components.msettings,*components.genparams,*components.hiresfix,*components.lucks,*components.txt2img_params],
outputs=[components.submit_result,components.currentmodel,*components.imagegal],
)
components.rand_merge.click(
fn=numanager,
inputs=[gr.Textbox(value="random",visible=False),*components.xysettings,*components.msettings,*components.genparams,*components.hiresfix,*components.lucks,*components.txt2img_params],
outputs=[components.submit_result,components.currentmodel,*components.imagegal],
)
components.frompromptb.click(
fn=frompromptf,
inputs=[*components.txt2img_params],
outputs=components.sml_loranames,
)
GenParamGetter.events_assigned = True
if __package__ == "GenParamGetter":
script_callbacks.on_app_started(GenParamGetter.get_params_components)
|