From 45f86661e31ea55e72383261f29c0d29b3e0adaa Mon Sep 17 00:00:00 2001
From: nofunstudio <jimmy@nofun.io>
Date: Wed, 9 Apr 2025 11:04:47 -0600
Subject: [PATCH] updates to work with comfydeploy structure

---
 __pycache__/ksampler_fun.cpython-312.pyc | Bin 9093 -> 9348 bytes
 ksampler_fun.py                          |  21 +++++++++++++++++----
 2 files changed, 17 insertions(+), 4 deletions(-)

diff --git a/__pycache__/ksampler_fun.cpython-312.pyc b/__pycache__/ksampler_fun.cpython-312.pyc
index 4b1a53c3e8d7d0eeb56837e74f53cab2977e3d2b..71d151d3cd22f761aa470cdb77285ef3b1223123 100644
GIT binary patch
delta 1172
zcmZuvU1%It6rQ^~`#-zcT{piw$?m2}aSZX7ineKpDcxn8?NXakYXzrc?%mCV*_q+a
zY`Q5c1~t;sKvQl-1415^)`!$B=!1$r2^J(`&~+XJAAImZNKqQ?llN{IgW?Q)=R4my
z_ug;L+?B$c#n|0wG|bWG&-d?F&*g5%f?>}=K>EUZ34$gk2Uv|g&_lD}9;ca?W+6AT
zyIr=><Ffq-`6f>`yHlQpKJs~Z@MP32HMxZWm+!}A!5(xo=(`2>IN+A(EBVWg->BPW
z-FCQNd0Ha<ky+41YLVk3yuD;1Cu(9;Q5&*inn;IEY|f}r#j<1#Eh3HFj@$rEk{6!r
zZKuS`{A#E@n7Pbv2M5|iotP#K;d!t{z7~eT-PUiyi1!q=`Lt_jKYkd!<#nw!yP8qj
z^~`?NWiST2-s4|Q*hl_R;$$p-6eNfmKUI)CPQcWws$MbhaoXX;sMkdmU6j?DQbm@;
zJHd*oAzi5<CuG%MsTdkW7$0N3p}j%KDkML~)7c?>h^ly;!6*YZTPL6!auq8O7fC8{
zAw9*khpNbyEoBkyog$lxJ&&i!wZu~rGtt={pP)=i;u$8KW~+h_>4s_{=zs>EB|j&|
zz!?%vK0JZXv2tL}sOvWM_F|{GhHM2Yw&L*2zY}8EImq$Tzfsq0D@psZ1Sb#si<DV+
zS|25g;5Bh*qp)7sD6N;aq$8i4`A(c(3+?#1M9(|n>*05w{b2qJ`-j5$Zwu$w!rSqe
zc7j}Pw4HmTEoL{0>&15NiF+a6KyuChM~p7n>U(^%e>1YxGqD{%x_c6E$=q6IC&~$9
zyF!OQu-IMv(wpxph~BS6UtyG-O-cR|74afjNgWIcJ_Y<Pj{7ECcT!h9V3|zy7UQq8
z`Wp;}7%aEedjmlHkJ_1;57VJQ+VN>$+0*?$9e9~!(%;h?c|TnOORZz#3h-ZHn_g*s
zGx#w8Gvuud25*zUGUZ3FGJTbShk_*Fi;OoIEKzVGbDCmVbnx>A#MenZ`$GRERy@pr
z-A&wdp+SDhmNP89B;@OQc4k(onHs`U0JBr_<#HaclG8c$K%S<Kpc@d$^GGuhws3{*
zGR5F!1{cUrx$*dOj7~FP7YA2KdU%9>*3|H!fq$EL5S*wPu&$v=e3M)o{wzEJChu_+
LcAMnvfye#^?CLF|

delta 889
zcmZuvT}TvB6y7`j>^jcs?CR>OtFEqui=<JBX0Ds1YmzRRSVGV?I#*qt-P!KUN|p~H
z3Q-iw3DRC9`XR6fWgiSeuqTNKghXPXH;H@*`k@Tzp?ehrqciZ`@1FDB@60{i@z#l!
zj5k&*Z=zBC<Y)g-@j`}$=YE^O$wvh}ljT_qim;HAg%GUrxyTNm_`-&rGpsgh8q8+v
z{<tCNxqBIzn?t)CnMUa6O$x%f^iCwe)AVK#1wa@c3y0ye(19L+-&)GekHRA7om#de
z?peQa`LgwS#$3{#g<kX5I1|=I9}*$kcGkU_Ow(S@Fvp_(;b@;~Wb~?$7{QtrQt&WV
zV9r*6T(Dv*LZ{%P?Q(&S>W!>^tcNsd7>B|UsUL@8l0HEEAlZ);H!(jOX{HNlV_J)h
zqiR^gvVpFWc384kpdf78%YBBWFRb9G6v2kMSB*z?OpG)w-qWWlGNqOX)+JfeB_o;X
zcGawrW}~*yo>i9`phTNY+{tQ17Yhn!Yi4R@>u2g$%j;e=y(?&(w7eG`_xUM)y7pQ3
zvNXxBiKjP2$Ex$tymOvk%{jg%)^8m%*-IvizgSH|#g?$cw>2xZeU;RFv;`%tpp+IK
zx*Xn}UPOg)$APr86bXIL@uc_?3CC>?jey_TDqg3D$S8v%1|x}Sry05a$?gm!h0x2&
zRPiz|$T|L_AR|!ZT1RpC;c7!l;$-0jN*!aSV~Mv#izu1R79P49;kCOG-GpE6VC6U~
zPcYypcm)z;{0aj?!AS2_Bu%3e9Z+R*3-po;c^WI$F{ot_--998C<zv`?KL3~kGjuv
zN|Bg?iPub6mq0KWAQRA08upx^TSJJdG7b%3C5DMcF3}WWPlTLj&<!6-4~h+p`Wdjt
sK|+x4DWkXZd#dyI|HQVXKBCHT1vijKaM!cMS0Ue56NRlF=qjuF1H;wtEdT%j

diff --git a/ksampler_fun.py b/ksampler_fun.py
index d77c9fb..2f0a347 100644
--- a/ksampler_fun.py
+++ b/ksampler_fun.py
@@ -62,6 +62,9 @@ class PreviewCallback:
         self.vae = vae
         self.model = model
         self.output_dir = folder_paths.get_output_directory()
+        self.type = "output"
+        self.prefix_append = ""
+        self.compress_level = 4
         # Initialize previewer using the model's latent_format
         self.previewer = latent_preview.get_previewer("cpu", model.model.latent_format)
         if self.previewer is None:
@@ -83,7 +86,7 @@ class PreviewCallback:
                     image = (image * 255).astype(np.uint8)
                     image = Image.fromarray(image[0])
                 
-                # Use the same approach as SaveImage for saving files
+                # Use the same approach as ComfyDeployOutputImage for saving files
                 filename_prefix = f"preview_step_{step:03d}"
                 full_output_folder, filename, counter, subfolder, filename_prefix = folder_paths.get_save_image_path(
                     filename_prefix, 
@@ -99,10 +102,20 @@ class PreviewCallback:
                 
                 # Save the image with metadata
                 file = f"{filename}_{counter:05}_.png"
-                filepath = os.path.join(full_output_folder, file)
-                image.save(filepath, pnginfo=metadata, compress_level=4)
+                file_path = os.path.join(full_output_folder, file)
                 
-                print(f"Saved preview image to: {filepath}")
+                # Save with PNG compression
+                image.save(file_path, pnginfo=metadata, compress_level=self.compress_level)
+                
+                # Create result entry similar to ComfyDeployOutputImage
+                result = {
+                    "filename": file,
+                    "subfolder": subfolder,
+                    "type": self.type,
+                    "output_id": "preview_images"
+                }
+                
+                print(f"Saved preview image to: {file_path}")
 
 class KSampler:
     @classmethod