Merge branch 'dev' into feat/interrupted-end

1 files changed, 32 insertions, 0 deletions

diff --git a/scripts/postprocessing_create_flipped_copies.py b/scripts/postprocessing_create_flipped_copies.py
new file mode 100644
index 00000000..b673003b
--- /dev/null
+++ b/scripts/postprocessing_create_flipped_copies.py
@@ -0,0 +1,32 @@
+from PIL import ImageOps, Image

+

+from modules import scripts_postprocessing, ui_components

+import gradio as gr

+

+

+class ScriptPostprocessingCreateFlippedCopies(scripts_postprocessing.ScriptPostprocessing):

+    name = "Create flipped copies"

+    order = 4030

+

+    def ui(self):

+        with ui_components.InputAccordion(False, label="Create flipped copies") as enable:

+            with gr.Row():

+                option = gr.CheckboxGroup(value=["Horizontal"], choices=["Horizontal", "Vertical", "Both"], show_label=False)

+

+        return {

+            "enable": enable,

+            "option": option,

+        }

+

+    def process(self, pp: scripts_postprocessing.PostprocessedImage, enable, option):

+        if not enable:

+            return

+

+        if "Horizontal" in option:

+            pp.extra_images.append(ImageOps.mirror(pp.image))

+

+        if "Vertical" in option:

+            pp.extra_images.append(pp.image.transpose(Image.Transpose.FLIP_TOP_BOTTOM))

+

+        if "Both" in option:

+            pp.extra_images.append(pp.image.transpose(Image.Transpose.FLIP_TOP_BOTTOM).transpose(Image.Transpose.FLIP_LEFT_RIGHT))