| 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) | |