add an option to enable sections from extras tab in txt2img/img2img

fix some style inconsistenices

1 files changed, 42 insertions, 0 deletions

diff --git a/modules/scripts_auto_postprocessing.py b/modules/scripts_auto_postprocessing.py
new file mode 100644
index 00000000..30d6d658
--- /dev/null
+++ b/modules/scripts_auto_postprocessing.py
@@ -0,0 +1,42 @@
+from modules import scripts, scripts_postprocessing, shared

+

+

+class ScriptPostprocessingForMainUI(scripts.Script):

+    def __init__(self, script_postproc):

+        self.script: scripts_postprocessing.ScriptPostprocessing = script_postproc

+        self.postprocessing_controls = None

+

+    def title(self):

+        return self.script.name

+

+    def show(self, is_img2img):

+        return scripts.AlwaysVisible

+

+    def ui(self, is_img2img):

+        self.postprocessing_controls = self.script.ui()

+        return self.postprocessing_controls.values()

+

+    def postprocess_image(self, p, script_pp, *args):

+        args_dict = {k: v for k, v in zip(self.postprocessing_controls, args)}

+

+        pp = scripts_postprocessing.PostprocessedImage(script_pp.image)

+        pp.info = {}

+        self.script.process(pp, **args_dict)

+        p.extra_generation_params.update(pp.info)

+        script_pp.image = pp.image

+

+

+def create_auto_preprocessing_script_data():

+    from modules import scripts

+

+    res = []

+

+    for name in shared.opts.postprocessing_enable_in_main_ui:

+        script = next(iter([x for x in scripts.postprocessing_scripts_data if x.script_class.name == name]), None)

+        if script is None:

+            continue

+

+        constructor = lambda s=script: ScriptPostprocessingForMainUI(s.script_class())

+        res.append(scripts.ScriptClassData(script_class=constructor, path=script.path, basedir=script.basedir, module=script.module))

+

+    return res