From 786b52bd46e7d86eb6454af2d53313ec637a1dec Mon Sep 17 00:00:00 2001
From: space-nuko <24979496+space-nuko@users.noreply.github.com>
Date: Fri, 17 Mar 2023 20:45:48 -0400
Subject: [PATCH 1/3] Enable transferring settings with "Send to img2img"

---
 scripts/two_shot.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/scripts/two_shot.py b/scripts/two_shot.py
index 0727bd6..35aa559 100644
--- a/scripts/two_shot.py
+++ b/scripts/two_shot.py
@@ -139,6 +139,7 @@ def ui(self, is_img2img):
         self.infotext_fields = [
             (extra_generation_params, "Latent Couple")
         ]
+        self.paste_field_names = ["Latent Couple"]
         return enabled, divisions, positions, weights, end_at_step
 
     def denoised_callback(self, params: CFGDenoisedParams):

From e2510b2e863184e93549a53825dca5e3fc3e00a2 Mon Sep 17 00:00:00 2001
From: space-nuko <24979496+space-nuko@users.noreply.github.com>
Date: Fri, 17 Mar 2023 20:49:30 -0400
Subject: [PATCH 2/3] Add copy button

---
 scripts/two_shot.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/scripts/two_shot.py b/scripts/two_shot.py
index 35aa559..ba30667 100644
--- a/scripts/two_shot.py
+++ b/scripts/two_shot.py
@@ -113,6 +113,12 @@ def do_apply(self, extra_generation_params: str):
 
         return raw_params.get('divisions', '1:1,1:2,1:2'), raw_params.get('positions', '0:0,0:0,0:1'), raw_params.get('weights', '0.2,0.8,0.8'), int(raw_params.get('step', '20'))
 
+    def do_copy(self, raw_divisions: str, raw_positions: str, raw_weights: str, raw_end_at_step: int):
+        #
+        # copy "Latent Couple" extra_generation_params to textbox
+        #
+        return f"divisions={raw_divisions} positions={raw_positions} weights={raw_weights} end at step={raw_end_at_step}"
+
     def ui(self, is_img2img):
         id_part = "img2img" if is_img2img else "txt2img"
 
@@ -132,9 +138,12 @@ def ui(self, is_img2img):
                 visualize_button.click(fn=self.do_visualize, inputs=[divisions, positions, weights], outputs=[visual_regions])
 
                 extra_generation_params = gr.Textbox(label="Extra generation params")
-                apply_button = gr.Button(value="Apply")
+                with gr.Column():
+                    apply_button = gr.Button(value="Apply")
+                    copy_button = gr.Button(value="Copy")
 
                 apply_button.click(fn=self.do_apply, inputs=[extra_generation_params], outputs=[divisions, positions, weights, end_at_step])
+                copy_button.click(fn=self.do_copy, inputs=[divisions, positions, weights, end_at_step], outputs=[extra_generation_params])
 
         self.infotext_fields = [
             (extra_generation_params, "Latent Couple")

From 2d84a66a7b1c41560137a4894d71d89a70593d8b Mon Sep 17 00:00:00 2001
From: space-nuko <24979496+space-nuko@users.noreply.github.com>
Date: Fri, 17 Mar 2023 20:59:08 -0400
Subject: [PATCH 3/3] Fix UI

---
 scripts/two_shot.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/scripts/two_shot.py b/scripts/two_shot.py
index ba30667..db5d384 100644
--- a/scripts/two_shot.py
+++ b/scripts/two_shot.py
@@ -138,8 +138,8 @@ def ui(self, is_img2img):
                 visualize_button.click(fn=self.do_visualize, inputs=[divisions, positions, weights], outputs=[visual_regions])
 
                 extra_generation_params = gr.Textbox(label="Extra generation params")
-                with gr.Column():
-                    apply_button = gr.Button(value="Apply")
+                with gr.Row():
+                    apply_button = gr.Button(value="Apply", variant="primary")
                     copy_button = gr.Button(value="Copy")
 
                 apply_button.click(fn=self.do_apply, inputs=[extra_generation_params], outputs=[divisions, positions, weights, end_at_step])