Clean up prior mask previews before writing new one

The preview node wrote a fresh timestamped PNG per run without ever
deleting the previous ones, so the temp directory would accumulate a
file per preview call over a session. Delete prior ssd_mask_*.png
files at the start of each preview so at most one mask image exists
on disk at a time. Keep the timestamped filename so the browser does
not cache a stale image.
This commit is contained in:
capacap
2026-04-21 15:25:42 +02:00
parent 3904d877f3
commit 20a2ebcd4c
+6
View File
@@ -1,5 +1,6 @@
from __future__ import annotations
import glob
import os
import time
@@ -167,6 +168,11 @@ class SelectiveSigmaDetailerMaskPreviewNode:
out_dir = folder_paths.get_temp_directory()
os.makedirs(out_dir, exist_ok=True)
for old in glob.glob(os.path.join(out_dir, "ssd_mask_*.png")):
try:
os.remove(old)
except OSError:
pass
filename = f"ssd_mask_{int(time.time() * 1000)}.png"
path = os.path.join(out_dir, filename)
arr = (img[0].numpy() * 255).clip(0, 255).astype(np.uint8)