Commit bb0f1a2c authored by yfszzx's avatar yfszzx
Browse files

inspiration finished

parent dc665406

javascript/inspiration.js

+16 −11
Original line number Diff line number Diff line 
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() {
@@ -28,10 +34,9 @@ document.addEventListener("DOMContentLoaded", function() { 
            if (node) {

                node.style.display = "None"; //parentNode.removeChild(node) 

            } 

            

            gallery.querySelectorAll('img').forEach(function(e){    

                e.onclick = inspiration_image_click

        }) 

            });



        }

modules/inspiration.py

+126 −66
Original line number Diff line number Diff line 
import os

import random

import gradio

inspiration_path = "inspiration"

inspiration_system_path = os.path.join(inspiration_path, "system")

def read_name_list(file):

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")

        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)

    print(ret)

        line = f.readline()

    return ret



def save_name_list(file, name): 

    print(file)

    f = open(file, "a")

    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)

        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":        

        abondened = read_name_list(os.path.join(inspiration_system_path, types + "_abondened.txt"))        

        all_names = os.listdir(path)

        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) < 25:

            while len(names) < get_num:

                name = random.choice(all_names)

            if name not in abondened:

                if name not in abandoned:

                    names.append(name)

        else:

        names = random.sample(os.listdir(path), 25)

    names = random.sample(names, 25)

            names = all_names

    else:

        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

        else:

            os.makedirs(opts.inspiration_dir)

        if not flag:            

            os.mkdir(inspiration_path)

            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

modules/shared.py

+6 −0
Original line number Diff line number Diff line
@@ -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

modules/ui.py

+1 −1
Original line number Diff line number Diff line
@@ -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):

webui.py

+2 −1
Original line number Diff line number Diff line
@@ -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)))

CRA Git | Maintained and supported by SUSTech CRA and CCSE