AutoFinish preview_master: node waits for the worker (cas/hires routes) and shows the REAL master as its canvas preview; rtx mode auto-skips (deadlock guard)

joyecho_autofinish.py

@@ -73,6 +73,17 @@ class JoyEcho_AutoFinish:
                                "detail per frame. Ignored when the run produced "
                                "shot_hires masters (those are used directly).",
                 }),
+                "preview_master": ("BOOLEAN", {
+                    "default": True,
+                    "tooltip": "Wait for the background worker and show the REAL "
+                               "finished MASTER as this node's canvas preview "
+                               "(the in-graph SaveVideo preview is a bit-starved "
+                               "re-encode and lies about quality). Adds ~30-90s "
+                               "after the render while the master is assembled. "
+                               "Auto-disabled in rtx mode - the RTX job queues "
+                               "behind this graph item, so waiting would "
+                               "deadlock the queue.",
+                }),
             },
         }
 
@@ -84,7 +95,7 @@ class JoyEcho_AutoFinish:
 
     def run(self, images, enabled, master_name, scale_factor, quality,
             batch_size, shots_subdir="joyecho",
-            upscale_mode="bicubic+cas (deterministic)"):
+            upscale_mode="bicubic+cas (deterministic)", preview_master=True):
         if not enabled:
             return (images,)
         out_root = (folder_paths.get_output_directory()
@@ -114,6 +125,45 @@ class JoyEcho_AutoFinish:
               f"(scale {scale_factor}); upscale jobs will queue after this "
               f"item. Progress: {shots_dir}\\_autofinish_<stamp>.log",
               flush=True)
+
+        # Optionally wait for the worker and preview the REAL master in-canvas.
+        # Never in rtx mode: the worker's RTX job queues behind THIS graph item,
+        # so blocking here would deadlock the queue. cas/hires routes are
+        # self-contained (ffmpeg only) and finish in seconds-to-minutes.
+        is_rtx = str(upscale_mode).lower().startswith("rtx")
+        if preview_master and not is_rtx:
+            import glob as _glob
+            import time as _time
+            t_spawn = _time.time()
+            deadline = t_spawn + 900
+            master, stable_size = None, -1
+            print("[JoyEcho] AutoFinish: waiting for the master to preview it "
+                  "in-canvas (preview_master=True, up to 15 min)...", flush=True)
+            while _time.time() < deadline:
+                cands = [p for p in _glob.glob(os.path.join(shots_dir, "*_MASTER.mp4"))
+                         if os.path.getmtime(p) >= t_spawn - 2]
+                if cands:
+                    p = max(cands, key=os.path.getmtime)
+                    sz = os.path.getsize(p)
+                    if sz > 0 and sz == stable_size:
+                        master = p
+                        break
+                    stable_size = sz
+                _time.sleep(2)
+            if master:
+                print(f"[JoyEcho] AutoFinish: master ready -> {master}", flush=True)
+                return {"ui": {"images": [{"filename": os.path.basename(master),
+                                           "subfolder": shots_subdir,
+                                           "type": "output"}],
+                               "animated": (True,)},
+                        "result": (images,)}
+            print("[JoyEcho] AutoFinish: master did not appear within the wait "
+                  "window - check the _autofinish log; continuing without the "
+                  "in-canvas preview.", flush=True)
+        elif preview_master and is_rtx:
+            print("[JoyEcho] AutoFinish: preview_master skipped in rtx mode "
+                  "(the RTX job runs after this graph item; waiting would "
+                  "deadlock the queue).", flush=True)
         return (images,)