
    ^(j                     L    d dl mZ d dlZd dlZd dlZ G d d      ZdeiZddiZ	y)    )ImageNc                   B    e Zd Zed        ZdZdZdZdZd Z	ed        Z
y)	SaveImageWebsocketc                     dddiiS )Nrequiredimages)IMAGE )ss    K/Users/danicosta/Desktop/Flux2/ComfyUI/custom_nodes/websocket_image_save.pyINPUT_TYPESzSaveImageWebsocket.INPUT_TYPES   s    {, 	    r
   save_imagesTimagec                    t         j                  j                  |j                  d         }d}|D ]  }d|j	                         j                         z  }t        j                  t        j                  |dd      j                  t        j                              }|j                  ||j                  d   d|d f       |dz  } i S )Nr   g     o@   PNG   )comfyutilsProgressBarshapecpunumpyr   	fromarraynpclipastypeuint8update_absolute)selfr   pbarstepr   iimgs          r   r   zSaveImageWebsocket.save_images   s    {{&&v||A7Euyy{((**A//"''!Q"4";";BHH"EFC  v||AT8JKAID	  	r   c                 *    t        j                          S )N)time)r   r   s     r   
IS_CHANGEDzSaveImageWebsocket.IS_CHANGED&   s    yy{r   N)__name__
__module____qualname__classmethodr   RETURN_TYPESFUNCTIONOUTPUT_NODECATEGORYr   r(   r
   r   r   r   r      sC     
 LHKH	  r   r   zSave Image (Websocket))
PILr   r   r   comfy.utilsr   r'   r   NODE_CLASS_MAPPINGSNODE_DISPLAY_NAME_MAPPINGSr
   r   r   <module>r5      s;        < , 
 2 r   