1580158
/
stable-diffusion-webui-dataset-tag-editor
mirror of https://github.com/toshiaki1729/stable-diffusion-webui-dataset-tag-editor.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Added a "Add ALL" button to "Filter by Selection" 

#11 2. Add a "Add all" Button to "Filter by Selection"
pull/41/head
toshiaki1729 2022-11-29 18:37:16 +09:00
parent af96b0fab1
commit d9e6cc4f5d
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
scripts/main.py
View File

@ -159,6 +159,14 @@ def add_image_selection(filter_tags: List[str], idx: int):
return [arrange_selection_order(tmp_selection_img_path_set), idx]
def add_all_displayed_image_selection(filter_tags: List[str]):
global tmp_selection_img_path_set
filter_tags = dataset_tag_editor.read_tags(filter_tags)
img_paths, _ = dataset_tag_editor.get_filtered_imgpath_and_tags(filter_tags=filter_tags)
tmp_selection_img_path_set |= set(img_paths)
return arrange_selection_order(tmp_selection_img_path_set)
def invert_image_selection():
global tmp_selection_img_path_set
img_paths = dataset_tag_editor.get_img_path_set()
@ -341,6 +349,7 @@ def on_ui_tabs():
with gr.Column(variant='panel'):
with gr.Row():
btn_add_image_selection = gr.Button(value='Add selection [Enter]', elem_id='dataset_tag_editor_btn_add_image_selection')
btn_add_all_displayed_image_selection = gr.Button(value='Add ALL')
gl_selected_images = gr.Gallery(label='Filter Images', elem_id="dataset_tag_editor_selection_gallery").style(grid=opts.dataset_editor_image_columns)
txt_selection = gr.HTML(value=get_current_txt_selection())
@ -459,6 +468,12 @@ def on_ui_tabs():
outputs=[gl_selected_images, lbl_hidden_image_index]
)
btn_add_all_displayed_image_selection.click(
fn=add_all_displayed_image_selection,
inputs=cbg_tags,
outputs=gl_selected_images
)
btn_invert_image_selection.click(
fn=invert_image_selection,
inputs=None,