o
    i                     @   sX   d dl Z d dlmZ d dlmZmZ G dd dejZG dd deZdefd	d
Z	dS )    N)override)IOComfyExtensionc                   @   s2   e Zd ZdZedd ZeddejfddZdS )	ImageComparez,Compares two images with a slider interface.c                 C   sB   t jdddddddt jjdddt jjd	ddt jd
gg d	S )Nr   zImage Comparez/Compares two images side by side with a slider.imagezImage ToolsTimage_a)optionalimage_bcompare_view)	node_iddisplay_namedescriptioncategoryessentials_categoryis_experimentalis_output_nodeinputsoutputs)r   SchemaImageInputr   )cls r   >/mnt/c/Users/fbmor/ComfyUI/comfy_extras/nodes_image_compare.pydefine_schema
   s   
zImageCompare.define_schemaNreturnc                 C   s~   g g d}t  }|d ur!t|dkr!||d}|d d |d< |d ur9t|dkr9||d}|d d |d< tj|d	S )
N)a_imagesb_imagesr   zcomfy.compare.auiimagesr   zcomfy.compare.br   )r   )nodesPreviewImagelensave_imagesr   
NodeOutput)r   r   r	   r
   resultpreview_nodesavedr   r   r   execute   s   
zImageCompare.execute)NNN)	__name__
__module____qualname____doc__classmethodr   r   r$   r(   r   r   r   r   r      s    
r   c                   @   s(   e Zd Zedeeej  fddZdS )ImageCompareExtensionr   c                    s   t gS N)r   )selfr   r   r   get_node_list.   s   z#ImageCompareExtension.get_node_listN)	r)   r*   r+   r   listtyper   	ComfyNoder1   r   r   r   r   r.   -   s    r.   r   c                      s   t  S r/   )r.   r   r   r   r   comfy_entrypoint5   s   r5   )
r    typing_extensionsr   comfy_api.latestr   r   r4   r   r.   r5   r   r   r   r   <module>   s    &