From cd462c4a5ad8f34755e112e9a5e4d9b2c93fbb20 Mon Sep 17 00:00:00 2001
From: Bingsu <ks2515@naver.com>
Date: Thu, 27 Apr 2023 14:58:57 +0900
Subject: [PATCH] fix: extraparams, params_txt, sd_model

---
 scripts/!adetailer.py | 49 +++++++++++++++++++++----------------------
 1 file changed, 24 insertions(+), 25 deletions(-)

diff --git a/scripts/!adetailer.py b/scripts/!adetailer.py
index d9bbf07..4a3ec18 100644
--- a/scripts/!adetailer.py
+++ b/scripts/!adetailer.py
@@ -10,7 +10,11 @@ from adetailer import __version__, get_models, mediapipe_predict, ultralytics_pr
 from adetailer.common import dilate_erode, is_all_black, offset
 from modules import images, safe, script_callbacks, scripts, shared
 from modules.paths import data_path, models_path
-from modules.processing import StableDiffusionProcessingImg2Img, process_images
+from modules.processing import (
+    StableDiffusionProcessingImg2Img,
+    create_infotext,
+    process_images,
+)
 from modules.shared import opts, state
 
 AFTER_DETAILER = "After Detailer"
@@ -55,26 +59,6 @@ class ADetailerArgs:
         }
 
 
-def with_params_txt(func):
-    params_txt = Path(data_path, "params.txt")
-    original_params = ""
-
-    def wrapper(*args, **kwargs):
-        nonlocal original_params
-
-        if not original_params and params_txt.exists():
-            original_params = params_txt.read_text(encoding="utf-8")
-
-        result = func(*args, **kwargs)
-
-        if original_params:
-            params_txt.write_text(original_params, encoding="utf-8")
-
-        return result
-
-    return wrapper
-
-
 class ChangeTorchLoad:
     def __enter__(self):
         self.orig = torch.load
@@ -286,18 +270,30 @@ class AfterDetailerScript(scripts.Script):
     def get_args(*args):
         return ADetailerArgs(*args)
 
-    @with_params_txt
+    def write_params_txt(self, p):
+        infotext = create_infotext(
+            p, p.all_prompts, p.all_seeds, p.all_subseeds, None, 0, 0
+        )
+        params_txt = Path(data_path, "params.txt")
+        params_txt.write_text(infotext, encoding="utf-8")
+
+    def process(self, p, *args_):
+        self.args = self.get_args(*args_)
+        if self.args.ad_model != "None":
+            extra_params = self.extra_params(**self.args.asdict())
+            p.extra_generation_params.update(extra_params)
+
     def postprocess_image(self, p, pp, *args_):
         if getattr(p, "_disable_adetailer", False):
             return
 
-        args = self.get_args(*args_)
+        if not hasattr(self, "args"):
+            self.args = self.get_args(*args_)
+        args = self.args
 
         if args.ad_model == "None":
             return
 
-        extra_params = self.extra_params(**args.asdict())
-        p.extra_generation_params.update(extra_params)
         p._idx = getattr(p, "_idx", -1) + 1
         i = p._idx
 
@@ -401,6 +397,9 @@ class AfterDetailerScript(scripts.Script):
         if processed is not None:
             pp.image = processed.images[0]
 
+        if i == len(p.all_prompts) - 1:
+            self.write_params_txt(p)
+
 
 def on_ui_settings():
     section = ("ADetailer", AFTER_DETAILER)