o · Ïiðã@s>ddlmZddlZddlZddlZGdd„dƒZdeiZdS)é)ÚImageNc@s<eZdZedd„ƒZdZdZdZdZdd„Z ed d „ƒZ d S) ÚSaveImageWebsocketcCs dddiiS)NÚrequiredÚimages)ÚIMAGE©)Úsrrú?/mnt/c/Users/fbmor/ComfyUI/custom_nodes/websocket_image_save.pyÚ INPUT_TYPESsÿzSaveImageWebsocket.INPUT_TYPESrÚ save_imagesTz api/imagecCsrtj |jd¡}d}|D])}d| ¡ ¡}t t  |dd¡  tj ¡¡}|  ||jdd|df¡|d7}q iS)Nrgào@éÿÚPNGé) ÚcomfyÚutilsÚ ProgressBarÚshapeÚcpuÚnumpyrÚ fromarrayÚnpÚclipÚastypeÚuint8Úupdate_absolute)ÚselfrÚpbarÚstepÚimageÚiÚimgrrr r s zSaveImageWebsocket.save_imagescCst ¡S)N)Útime)rrrrr Ú IS_CHANGED&szSaveImageWebsocket.IS_CHANGEDN) Ú__name__Ú __module__Ú __qualname__Ú classmethodr Ú RETURN_TYPESÚFUNCTIONÚ OUTPUT_NODEÚCATEGORYr r"rrrr r s  r) ÚPILrrrÚ comfy.utilsrr!rÚNODE_CLASS_MAPPINGSrrrr Ús  ÿ