From f0e2fc131dd919e52fba06b865253b966b399005 Mon Sep 17 00:00:00 2001
From: tazlin <tazlin.on.github@gmail.com>
Date: Thu, 21 Mar 2024 09:15:29 -0400
Subject: [PATCH] fix: callback during each step of post-proc.

---
 hordelib/horde.py | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/hordelib/horde.py b/hordelib/horde.py
index 53582b2c..167119e8 100644
--- a/hordelib/horde.py
+++ b/hordelib/horde.py
@@ -1127,6 +1127,17 @@ def _default_progress_callback(comfyui_progress: ComfyUIProgress, message: str)
                 final_image = ret.image
                 final_rawpng = ret.rawpng
 
+                if progress_callback is not None:
+                    try:
+                        progress_callback(
+                            ProgressReport(
+                                hordelib_progress_state=ProgressState.progress,
+                                hordelib_message="Post Processing new image.",
+                            ),
+                        )
+                    except Exception as e:
+                        logger.error(f"Progress callback failed ({type(e)}): {e}")
+
                 # Ensure facefixers always happen first
                 post_processing_requested = sorted(
                     post_processing_requested,