"
- load_info += f"{total_num} images in this directory. Loaded {num} images during {date} - {date_to_bak}, divided into {int((num + 1) // opts.images_history_num_per_page + 1)} pages"
- load_info += "
"
- _, image_list, _, _, visible_num = get_recent_images(1, 0, filenames)
- return (
- date_to,
- load_info,
- filenames,
- 1,
- image_list,
- "",
- "",
- visible_num,
- last_date_from,
- gradio.update(visible=total_num > num)
- )
-
-def delete_image(delete_num, name, filenames, image_index, visible_num):
- if name == "":
- return filenames, delete_num
- else:
- delete_num = int(delete_num)
- visible_num = int(visible_num)
- image_index = int(image_index)
- index = list(filenames).index(name)
- i = 0
- new_file_list = []
- for name in filenames:
- if i >= index and i < index + delete_num:
- if os.path.exists(name):
- if visible_num == image_index:
- new_file_list.append(name)
- i += 1
- continue
- print(f"Delete file {name}")
- os.remove(name)
- visible_num -= 1
- txt_file = os.path.splitext(name)[0] + ".txt"
- if os.path.exists(txt_file):
- os.remove(txt_file)
- else:
- print(f"Not exists file {name}")
- else:
- new_file_list.append(name)
- i += 1
- return new_file_list, 1, visible_num
-
-def save_image(file_name):
- if file_name is not None and os.path.exists(file_name):
- shutil.copy(file_name, opts.outdir_save)
-
-def get_recent_images(page_index, step, filenames):
- page_index = int(page_index)
- num_of_imgs_per_page = int(opts.images_history_num_per_page)
- max_page_index = len(filenames) // num_of_imgs_per_page + 1
- page_index = max_page_index if page_index == -1 else page_index + step
- page_index = 1 if page_index < 1 else page_index
- page_index = max_page_index if page_index > max_page_index else page_index
- idx_frm = (page_index - 1) * num_of_imgs_per_page
- image_list = filenames[idx_frm:idx_frm + num_of_imgs_per_page]
- length = len(filenames)
- visible_num = num_of_imgs_per_page if idx_frm + num_of_imgs_per_page <= length else length % num_of_imgs_per_page
- visible_num = num_of_imgs_per_page if visible_num == 0 else visible_num
- return page_index, image_list, "", "", visible_num
-
-def loac_batch_click(date_to):
- if date_to is None:
- return time.strftime("%Y%m%d",time.localtime(time.time())), []
- else:
- return None, []
-def forward_click(last_date_from, date_to_recorder):
- if len(date_to_recorder) == 0:
- return None, []
- if last_date_from == date_to_recorder[-1]:
- date_to_recorder = date_to_recorder[:-1]
- if len(date_to_recorder) == 0:
- return None, []
- return date_to_recorder[-1], date_to_recorder[:-1]
-
-def backward_click(last_date_from, date_to_recorder):
- if last_date_from is None or last_date_from == "":
- return time.strftime("%Y%m%d",time.localtime(time.time())), []
- if len(date_to_recorder) == 0 or last_date_from != date_to_recorder[-1]:
- date_to_recorder.append(last_date_from)
- return last_date_from, date_to_recorder
-
-
-def first_page_click(page_index, filenames):
- return get_recent_images(1, 0, filenames)
-
-def end_page_click(page_index, filenames):
- return get_recent_images(-1, 0, filenames)
-
-def prev_page_click(page_index, filenames):
- return get_recent_images(page_index, -1, filenames)
-
-def next_page_click(page_index, filenames):
- return get_recent_images(page_index, 1, filenames)
-
-def page_index_change(page_index, filenames):
- return get_recent_images(page_index, 0, filenames)
-
-def show_image_info(tabname_box, num, page_index, filenames):
- file = filenames[int(num) + int((page_index - 1) * int(opts.images_history_num_per_page))]
- tm = "" + time.strftime("%Y-%m-%d %H:%M:%S",time.localtime(os.path.getmtime(file))) + "
"
- return file, tm, num, file
-
-def enable_page_buttons():
- return gradio.update(visible=True)
-
-def change_dir(img_dir, date_to):
- warning = None
- try:
- if os.path.exists(img_dir):
- try:
- f = os.listdir(img_dir)
- except:
- warning = f"'{img_dir} is not a directory"
- else:
- warning = "The directory is not exist"
- except:
- warning = "The format of the directory is incorrect"
- if warning is None:
- today = time.strftime("%Y%m%d",time.localtime(time.time()))
- return gradio.update(visible=False), gradio.update(visible=True), None, None if date_to != today else today, gradio.update(visible=True), gradio.update(visible=True)
- else:
- return gradio.update(visible=True), gradio.update(visible=False), warning, date_to, gradio.update(visible=False), gradio.update(visible=False)
-
-def show_images_history(gr, opts, tabname, run_pnginfo, switch_dict):
- custom_dir = False
- if tabname == "txt2img":
- dir_name = opts.outdir_txt2img_samples
- elif tabname == "img2img":
- dir_name = opts.outdir_img2img_samples
- elif tabname == "extras":
- dir_name = opts.outdir_extras_samples
- elif tabname == faverate_tab_name:
- dir_name = opts.outdir_save
- else:
- custom_dir = True
- dir_name = None
-
- if not custom_dir:
- d = dir_name.split("/")
- dir_name = d[0]
- for p in d[1:]:
- dir_name = os.path.join(dir_name, p)
- if not os.path.exists(dir_name):
- os.makedirs(dir_name)
-
- with gr.Column() as page_panel:
- with gr.Row():
- with gr.Column(scale=1, visible=not custom_dir) as load_batch_box:
- load_batch = gr.Button('Load', elem_id=tabname + "_images_history_start", full_width=True)
- with gr.Column(scale=4):
- with gr.Row():
- img_path = gr.Textbox(dir_name, label="Images directory", placeholder="Input images directory", interactive=custom_dir)
- with gr.Row():
- with gr.Column(visible=False, scale=1) as batch_panel:
- with gr.Row():
- forward = gr.Button('Prev batch')
- backward = gr.Button('Next batch')
- with gr.Column(scale=3):
- load_info = gr.HTML(visible=not custom_dir)
- with gr.Row(visible=False) as warning:
- warning_box = gr.Textbox("Message", interactive=False)
-
- with gr.Row(visible=not custom_dir, elem_id=tabname + "_images_history") as main_panel:
- with gr.Column(scale=2):
- with gr.Row(visible=True) as turn_page_buttons:
- #date_to = gr.Dropdown(label="Date to")
- first_page = gr.Button('First Page')
- prev_page = gr.Button('Prev Page')
- page_index = gr.Number(value=1, label="Page Index")
- next_page = gr.Button('Next Page')
- end_page = gr.Button('End Page')
-
- history_gallery = gr.Gallery(show_label=False, elem_id=tabname + "_images_history_gallery").style(grid=opts.images_history_grid_num)
- with gr.Row():
- delete_num = gr.Number(value=1, interactive=True, label="number of images to delete consecutively next")
- delete = gr.Button('Delete', elem_id=tabname + "_images_history_del_button")
-
- with gr.Column():
- with gr.Row():
- with gr.Column():
- img_file_info = gr.Textbox(label="Generate Info", interactive=False, lines=6)
- gr.HTML("To activate inspiration function, you need get "inspiration" images first.
- You can create these images by run "Create inspiration images" script in txt2img page,
you can get the artists or art styles list from here
-
https://github.com/pharmapsychotic/clip-interrogator/tree/main/data
- download these files, and select these files in the "Create inspiration images" script UI
- There about 6000 artists and art styles in these files.
This takes server hours depending on your GPU type and how many pictures you generate for each artist/style
-
I suggest at least four images for each
-
You can also download generated pictures from here:
-
https://huggingface.co/datasets/yfszzx/inspiration
- unzip the file to the project directory of webui
- and restart webui, and enjoy the joy of creation!