96 lines
3.9 KiB
Python
96 lines
3.9 KiB
Python
|
import json
|
||
|
|
|
||
|
|
import gradio as gr
|
||
|
|
|
||
|
|
from modules import scripts, ui, errors
|
||
|
|
from modules.shared import cmd_opts
|
||
|
|
from modules.ui_components import ToolButton
|
||
|
|
|
||
|
|
|
||
|
|
class ScriptSeed(scripts.ScriptBuiltin):
|
||
|
|
section = "seed"
|
||
|
|
create_group = False
|
||
|
|
|
||
|
|
def __init__(self):
|
||
|
|
self.seed = None
|
||
|
|
self.reuse_seed = None
|
||
|
|
self.reuse_subseed = None
|
||
|
|
|
||
|
|
def title(self):
|
||
|
|
return "Seed"
|
||
|
|
|
||
|
|
def show(self, is_img2img):
|
||
|
|
return scripts.AlwaysVisible
|
||
|
|
|
||
|
|
def ui(self, is_img2img):
|
||
|
|
with gr.Row(elem_id=self.elem_id("seed_row")):
|
||
|
|
if cmd_opts.use_textbox_seed:
|
||
|
|
self.seed = gr.Textbox(label='Seed', value="", elem_id=self.elem_id("seed"))
|
||
|
|
else:
|
||
|
|
self.seed = gr.Number(label='Seed', value=-1, elem_id=self.elem_id("seed"), precision=0)
|
||
|
|
|
||
|
|
random_seed = ToolButton(ui.random_symbol, elem_id=self.elem_id("random_seed"), label='Random seed')
|
||
|
|
reuse_seed = ToolButton(ui.reuse_symbol, elem_id=self.elem_id("reuse_seed"), label='Reuse seed')
|
||
|
|
|
||
|
|
subseed = gr.Number(label='Variation seed', value=-1, elem_id=self.elem_id("subseed"), precision=0)
|
||
|
|
|
||
|
|
random_subseed = ToolButton(ui.random_symbol, elem_id=self.elem_id("random_subseed"))
|
||
|
|
reuse_subseed = ToolButton(ui.reuse_symbol, elem_id=self.elem_id("reuse_subseed"))
|
||
|
|
|
||
|
|
subseed_strength = gr.Slider(label='Variation strength', value=0.0, minimum=0, maximum=1, step=0.01, elem_id=self.elem_id("subseed_strength"))
|
||
|
|
|
||
|
|
random_seed.click(fn=None, _js="function(){setRandomSeed('" + self.elem_id("seed") + "')}", show_progress=False, inputs=[], outputs=[])
|
||
|
|
random_subseed.click(fn=None, _js="function(){setRandomSeed('" + self.elem_id("subseed") + "')}", show_progress=False, inputs=[], outputs=[])
|
||
|
|
|
||
|
|
self.infotext_fields = [
|
||
|
|
(self.seed, "Seed"),
|
||
|
|
(subseed, "Variation seed"),
|
||
|
|
(subseed_strength, "Variation seed strength"),
|
||
|
|
]
|
||
|
|
|
||
|
|
self.on_after_component(lambda x: connect_reuse_seed(self.seed, reuse_seed, x.component, False), elem_id=f'generation_info_{self.tabname}')
|
||
|
|
self.on_after_component(lambda x: connect_reuse_seed(self.seed, reuse_subseed, x.component, True), elem_id=f'generation_info_{self.tabname}')
|
||
|
|
|
||
|
|
return self.seed, subseed, subseed_strength
|
||
|
|
|
||
|
|
def before_process(self, p, seed, subseed, subseed_strength):
|
||
|
|
p.seed = seed
|
||
|
|
|
||
|
|
if subseed_strength > 0:
|
||
|
|
p.subseed = subseed
|
||
|
|
p.subseed_strength = subseed_strength
|
||
|
|
|
||
|
|
|
||
|
|
def connect_reuse_seed(seed: gr.Number, reuse_seed: gr.Button, generation_info: gr.Textbox, is_subseed):
|
||
|
|
""" Connects a 'reuse (sub)seed' button's click event so that it copies last used
|
||
|
|
(sub)seed value from generation info the to the seed field. If copying subseed and subseed strength
|
||
|
|
was 0, i.e. no variation seed was used, it copies the normal seed value instead."""
|
||
|
|
|
||
|
|
def copy_seed(gen_info_string: str, index):
|
||
|
|
res = -1
|
||
|
|
|
||
|
|
try:
|
||
|
|
gen_info = json.loads(gen_info_string)
|
||
|
|
index -= gen_info.get('index_of_first_image', 0)
|
||
|
|
|
||
|
|
if is_subseed and gen_info.get('subseed_strength', 0) > 0:
|
||
|
|
all_subseeds = gen_info.get('all_subseeds', [-1])
|
||
|
|
res = all_subseeds[index if 0 <= index < len(all_subseeds) else 0]
|
||
|
|
else:
|
||
|
|
all_seeds = gen_info.get('all_seeds', [-1])
|
||
|
|
res = all_seeds[index if 0 <= index < len(all_seeds) else 0]
|
||
|
|
|
||
|
|
except json.decoder.JSONDecodeError:
|
||
|
|
if gen_info_string:
|
||
|
|
errors.report(f"Error parsing JSON generation info: {gen_info_string}")
|
||
|
|
|
||
|
|
return [res, gr.update()]
|
||
|
|
|
||
|
|
reuse_seed.click(
|
||
|
|
fn=copy_seed,
|
||
|
|
_js="(x, y) => [x, selected_gallery_index()]",
|
||
|
|
show_progress=False,
|
||
|
|
inputs=[generation_info, seed],
|
||
|
|
outputs=[seed, seed]
|
||
|
|
)
|