Herman Kwan
82336a25be
- Migrate frontend to modular ES modules (timer, progress, renderer) - Rewrite batch tracking logic to support Sequential Image Loaders and native batches - Introduce robust queue-empty fallback debounce to prevent infinite timers - Decouple timer interactions from progress counters for absolute independence - Redesign UI layout, fix node resizing conflicts, and add dynamic status colors - Update README with new usage instructions and v0.2.0 changelog
65 lines
1.6 KiB
Python
65 lines
1.6 KiB
Python
from aiohttp import web
|
|
from server import PromptServer
|
|
|
|
|
|
class TimerState:
|
|
def __init__(self):
|
|
self.completed_count = 0
|
|
self.total_images = 1
|
|
self.is_active = False
|
|
|
|
|
|
STATE = TimerState()
|
|
|
|
|
|
@PromptServer.instance.routes.post("/batch_timer/reset")
|
|
async def reset_timer(request):
|
|
STATE.completed_count = 0
|
|
STATE.is_active = False
|
|
return web.json_response({"status": "ok"})
|
|
|
|
|
|
class BatchImageTimer:
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"total_images": ("INT", {"default": 1, "min": 1, "max": 999999}),
|
|
},
|
|
"hidden": {
|
|
"unique_id": "UNIQUE_ID",
|
|
},
|
|
}
|
|
|
|
RETURN_TYPES = ()
|
|
FUNCTION = "process"
|
|
CATEGORY = "utils/timer"
|
|
OUTPUT_NODE = True
|
|
|
|
@classmethod
|
|
def IS_CHANGED(cls, **kwargs):
|
|
return float("nan")
|
|
|
|
def process(self, total_images, unique_id=None):
|
|
# Auto-reset if a new batch starts or if total_images changed
|
|
if (
|
|
not STATE.is_active
|
|
or STATE.completed_count >= STATE.total_images
|
|
or STATE.total_images != total_images
|
|
):
|
|
STATE.is_active = True
|
|
STATE.completed_count = 0
|
|
STATE.total_images = total_images
|
|
|
|
STATE.completed_count += 1
|
|
|
|
PromptServer.instance.send_sync(
|
|
"batch_timer_update",
|
|
{
|
|
"current": STATE.completed_count,
|
|
"total": STATE.total_images,
|
|
"node_id": unique_id,
|
|
},
|
|
)
|
|
return ()
|