adds components to infotext_fields (4d634dc5) · Commits · github_fork / Stable Diffusion Webui

modules/scripts.py

+14 −0

Original line number	Diff line number	Diff line
		@@ -330,6 +330,20 @@ class ScriptRunner:
		outputs=[script.group for script in self.selectable_scripts]
		)

		self.script_load_ctr = 0
		def onload_script_visibility(params):
		title = params.get('Script', None)
		if title:
		title_index = self.titles.index(title)
		visibility = title_index == self.script_load_ctr
		self.script_load_ctr = (self.script_load_ctr + 1) % len(self.titles)
		return gr.update(visible=visibility)
		else:
		return gr.update(visible=False)

		self.infotext_fields.append( (dropdown, lambda x: gr.update(value=x.get('Script', 'None'))) )
		self.infotext_fields.extend( [(script.group, onload_script_visibility) for script in self.selectable_scripts] )

		return inputs

		def run(self, p: StableDiffusionProcessing, *args):

+10 −0

Original line number	Diff line number	Diff line
		@@ -383,6 +383,15 @@ class Script(scripts.Script):
		y_type.change(fn=select_axis, inputs=[y_type], outputs=[fill_y_button])
		z_type.change(fn=select_axis, inputs=[z_type], outputs=[fill_z_button])

		self.infotext_fields = (
		(x_type, "X Type"),
		(x_values, "X Values"),
		(y_type, "Y Type"),
		(y_values, "Y Values"),
		(z_type, "Z Type"),
		(z_values, "Z Values"),
		)

		return [x_type, x_values, y_type, y_values, z_type, z_values, draw_legend, include_lone_images, include_sub_grids, no_fixed_seeds]

		def run(self, p, x_type, x_values, y_type, y_values, z_type, z_values, draw_legend, include_lone_images, include_sub_grids, no_fixed_seeds):
		@@ -541,6 +550,7 @@ class Script(scripts.Script):

		if grid_infotext[0] is None:
		pc.extra_generation_params = copy(pc.extra_generation_params)
		pc.extra_generation_params['Script'] = self.title()

		if x_opt.label != 'Nothing':
		pc.extra_generation_params["X Type"] = x_opt.label