feat: support registering InfiniteZoom result to gallery

iz_helpers/run.py

@@ -7,6 +7,8 @@
 from modules.paths_internal import script_path
 from modules.paths import Paths
 from modules.model_info import AllModelInfo
+from modules import script_callbacks
+
 from .helpers import (
     fix_env_Path_ffprobe,
     closest_upper_divisible_by_eight,
@@ -101,6 +103,7 @@ def outpaint_steps(
     main_frames = [init_img.convert("RGB")]
 
     processed = None
+    p = None
 
     for i in range(outpaint_steps):
         print_out = (
@@ -130,7 +133,7 @@ def outpaint_steps(
             save2Collect(current_image, out_config, f"exit_img.png")
         else:
             pr = prompts[max(k for k in prompts.keys() if k <= i)]
-            processed, newseed = renderImg2Img(
+            processed, newseed, p = renderImg2Img(
                 request,
                 f"{common_prompt_pre}\n{pr}\n{common_prompt_suf}".strip(),
                 negative_prompt,
@@ -173,7 +176,8 @@ def outpaint_steps(
             main_frames[i] = corrected_frame
         # else :TEST
         # current_image.paste(prev_image, mask=prev_image)
-    return main_frames, processed
+
+    return main_frames, processed, p
 
 
 def create_zoom(
@@ -392,7 +396,7 @@ def create_zoom_single(
         )
 
         pr = prompts[min(k for k in prompts.keys() if k >= 0)]
-        processed, newseed = renderTxt2Img(
+        processed, newseed, p = renderTxt2Img(
             request,
             f"{common_prompt_pre}\n{pr}\n{common_prompt_suf}".strip(),
             negative_prompt,
@@ -404,6 +408,7 @@ def create_zoom_single(
             height,
             all_model_info,
         )
+
         if len(processed.images) > 0:
             current_image = processed.images[0]
             save2Collect(current_image, out_config, f"init_txt2img.png")
@@ -426,7 +431,7 @@ def create_zoom_single(
         all_model_info,
         specified_model=main_sd_model
     )
-    main_frames, processed = outpaint_steps(
+    main_frames, processed, p = outpaint_steps(
         request,
         width,
         height,
@@ -451,6 +456,7 @@ def create_zoom_single(
         custom_exit_image,
         all_model_info,
     )
+
     all_frames.append(
         do_upscaleImg(main_frames[0], upscale_do, upscaler_name, upscale_by)
         if upscale_do
@@ -543,6 +549,12 @@ def create_zoom_single(
         int(video_start_frame_dupe_amount),
         int(video_last_frame_dupe_amount),
     )
+    p.feature = "INFINITE_ZOOM"
+    script_callbacks.image_saved_callback(
+        script_callbacks.ImageSaveParams(
+            None, p, out_config["video_filename"], {"parameters": processed.info}
+        )
+    )
     print("Video saved in: " + os.path.join(script_path, out_config["video_filename"]))
     return (
         out_config["video_filename"],

iz_helpers/sd_helpers.py

@@ -32,6 +32,9 @@ def renderTxt2Img(
     )
     p.set_request(request)
     p.set_all_model_info(all_model_info)
+    p.do_not_save_samples = True
+    p.do_not_save_grid = True
+
     with monitor_call_context(
         request,
         get_function_name_from_processing(p),
@@ -42,7 +45,7 @@ def renderTxt2Img(
     ):
         processed = process_images(p)
     newseed = p.seed
-    return processed, newseed
+    return processed, newseed, p
 
 
 def renderImg2Img(
@@ -90,6 +93,8 @@ def renderImg2Img(
     # p.latent_mask = Image.new("RGB", (p.width, p.height), "white")
     p.set_request(request)
     p.set_all_model_info(all_model_info)
+    p.do_not_save_samples = True
+    p.do_not_save_grid = True
 
     with monitor_call_context(
         request,
@@ -106,4 +111,4 @@ def renderImg2Img(
         print("\nGrid image detected applying patch")
 
     newseed = p.seed
-    return processed, newseed
+    return processed, newseed, p