+jTddlmZddlZddlZddlZGddZdeiZddiZ dS))ImageNcPeZdZedZdZdZdZdZdZ edZ dS) SaveImageWebsocketcdddiiS)Nrequiredimages)IMAGE)ss A/home/wildlama/comfy/ComfyUI/custom_nodes/websocket_image_save.py INPUT_TYPESzSaveImageWebsocket.INPUT_TYPESs{, r save_imagesTimagectj|jd}d}|D]}d|z}t jtj |dd tj }| ||jdd|df|dz }iS)Nrgo@PNG) comfyutils ProgressBarshapecpunumpyr fromarraynpclipastypeuint8update_absolute)selfrpbarstepriimgs r rzSaveImageWebsocket.save_imagess{&&v|A77  Euyy{{((***A/"'!Q"4"4";";BH"E"EFFC  v|AT8J K K K AIDD rc(tjS)N)time)r rs r IS_CHANGEDzSaveImageWebsocket.IS_CHANGED&sy{{rN) __name__ __module__ __qualname__ classmethodr RETURN_TYPESFUNCTION OUTPUT_NODECATEGORYrr(r rr rr sm[ LHKH   [rrzSave Image (Websocket)) PILrrr comfy.utilsrr'rNODE_CLASS_MAPPINGSNODE_DISPLAY_NAME_MAPPINGSr rr r5s <, 2r