o پi@sddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddl m Z deddfddZde jdejfd d Zde jde jfd d Zdde jdedefddZde jdefddZdede jfddZGddde Zdededede fddZdS)N) VideoInput)ImagepathreturncCstj|dddS)NTexist_ok)osmakedirs)rr i/home/ubuntu/.local/lib/python3.10/site-packages/sglang/multimodal_gen/apps/ComfyUI_SGLDiffusion/utils.py _ensure_dirsr imagecCs|dkr |d}|dkr|jddvr|ddd}n |dkr*|d}|}t|dd }|d  tj }|jddkrPtj |ddd }|S) z tolerancec Csb|durdSt|}|ddddf}|j\}}}|d|}t||}t|} | |kS)a] Check if the input image is an empty/solid color image (like ComfyUI's empty image). Args: image: Input tensor image in ComfyUI format (BCHW, CHW, HWC, etc.) tolerance: Tolerance for floating point comparison (default: 1e-6) Returns: True if the image is empty (all pixels have same color), False otherwise NTrr)r*rreshaper'absmaxitem) r r,img_hwc first_pixelhwcpixelsdiffmax_diffr r r is_empty_image4s     r9cCsht}td}tjdd}d|d|d}tj ||}t |}t |}|j |dd|S) zQ Save tensor image to ComfyUI temp directory as PNG and return the path. z %Y%m%d-%H%M%SN sgl_output__z.pngPNG)format) folder_pathsget_temp_directorytimestrftimeuuiduuid4hexrrjoinr%r fromarraysave)r temp_dirtsunique file_name file_pathr$r)r r r get_image_pathPs  rN b64_imagecCs`t|}tt|}|jdkr|d}t | tj d}|tj df}t |}|S)a. Convert base64 encoded image to ComfyUI IMAGE format (torch.Tensor). Args: b64_image: Base64 encoded image string Returns: torch.Tensor with shape [batch_size, height, width, channels] (BHWC format), values normalized to [0, 1] range, RGB format (3 channels) RGBgo@.)base64 b64decoderopenioBytesIOmodeconvertrarrayr!float32newaxisr' from_numpy)rO image_bytes pil_image image_array tensor_imager r r convert_b64_to_tensor_imageds    r`csXeZdZdededeffdd ZdeeeffddZd d Zdd efd dZ Z S)SGLDVideoInput video_pathheightwidthcs t||_||_||_dS)N)super__init__rbrcrd)selfrbrcrd __class__r r rfs  zSGLDVideoInput.__init__rcCs |j|jfS)zs Returns the dimensions of the video input. Returns: Tuple of (width, height) )rdrcrgr r r get_dimensionss zSGLDVideoInput.get_dimensionscCs|jgS)z| Returns the components of the video input. This is required by the VideoInput abstract base class. )rbrjr r r get_componentsszSGLDVideoInput.get_componentsNrcCsF|}tj|jr!tj|}|rtj|ddt|j|dSdS)zD Abstract method to save the video input to a file. TrN)rrexistsrbdirnamer shutilcopy2)rgrr>codecmetadata save_pathsave_dirr r r save_tos zSGLDVideoInput.save_to)NNN) __name__ __module__ __qualname__strintrftuplerkrlru __classcell__r r rhr ras  rarbrcrdcCst|||}|S)z= Convert video to ComfyUI VIDEO format (VideoInput). )ra)rbrcrd video_inputr r r convert_video_to_comfy_videos r~)r+)rQrTrrorArCr?rrr'comfy_api.inputrPILrryr Tensorndarrayr%r*floatboolr9rNr`rarzr~r r r r s6  &