From 23e3c5d7783827462420cdcbf64d7ef31d397d6e Mon Sep 17 00:00:00 2001
From: Dowon <ks2515@naver.com>
Date: Sat, 3 Aug 2024 13:05:45 +0900
Subject: [PATCH] feat(script): same sampler option

---
 aaaaaa/ui.py          | 9 +++++++--
 scripts/!adetailer.py | 3 +++
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/aaaaaa/ui.py b/aaaaaa/ui.py
index f7a5148..e4efb54 100644
--- a/aaaaaa/ui.py
+++ b/aaaaaa/ui.py
@@ -550,11 +550,16 @@ def inpainting(w: Widgets, n: int, is_img2img: bool, webui_info: WebuiInfo):
                 elem_id=eid("ad_use_sampler"),
             )
 
+            sampler_names = [
+                "Use same sampler",
+                *webui_info.sampler_names,
+            ]
+
             with gr.Row():
                 w.ad_sampler = gr.Dropdown(
                     label="ADetailer sampler" + suffix(n),
-                    choices=webui_info.sampler_names,
-                    value=webui_info.sampler_names[0],
+                    choices=sampler_names,
+                    value=sampler_names[1],
                     visible=True,
                     elem_id=eid("ad_sampler"),
                 )
diff --git a/scripts/!adetailer.py b/scripts/!adetailer.py
index 43865e5..b655e6e 100644
--- a/scripts/!adetailer.py
+++ b/scripts/!adetailer.py
@@ -374,7 +374,10 @@ class AfterDetailerScript(scripts.Script):
 
     def get_sampler(self, p, args: ADetailerArgs) -> str:
         if args.ad_use_sampler:
+            if args.ad_sampler == "Use same sampler":
+                return p.sampler_name
             return args.ad_sampler
+
         if hasattr(p, "_ad_orig"):
             return p._ad_orig.sampler_name
         return p.sampler_name