Update image_browser.py

Added "All" tab allowing users to view images in all subfolders together in a single view
pull/184/head
Ezekiel-Rage 2023-05-27 12:09:39 -07:00 committed by GitHub
parent c61fae964a
commit 5aec351162
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 1 deletions

View File

@ -96,7 +96,8 @@ def check_image_browser_active_tabs():
shared.opts.save(shared.config_filename)
favorite_tab_name = "Favorites"
default_tab_options = ["txt2img", "img2img", "txt2img-grids", "img2img-grids", "Extras", favorite_tab_name, "Others"]
default_output_path = "outputs"
default_tab_options = ["All", "txt2img", "img2img", "txt2img-grids", "img2img-grids", "Extras", favorite_tab_name, "Others"]
check_image_browser_active_tabs()
tabs_list = [tab.strip() for tab in chain.from_iterable(csv.reader(StringIO(opts.image_browser_active_tabs))) if tab] if hasattr(opts, "image_browser_active_tabs") else default_tab_options
try:
@ -106,6 +107,7 @@ except AttributeError:
tabs_list.append("Maintenance") # mandatory tab
path_maps = {
"All": opts.outdir_samples or default_output_path,
"txt2img": opts.outdir_samples or opts.outdir_txt2img_samples,
"img2img": opts.outdir_samples or opts.outdir_img2img_samples,
"txt2img-grids": opts.outdir_grids or opts.outdir_txt2img_grids,
@ -114,6 +116,7 @@ path_maps = {
favorite_tab_name: opts.outdir_save
}
class ImageBrowserTab():
seen_base_tags = set()