138 lines
6.5 KiB
Python
138 lines
6.5 KiB
Python
import gradio as gr
|
|
import scripts.mergers.components as components
|
|
from scripts.mergers.mergers import smergegen, simggen
|
|
from scripts.mergers.xyplot import numanager
|
|
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],
|
|
)
|
|
GenParamGetter.events_assigned = True
|
|
|
|
if __package__ == "GenParamGetter":
|
|
script_callbacks.on_app_started(GenParamGetter.get_params_components)
|