From cc0a11393c725e9e970146b7b5dad86da2b6c332 Mon Sep 17 00:00:00 2001 From: bluelovers Date: Sat, 24 Aug 2024 12:16:12 +0800 Subject: [PATCH] feat: add `ad_save_images_dir` (#689) * feat: add `ad-save-images` https://github.com/Bing-su/adetailer/issues/688 * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * feat: only enable `ad_save_images_dir` when not empty https://github.com/Bing-su/adetailer/pull/689#issuecomment-2300057990 * fix: F401 `pathlib.PurePath` imported but unused https://results.pre-commit.ci/run/github/632823919/1724229072.nxpHBQU4QvaIJ8IOFayT1Q --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> --- scripts/!adetailer.py | 17 ++++++++++++++++- 1 file changed, 16 insertions(+), 1 deletion(-) diff --git a/scripts/!adetailer.py b/scripts/!adetailer.py index 64ed90e..53d18d7 100644 --- a/scripts/!adetailer.py +++ b/scripts/!adetailer.py @@ -566,9 +566,14 @@ class AfterDetailerScript(scripts.Script): seed, _ = self.get_seed(p) if opts.data.get(condition, False): + ad_save_images_dir = opts.data.get("ad_save_images_dir", None) + + if not bool(ad_save_images_dir and ad_save_images_dir.strip()): + ad_save_images_dir = p.outpath_samples + images.save_image( image=image, - path=p.outpath_samples, + path=ad_save_images_dir, basename="", seed=seed, prompt=save_prompt, @@ -973,6 +978,16 @@ def on_ui_settings(): .needs_reload_ui(), ) + shared.opts.add_option( + "ad_save_images_dir", + shared.OptionInfo( + default="", + label="Output directory for adetailer images", + component=gr.Textbox, + section=section, + ), + ) + shared.opts.add_option( "ad_save_previews", shared.OptionInfo(False, "Save mask previews", section=section),