add img2img

app.py

@@ -7,6 +7,67 @@ import json
 from PIL import Image
 import ui_functions as uifn
 
+txt2img_defaults = {
+    'prompt': '',
+    'ddim_steps': 50,
+    'toggles': [1, 2, 3],
+    'sampler_name': 'k_lms',
+    'ddim_eta': 0.0,
+    'n_iter': 1,
+    'batch_size': 1,
+    'cfg_scale': 7.5,
+    'seed': '',
+    'height': 512,
+    'width': 512,
+    'fp': None,
+    'variant_amount': 0.0,
+    'variant_seed': '',
+    'submit_on_enter': 'Yes',
+}
+
+img2img_defaults = {
+    'prompt': '',
+    'ddim_steps': 50,
+    'toggles': [1, 4, 5],
+    'sampler_name': 'k_lms',
+    'ddim_eta': 0.0,
+    'n_iter': 1,
+    'batch_size': 1,
+    'cfg_scale': 5.0,
+    'denoising_strength': 0.75,
+    'mask_mode': 1,
+    'resize_mode': 0,
+    'seed': '',
+    'height': 512,
+    'width': 512,
+    'fp': None,
+}
+sample_img2img = None
+job_manager = None
+RealESRGAN = True
+show_embeddings = False
+
+img2img_resize_modes = [
+    "Just resize",
+    "Crop and resize",
+    "Resize and fill",
+]
+
+img2img_toggles = [
+    'Create prompt matrix (separate multiple prompts using |, and get all combinations of them)',
+    'Normalize Prompt Weights (ensure sum of weights add up to 1.0)',
+    'Loopback (use images from previous batch when creating next batch)',
+    'Random loopback seed',
+    'Save individual images',
+    'Save grid',
+    'Sort samples by prompt',
+    'Write sample info files',
+    'Write sample info to one file',
+    'jpg samples',
+]
+
+img2img_toggle_defaults = [img2img_toggles[i] for i in img2img_defaults['toggles']]
+
 def read_content(file_path: str) -> str:
     """read the content of target file
     """
@@ -60,6 +121,11 @@ def request_images(raw_text, class_draw, style_draw, batch_size):
 
     return images 
 
+def call_JS(sd_method, **kwargs):
+    param_str = json.dumps(kwargs)
+    return f"async (...x) => {{ return await SD.{sd_method}({{ x, ...{param_str} }}) ?? []; }}"
+
+
 examples = [
     '水墨蝴蝶和牡丹花，国画',
     '苍劲有力的墨竹，国画',
@@ -292,14 +358,14 @@ if __name__ == "__main__":
                 #     img2img_image_mask
                 # )
 
-                output_txt2img_copy_to_input_btn.click(
-                    uifn.copy_img_to_input,
-                    [output_txt2img_gallery],
-                    [img2img_image_editor, img2img_image_mask, tabs],
-                    _js=call_JS("moveImageFromGallery",
-                                fromId="txt2img_gallery_output",
-                                toId="img2img_editor")
-                )
+                # output_txt2img_copy_to_input_btn.click(
+                #     uifn.copy_img_to_input,
+                #     [output_txt2img_gallery],
+                #     [img2img_image_editor, img2img_image_mask, tabs],
+                #     _js=call_JS("moveImageFromGallery",
+                #                 fromId="txt2img_gallery_output",
+                #                 toId="img2img_editor")
+                # )
 
                 output_img2img_copy_to_input_btn.click(
                     uifn.copy_img_to_edit,