Mirror/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2024-06-07 21:20:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Process interrogation on all img2img subtabs

Browse Source
This commit is contained in:
Vladimir Repin 2023-01-18 20:16:52 +03:00
parent 3a0d6b7729
commit 8683427bd9
2 changed files with 50 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
javascript/ui.js
View File

@ -109,6 +109,13 @@ function get_extras_tab_index(){
return [get_tab_index('mode_extras'), get_tab_index('extras_resize_mode'), ...args] return [get_tab_index('mode_extras'), get_tab_index('extras_resize_mode'), ...args]
} }
function get_img2img_tab_index() {
let res = args_to_array(arguments)
res.splice(-2)
res[0] = get_tab_index('mode_img2img')
return res
}
function create_submit_args(args){ function create_submit_args(args){
res = [] res = []
for(var i=0;i<args.length;i++){ for(var i=0;i<args.length;i++){

50
modules/ui.py
View File

@ -204,9 +204,31 @@ def apply_styles(prompt, prompt_neg, style1_name, style2_name):
return [gr.Textbox.update(value=prompt), gr.Textbox.update(value=prompt_neg), gr.Dropdown.update(value="None"), gr.Dropdown.update(value="None")] return [gr.Textbox.update(value=prompt), gr.Textbox.update(value=prompt_neg), gr.Dropdown.update(value="None"), gr.Dropdown.update(value="None")]
def process_interrogate(interrogation_function, mode, ii_input_dir, ii_output_dir, *ii_singles):
if mode in {0, 1, 3, 4}:
return [interrogation_function(ii_singles[mode]), None]
elif mode == 2:
return [interrogation_function(ii_singles[mode]["image"]), None]
elif mode == 5:
assert not shared.cmd_opts.hide_ui_dir_config, "Launched with --hide-ui-dir-config, batch img2img disabled"
images = shared.listfiles(ii_input_dir)
print(f"Will process {len(images)} images.")
if ii_output_dir != "":
os.makedirs(ii_output_dir, exist_ok=True)
else:
ii_output_dir = ii_input_dir
for image in images:
img = Image.open(image)
filename = os.path.basename(image)
left, _ = os.path.splitext(filename)
print(interrogation_function(img), file=open(os.path.join(ii_output_dir, left + ".txt"), 'a'))
return [gr_show(True), None]
def interrogate(image): def interrogate(image):
prompt = shared.interrogator.interrogate(image.convert("RGB")) prompt = shared.interrogator.interrogate(image.convert("RGB"))
return gr_show(True) if prompt is None else prompt return gr_show(True) if prompt is None else prompt
@ -983,19 +1005,33 @@ def create_ui():
show_progress=False, show_progress=False,
) )
interrogate_args = dict(
_js="get_img2img_tab_index",
inputs=[
dummy_component,
img2img_batch_input_dir,
img2img_batch_output_dir,
init_img,
sketch,
init_img_with_mask,
inpaint_color_sketch,
init_img_inpaint,
],
outputs=[img2img_prompt, dummy_component],
show_progress=False,
)
img2img_prompt.submit(**img2img_args) img2img_prompt.submit(**img2img_args)
submit.click(**img2img_args) submit.click(**img2img_args)
img2img_interrogate.click( img2img_interrogate.click(
fn=interrogate, fn=lambda *args : process_interrogate(interrogate, *args),
inputs=[init_img], **interrogate_args,
outputs=[img2img_prompt],
) )
img2img_deepbooru.click( img2img_deepbooru.click(
fn=interrogate_deepbooru, fn=lambda *args : process_interrogate(interrogate_deepbooru, *args),
inputs=[init_img], **interrogate_args,
outputs=[img2img_prompt],
) )
prompts = [(txt2img_prompt, txt2img_negative_prompt), (img2img_prompt, img2img_negative_prompt)] prompts = [(txt2img_prompt, txt2img_negative_prompt), (img2img_prompt, img2img_negative_prompt)]