aboutsummaryrefslogtreecommitdiff
path: root/modules
diff options
context:
space:
mode:
Diffstat (limited to 'modules')
-rw-r--r--modules/inspiration.py192
-rw-r--r--modules/shared.py6
-rw-r--r--modules/ui.py2
3 files changed, 133 insertions, 67 deletions
diff --git a/modules/inspiration.py b/modules/inspiration.py
index 456bfcb5..f72ebf3a 100644
--- a/modules/inspiration.py
+++ b/modules/inspiration.py
@@ -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
diff --git a/modules/shared.py b/modules/shared.py
index ae033710..564b1b8d 100644
--- a/modules/shared.py
+++ b/modules/shared.py
@@ -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
diff --git a/modules/ui.py b/modules/ui.py
index 6a0a3c3b..b651eb9c 100644
--- a/modules/ui.py
+++ b/modules/ui.py
@@ -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):
generated by cgit v1.2.1 (git 2.18.0) at 2024-09-20 09:43:41 +0000