michael/stable-diffusion-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

inspiration finished

Browse source
This commit is contained in:
yfszzx 2022-10-22 01:23:00 +08:00
parent dc66540629
commit bb0f1a2cda
5 changed files with 151 additions and 79 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
javascript/inspiration.js
View file

@ -1,25 +1,31 @@
function public_image_index_in_gallery(item, gallery){
var imgs = gallery.querySelectorAll("img.h-full")
var index;
var i = 0;
gallery.querySelectorAll("img").forEach(function(e){
imgs.forEach(function(e){
if (e == item)
index = i;
i += 1;
});
var num = imgs.length / 2
index = (index < num) ? index : (index - num)
return index;
}
function inspiration_selected(name, types, name_list){
function inspiration_selected(name, name_list){
var btn = gradioApp().getElementById("inspiration_select_button")
return [gradioApp().getElementById("inspiration_select_button").getAttribute("img-index"), types];
}
return [gradioApp().getElementById("inspiration_select_button").getAttribute("img-index")];
}
function inspiration_click_get_button(){
gradioApp().getElementById("inspiration_get_button").click();
}
var inspiration_image_click = function(){
var index = public_image_index_in_gallery(this, gradioApp().getElementById("inspiration_gallery"));
var btn = gradioApp().getElementById("inspiration_select_button")
btn.setAttribute("img-index", index)
setTimeout(function(btn){btn.click();}, 10, btn)
var btn = gradioApp().getElementById("inspiration_select_button");
btn.setAttribute("img-index", index);
setTimeout(function(btn){btn.click();}, 10, btn);
}
document.addEventListener("DOMContentLoaded", function() {
var mutationObserver = new MutationObserver(function(m){
var gallery = gradioApp().getElementById("inspiration_gallery")
@ -27,11 +33,10 @@ document.addEventListener("DOMContentLoaded", function() {
var node = gallery.querySelector(".absolute.backdrop-blur.h-full")
if (node) {
node.style.display = "None"; //parentNode.removeChild(node)
}
}
gallery.querySelectorAll('img').forEach(function(e){
e.onclick = inspiration_image_click
})
});
}

192
modules/inspiration.py
View file

@ -1,122 +1,182 @@
import os
import random
import gradio
inspiration_path = "inspiration"
inspiration_system_path = os.path.join(inspiration_path, "system")
def read_name_list(file):
import gradio
from modules.shared import opts
inspiration_system_path = os.path.join(opts.inspiration_dir, "system")
def read_name_list(file, types=None, keyword=None):
if not os.path.exists(file):
return []
f = open(file, "r")
ret = []
f = open(file, "r")
line = f.readline()
while len(line) > 0:
line = line.rstrip("\n")
ret.append(line)
print(ret)
if types is not None:
dirname = os.path.split(line)
if dirname[0] in types and keyword in dirname[1]:
ret.append(line)
else:
ret.append(line)
line = f.readline()
return ret
def save_name_list(file, name):
print(file)
f = open(file, "a")
f.write(name + "\n")
with open(file, "a") as f:
f.write(name + "\n")
def get_inspiration_images(source, types):
path = os.path.join(inspiration_path , types)
def get_types_list():
files = os.listdir(opts.inspiration_dir)
types = []
for x in files:
path = os.path.join(opts.inspiration_dir, x)
if x[0] == ".":
continue
if not os.path.isdir(path):
continue
if path == inspiration_system_path:
continue
types.append(x)
return types
def get_inspiration_images(source, types, keyword):
get_num = int(opts.inspiration_rows_num * opts.inspiration_cols_num)
if source == "Favorites":
names = read_name_list(os.path.join(inspiration_system_path, types + "_faverites.txt"))
names = random.sample(names, 25)
names = read_name_list(os.path.join(inspiration_system_path, "faverites.txt"), types, keyword)
names = random.sample(names, get_num) if len(names) > get_num else names
elif source == "Abandoned":
names = read_name_list(os.path.join(inspiration_system_path, types + "_abondened.txt"))
names = random.sample(names, 25)
elif source == "Exclude abandoned":
abondened = read_name_list(os.path.join(inspiration_system_path, types + "_abondened.txt"))
all_names = os.listdir(path)
names = []
while len(names) < 25:
name = random.choice(all_names)
if name not in abondened:
names.append(name)
names = read_name_list(os.path.join(inspiration_system_path, "abandoned.txt"), types, keyword)
print(names)
names = random.sample(names, get_num) if len(names) > get_num else names
elif source == "Exclude abandoned":
abandoned = read_name_list(os.path.join(inspiration_system_path, "abandoned.txt"), types, keyword)
all_names = []
for tp in types:
name_list = os.listdir(os.path.join(opts.inspiration_dir, tp))
all_names += [os.path.join(tp, x) for x in name_list if keyword in x]
if len(all_names) > get_num:
names = []
while len(names) < get_num:
name = random.choice(all_names)
if name not in abandoned:
names.append(name)
else:
names = all_names
else:
names = random.sample(os.listdir(path), 25)
names = random.sample(names, 25)
all_names = []
for tp in types:
name_list = os.listdir(os.path.join(opts.inspiration_dir, tp))
all_names += [os.path.join(tp, x) for x in name_list if keyword in x]
names = random.sample(all_names, get_num) if len(all_names) > get_num else all_names
image_list = []
for a in names:
image_path = os.path.join(path, a)
image_path = os.path.join(opts.inspiration_dir, a)
images = os.listdir(image_path)
image_list.append(os.path.join(image_path, random.choice(images)))
return image_list, names
image_list.append((os.path.join(image_path, random.choice(images)), a))
return image_list, names, ""
def select_click(index, types, name_list):
def select_click(index, name_list):
name = name_list[int(index)]
path = os.path.join(inspiration_path, types, name)
path = os.path.join(opts.inspiration_dir, name)
images = os.listdir(path)
return name, [os.path.join(path, x) for x in images]
return name, [os.path.join(path, x) for x in images], ""
def give_up_click(name, types):
file = os.path.join(inspiration_system_path, types + "_abandoned.txt")
def give_up_click(name):
file = os.path.join(inspiration_system_path, "abandoned.txt")
name_list = read_name_list(file)
if name not in name_list:
save_name_list(file, name)
return "Added to abandoned list"
def collect_click(name, types):
file = os.path.join(inspiration_system_path, types + "_faverites.txt")
print(file)
def collect_click(name):
file = os.path.join(inspiration_system_path, "faverites.txt")
name_list = read_name_list(file)
print(name_list)
if name not in name_list:
save_name_list(file, name)
return "Added to faverite list"
def moveout_click(name, types):
file = os.path.join(inspiration_system_path, types + "_faverites.txt")
def moveout_click(name, source):
if source == "Abandoned":
file = os.path.join(inspiration_system_path, "abandoned.txt")
if source == "Favorites":
file = os.path.join(inspiration_system_path, "faverites.txt")
else:
return None
name_list = read_name_list(file)
if name not in name_list:
save_name_list(file, name)
os.remove(file)
with open(file, "a") as f:
for a in name_list:
if a != name:
f.write(a)
return "Moved out {name} from {source} list"
def source_change(source):
if source == "Abandoned" or source == "Favorites":
return gradio.Button.update(visible=True, value=f"Move out {source}")
if source in ["Abandoned", "Favorites"]:
return gradio.update(visible=True), []
else:
return gradio.Button.update(visible=False)
return gradio.update(visible=False), []
def add_to_prompt(name, prompt):
print(name, prompt)
name = os.path.basename(name)
return prompt + "," + name
def ui(gr, opts):
def ui(gr, opts, txt2img_prompt, img2img_prompt):
with gr.Blocks(analytics_enabled=False) as inspiration:
flag = os.path.exists(inspiration_path)
flag = os.path.exists(opts.inspiration_dir)
if flag:
types = os.listdir(inspiration_path)
types = [x for x in types if x != "system"]
types = get_types_list()
flag = len(types) > 0
if not flag:
os.mkdir(inspiration_path)
else:
os.makedirs(opts.inspiration_dir)
if not flag:
gr.HTML("""
<div align='center' width="50%>You need get "inspiration" images first. You can create these images by run "Create inspiration images" script in txt2img page, or download zip file from here and unzip these file under fold "inpiration".</div>"
<div align='center' width="50%"><h2>To activate inspiration function, you need get "inspiration" images first. </h2><br>
You can create these images by run "Create inspiration images" script in txt2img page, <br> you can get the artists or art styles list from here<br>
<a>https://github.com/pharmapsychotic/clip-interrogator/tree/main/data</a><br>
download these files, and select these files in the "Create inspiration images" script UI<br>
There about 6000 artists and art styles in these files. <br>This takes server hours depending on your GPU type and how many pictures you generate for each artist/style
<br>I suggest at least four images for each<br><br><br>
<h2>You can also download generated pictures from here:</h2><br>
<a>https://huggingface.co/datasets/yfszzx/inspiration</a><br>
unzip the file to the project directory of webui<br>
and restart webui, and enjoy the joy of creation!<br></div>
""")
return inspiration
if not os.path.exists(inspiration_system_path):
os.mkdir(inspiration_system_path)
gallery, names = get_inspiration_images("Exclude abandoned", types[0])
with gr.Row():
with gr.Column(scale=2):
inspiration_gallery = gr.Gallery(gallery, show_label=False, elem_id="inspiration_gallery").style(grid=5, height='auto')
inspiration_gallery = gr.Gallery(show_label=False, elem_id="inspiration_gallery").style(grid=opts.inspiration_cols_num, height='auto')
with gr.Column(scale=1):
types = gr.Dropdown(choices=types, value=types[0], label="Type", visible=len(types) > 1)
print(types)
types = gr.CheckboxGroup(choices=types, value=types)
keyword = gr.Textbox("", label="Key word")
with gr.Row():
source = gr.Dropdown(choices=["All", "Favorites", "Exclude abandoned", "Abandoned"], value="Exclude abandoned", label="Source")
get_inspiration = gr.Button("Get inspiration")
get_inspiration = gr.Button("Get inspiration", elem_id="inspiration_get_button")
name = gr.Textbox(show_label=False, interactive=False)
with gr.Row():
send_to_txt2img = gr.Button('to txt2img')
send_to_img2img = gr.Button('to img2img')
style_gallery = gr.Gallery(show_label=False, elem_id="inspiration_style_gallery").style(grid=2, height='auto')
style_gallery = gr.Gallery(show_label=False).style(grid=2, height='auto')
collect = gr.Button('Collect')
give_up = gr.Button("Don't show any more")
give_up = gr.Button("Don't show again")
moveout = gr.Button("Move out", visible=False)
with gr.Row():
warning = gr.HTML()
with gr.Row(visible=False):
select_button = gr.Button('set button', elem_id="inspiration_select_button")
name_list = gr.State(names)
source.change(source_change, inputs=[source], outputs=[moveout])
get_inspiration.click(get_inspiration_images, inputs=[source, types], outputs=[inspiration_gallery, name_list])
select_button.click(select_click, _js="inspiration_selected", inputs=[name, types, name_list], outputs=[name, style_gallery])
give_up.click(give_up_click, inputs=[name, types], outputs=None)
collect.click(collect_click, inputs=[name, types], outputs=None)
name_list = gr.State()
get_inspiration.click(get_inspiration_images, inputs=[source, types, keyword], outputs=[inspiration_gallery, name_list, keyword])
source.change(source_change, inputs=[source], outputs=[moveout, style_gallery])
source.change(fn=None, _js="inspiration_click_get_button", inputs=None, outputs=None)
keyword.submit(fn=None, _js="inspiration_click_get_button", inputs=None, outputs=None)
select_button.click(select_click, _js="inspiration_selected", inputs=[name, name_list], outputs=[name, style_gallery, warning])
give_up.click(give_up_click, inputs=[name], outputs=[warning])
collect.click(collect_click, inputs=[name], outputs=[warning])
moveout.click(moveout_click, inputs=[name, source], outputs=[warning])
send_to_txt2img.click(add_to_prompt, inputs=[name, txt2img_prompt], outputs=[txt2img_prompt])
send_to_img2img.click(add_to_prompt, inputs=[name, img2img_prompt], outputs=[img2img_prompt])
send_to_txt2img.click(None, _js='switch_to_txt2img', inputs=None, outputs=None)
send_to_img2img.click(None, _js="switch_to_img2img_img2img", inputs=None, outputs=None)
return inspiration

6
modules/shared.py
View file

@ -316,6 +316,12 @@ options_templates.update(options_section(('sampler-params', "Sampler parameters"
'eta_noise_seed_delta': OptionInfo(0, "Eta noise seed delta", gr.Number, {"precision": 0}),
}))
options_templates.update(options_section(('inspiration', "Inspiration"), {
"inspiration_dir": OptionInfo("inspiration", "Directory of inspiration", component_args=hide_dirs),
"inspiration_max_samples": OptionInfo(4, "Maximum number of samples, used to determine which folders to skip when continue running the create script", gr.Slider, {"minimum": 1, "maximum": 20, "step": 1}),
"inspiration_rows_num": OptionInfo(4, "Rows of inspiration interface frame", gr.Slider, {"minimum": 4, "maximum": 16, "step": 1}),
"inspiration_cols_num": OptionInfo(8, "Columns of inspiration interface frame", gr.Slider, {"minimum": 4, "maximum": 16, "step": 1}),
}))
class Options:
data = None

2
modules/ui.py
View file

@ -1180,7 +1180,7 @@ def create_ui(wrap_gradio_gpu_call):
}
browser_interface = images_history.create_history_tabs(gr, opts, wrap_gradio_call(modules.extras.run_pnginfo), images_history_switch_dict)
inspiration_interface = inspiration.ui(gr, opts)
inspiration_interface = inspiration.ui(gr, opts, txt2img_prompt, img2img_prompt)
with gr.Blocks() as modelmerger_interface:
with gr.Row().style(equal_height=False):

3
webui.py
View file

@ -72,6 +72,7 @@ def wrap_gradio_gpu_call(func, extra_outputs=None):
return modules.ui.wrap_gradio_call(f, extra_outputs=extra_outputs)
def initialize():
modules.scripts.load_scripts(os.path.join(script_path, "scripts"))
if cmd_opts.ui_debug_mode:
class enmpty():
name = None
@ -84,7 +85,7 @@ def initialize():
shared.face_restorers.append(modules.face_restoration.FaceRestoration())
modelloader.load_upscalers()
modules.scripts.load_scripts(os.path.join(script_path, "scripts"))
shared.sd_model = modules.sd_models.load_model()
shared.opts.onchange("sd_model_checkpoint", wrap_queued_call(lambda: modules.sd_models.reload_model_weights(shared.sd_model)))