From cd180add5e93afac23efd3d731c921cc9d501ff0 Mon Sep 17 00:00:00 2001
From: LuqP2 <kelipot.l@gmail.com>
Date: Wed, 24 Dec 2025 02:05:06 -0300
Subject: [PATCH] feat: use ComfyUI exec_time for accurate workflow timing

---
 metahub_save_node.py | 32 +++++++++++++++++++++-----------
 1 file changed, 21 insertions(+), 11 deletions(-)

diff --git a/metahub_save_node.py b/metahub_save_node.py
index 74d95c7..1e79785 100644
--- a/metahub_save_node.py
+++ b/metahub_save_node.py
@@ -145,7 +145,8 @@ class MetaHubSaveNode:
             "hidden": {
                 "prompt": "PROMPT",
                 "extra_pnginfo": "EXTRA_PNGINFO",
-                "unique_id": "UNIQUE_ID"
+                "unique_id": "UNIQUE_ID",
+                "exec_time": "EXEC_TIME"
             },
         }
 
@@ -184,13 +185,10 @@ class MetaHubSaveNode:
         prompt=None,
         extra_pnginfo=None,
         unique_id=None,
+        exec_time=None,
     ):
         global _HINT_SHOWN
 
-        # Start timing for auto-measurement
-        import time
-        start_time = time.time()
-
         try:
             workflow_json = utils.get_workflow_json(extra_pnginfo)
             prompt_data = prompt if isinstance(prompt, dict) else workflow_json.get("prompt", {})
@@ -253,16 +251,28 @@ class MetaHubSaveNode:
 
             height, width = images[0].shape[0], images[0].shape[1]
 
-            # Calculate elapsed time and collect performance metrics
-            elapsed_seconds = time.time() - start_time
+            # Determine final generation time with priority:
+            # 1. Manual override (generation_time_override input)
+            # 2. ComfyUI exec_time (total workflow execution time from ComfyUI)
+            # 3. Legacy generation_time input (deprecated)
+            # 4. Fallback to 0
+            print(f"[MetaHub] exec_time from ComfyUI: {exec_time}")
+            print(f"[MetaHub] generation_time_override: {generation_time_override}")
+            print(f"[MetaHub] generation_time (legacy): {generation_time}")
 
-            # Determine final generation time (use override if provided)
             if generation_time_override is not None:
                 final_time = generation_time_override
-            elif elapsed_seconds > 0:
-                final_time = elapsed_seconds
+                print(f"[MetaHub] Using generation_time_override: {final_time}s")
+            elif exec_time is not None and exec_time > 0:
+                # exec_time is the total workflow execution time from ComfyUI (in seconds)
+                final_time = exec_time
+                print(f"[MetaHub] Using exec_time from ComfyUI: {final_time}s")
+            elif generation_time > 0:
+                final_time = generation_time
+                print(f"[MetaHub] Using legacy generation_time: {final_time}s")
             else:
-                final_time = generation_time or 0.0
+                final_time = 0.0
+                print(f"[MetaHub] No timing data available, using fallback: {final_time}s")
 
             # Collect GPU metrics (auto-detect)
             gpu_metrics = utils.collect_gpu_metrics()