36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
import server
|
|
from aiohttp import web
|
|
from .nodes import PSampler, PSamplerAdvanced, PSamplerCustom, PSamplerCustomAdvanced
|
|
from .shared import PAUSE_STATE, state_lock
|
|
|
|
# API Route
|
|
@server.PromptServer.instance.routes.post("/comfy/pause_signal")
|
|
async def set_pause_command(request):
|
|
json_data = await request.json()
|
|
command = json_data.get("command", "PROCEED")
|
|
|
|
with state_lock:
|
|
PAUSE_STATE["command"] = command
|
|
|
|
print(f"[P-Sampler] 📡 Signal Received: {command}")
|
|
return web.json_response({"status": "success"})
|
|
|
|
# Node Mappings
|
|
NODE_CLASS_MAPPINGS = {
|
|
"PSampler": PSampler,
|
|
"PSamplerAdvanced": PSamplerAdvanced,
|
|
"PSamplerCustom": PSamplerCustom,
|
|
"PSamplerCustomAdvanced": PSamplerCustomAdvanced
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"PSampler": "⏸️ PSampler",
|
|
"PSamplerAdvanced": "⏸️ PSampler (Advanced)",
|
|
"PSamplerCustom": "⏸ PSampler (Custom)",
|
|
"PSamplerCustomAdvanced": "⏸ PSampler (Custom Advanced)"
|
|
}
|
|
|
|
WEB_DIRECTORY = "./js"
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
|