From f3bf178abc1dff7c20c69562ff56c8867efbb284 Mon Sep 17 00:00:00 2001 From: nofunstudio Date: Wed, 9 Apr 2025 10:53:10 -0600 Subject: [PATCH] change file save path test --- __pycache__/ksampler_fun.cpython-312.pyc | Bin 7832 -> 9093 bytes ksampler_fun.py | 32 ++++++++++++++++++++--- 2 files changed, 28 insertions(+), 4 deletions(-) diff --git a/__pycache__/ksampler_fun.cpython-312.pyc b/__pycache__/ksampler_fun.cpython-312.pyc index fd9340c9882628fc1848d4c5eee8a7f7c86a6329..4b1a53c3e8d7d0eeb56837e74f53cab2977e3d2b 100644 GIT binary patch delta 2568 zcmaJ@Z%iBK8Nb(@|DEyqzy^#lU`zt>5)8>gAq3L2OCU)T5|XTmSk%kmI~+TQP2M{v zIJ3*NQLU>w3G{vlT@)2*szgMkIHHyEG3|#|>a@uWPq14jO=|n0(mrfsTBiNc_CDvj zq*2oM!|#3G-}C4F_q@;D;?Qd&yME<%a|~_0@BAZvwUuE$gF&U)%L3Z^2rcZIYGhOueC6WE7?JQ;|PSueRNPKTNU4J=8k0redWuTtWDvPdq`Wgx3{|z ze7jeC73-&X+AE$z-|C5*={#JrXFF;|Qhb!(NA@@aFQ<^~_~sO{M|RGk@qmRHVPtNO zkzFg6vg3G@$yHO%{WTYqc_o-Y1mt|O-uTffzZviXBsFo2~F3iNyjry5?bg6PUzBvstkmNV>RY# z@BjBw6>^htcEaYQ1m?(xY!cOzF6)Up4-f(X*y36MFL~2?m^%F0df4Ryq7J}L{I;K1 z4^Xjwb;BT<9%XORPV6Iwr-Dc(LJ8a&g!%3Sh*#nOUeHXS} z!{t1cx$*+tXfLqsLXMZ+@A513Orar0?<-3TBYScJsrZ+0SP_e1r*eX(x53BKQ`1<{G*MNqE9y=EIqAA0 zhbVPDq1*^1rlhzM($fQ>NU)qPW{&KnQE&hzw5QTy980nVe~;Ms9X#yOu*pg+Ei2eT z`uVO|JOrFKuIQphgNtTN5nc~|9^wQ6AeRh>B+H_%%;-jyrenjB)(jTx3^tigr0C|T zJ3c}!BQ^Lac%3x8WF?hOXo_s08GH&DK9<2)N$H|0=?ZzDZ=A)WzS|Lyj56*i&J%4%QyuNl`-}JR_`uw*}FPz@= zHQqY0aANV=!l}o^EvNPSsCG>lD+=MIW6SYE_F?zm1t!>j`@pEtRo!RCqxEeuQ*NNE`y;~nCF1QAOQfSxhc#sf3rBP656V7w+@vp!>hN1xCOP4 z|J4RvS!(m8S<`sZ;eB5CAsyo!Knp;2HRH9T;1jHw6;@0yPo@@&zEMlvQ!H?n{JVb6 z1~muQlWR@QJ$pdh3jkS+%~A`WDGcACh!r=`&Tweb)U>K#4QI%28+M|8^3R42q^`F5 zUqwFHSHyyU3{C)4?jrXa-$ee^Ljf0Re>@AHf&NXMr)nHUJy^eWF9`YK|a#ETyO4e|K_Iu=$bpa?(X>!_j|@#baXCrOSz(Le~IZe z$C2gmz0SLx_o8>B4+87oIZ})qE!u`l%!sLt7RSzQj9pwCySOnXu8oPs%gNH%)#dCm zzL&k5eGqtf_Tvj5Tqt#)Dqg!@jNT~PW=f1VyTyi4!=q!U8x8%Hq3~#eBwBW}a4r7_ DndWg- delta 1447 zcmZ{kU1%d!6oBXEZ<5S3Nz$;jP#NFDhtz9K7cB#^rVVaqyKgmqI zGg)^_cVEnc4?c9c54ND}YgYveK@b!~@X37#*GJ=y=ct|JKFTkQQGj%N=S+4MCr{=8=SZI6)$|py;|`Db~87n?)G=7ph8G zYrg3zd5Oq=@vj7e;7CL4a$tib*=K=^$p|Wxea!gl7Q5-g@4(1(rYS+_V$o?_35dU^`9 zB{a2K00b|`k;AJVc2~@jO!HUqfW%e)UGMdf)z@;{=x6uE4w7T9MLN9P#t`<=P^((k z!`u9ZggO#tPsZ}TU=g-ai46~L`6g<5SxUMoJ9!R5Oxg{wdEJo9R8i>+OSBJ0a6@)L z)k=n{QJ&4UkI&d1!?b-h#agp{8^*d$&!LacRIX@}Kxf^7rPb`9s_DkMsi`)(Y6}(| zPDM7sV6vauhb5cHc5rpQqUlOib0v&v3J}1lkqfBQ&G;~nYnaCzYsNo%wRd#?%z=0R z@q^i~Qd2v@A5uNH^LzQ*g}uVQ@5AVw=*OqNPR;BDzl$Z`4&Mszp8X~^c+}0svpbpJ z+Bk9aNOa_5vkhW??!7QDJL&ym(my-JmJ(9oc~DXXaEGIKFf@&c~`4SV!PA zy4l?8;z|01_7yPMernWp6+AvYRrDnP!-HvxBBmtyMH?(lUFH%8@)vr-#Vp=7sbPlE4Z`9%|tv1ju9T+n%>bS97ITAR{M2 z5{gemia57`p2%eW|1vQ#$|FY(qFePY&~HUwTyztha1*VuKQeRAU}8ndORqprUq|4N zLlWr<$|Zyfu!T}ZG0m!Gtr;raX7#?8doe225dWNS8Hb2P_H*CDl4HQ(mN-|}`&t=D zytCe1u{ih0nn^n>%d2&(PBmGk@NOTzq)5rAu3meXHI%oLHM04_d9d4}4$nzm(<(Kr z`yx&$Ac}}f=;6Jz@DjzJrBhM&{*~Eu|I-8i&ipu7s2Xa$qMf5V;7#B&HMZM77C-d$ Xldhu|NQUI@bAY26wlpy6!>#=VJ@-q& diff --git a/ksampler_fun.py b/ksampler_fun.py index 6f2615d..d77c9fb 100644 --- a/ksampler_fun.py +++ b/ksampler_fun.py @@ -10,6 +10,14 @@ import os import folder_paths import numpy as np from PIL import Image +from PIL.PngImagePlugin import PngInfo +import json + +# Add debug print for output directory +output_dir = folder_paths.get_output_directory() +print(f"Debug - Output directory path: {output_dir}") +print(f"Debug - Output directory exists: {os.path.exists(output_dir)}") +print(f"Debug - Output directory is writable: {os.access(output_dir, os.W_OK) if os.path.exists(output_dir) else False}") def fun_ksampler(model, seed, steps, cfg, sampler_name, scheduler, positive, negative, latent, vae, denoise=1.0, disable_noise=False, start_step=None, last_step=None, force_full_denoise=False, preview_steps=5, skip_steps=0): latent_image = latent["samples"] @@ -75,10 +83,26 @@ class PreviewCallback: image = (image * 255).astype(np.uint8) image = Image.fromarray(image[0]) - # Save the image - filename = f"preview_step_{step:03d}.png" - filepath = os.path.join(self.output_dir, filename) - image.save(filepath) + # Use the same approach as SaveImage 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, + self.output_dir, + image.size[0], # width + image.size[1] # height + ) + + # Create metadata + metadata = PngInfo() + metadata.add_text("step", str(step)) + metadata.add_text("total_steps", str(total_steps)) + + # 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) + + print(f"Saved preview image to: {filepath}") class KSampler: @classmethod