add callback after image is saved

1 files changed, 11 insertions, 1 deletions

diff --git a/modules/script_callbacks.py b/modules/script_callbacks.py
index 5bcccd67..5836e4b9 100644
--- a/modules/script_callbacks.py
+++ b/modules/script_callbacks.py
@@ -2,11 +2,12 @@
 callbacks_model_loaded = []

 callbacks_ui_tabs = []

 callbacks_ui_settings = []

-

+callbacks_image_saved = []

 

 def clear_callbacks():

     callbacks_model_loaded.clear()

     callbacks_ui_tabs.clear()

+    callbacks_image_saved.clear()

 

 

 def model_loaded_callback(sd_model):

@@ -28,6 +29,10 @@ def ui_settings_callback():
         callback()

 

 

+def image_saved_callback(image, p, fullfn, txt_fullfn):

+    for callback in callbacks_image_saved:

+        callback(image, p, fullfn, txt_fullfn)

+

 def on_model_loaded(callback):

     """register a function to be called when the stable diffusion model is created; the model is

     passed as an argument"""

@@ -51,3 +56,8 @@ def on_ui_settings(callback):
     """register a function to be called before UI settings are populated; add your settings

     by using shared.opts.add_option(shared.OptionInfo(...)) """

     callbacks_ui_settings.append(callback)

+

+

+def on_save_imaged(callback):

+    """register a function to call after modules.images.save_image is called returning same values, original image and p """

+    callbacks_image_saved.append(callback)