
    +j                     x    d dl Z d dlmZ d dlmZmZ  G d dej                  Z G d de          Zdefd	Z	dS )
    N)override)IOComfyExtensionc                   P    e Zd ZdZed             Zeddej        fd            ZdS )ImageComparez,Compares two images with a slider interface.c                     t          j        dddddddt           j                            dd          t           j                            d	d          t           j                            d
          gg 	  	        S )Nr   zCompare Imagesz/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   )clss    @/home/wildlama/comfy/ComfyUI/comfy_extras/nodes_image_compare.pydefine_schemazImageCompare.define_schema
   sw    y")I - y488y488%%n55
 
 
 
 	
    Nreturnc                 L   g g d}t          j                    }|:t          |          dk    r'|                    |d          }|d         d         |d<   |:t          |          dk    r'|                    |d          }|d         d         |d<   t	          j        |	          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saveds          r   executezImageCompare.execute   s     b11)++3w<<!#3#3 ,,W6GHHE!&tX!6F:3w<<!#3#3 ,,W6GHHE!&tX!6F:}''''r   )NNN)	__name__
__module____qualname____doc__classmethodr   r   r(   r,    r   r   r   r      sa        66
 
 [
" ( (r} ( ( ( [( ( (r   r   c                   L    e Zd Zedeeej                          fd            ZdS )ImageCompareExtensionr   c                    K   t           gS N)r   )selfs    r   get_node_listz#ImageCompareExtension.get_node_list.   s       
 	
r   N)	r-   r.   r/   r   listtyper   	ComfyNoder8   r2   r   r   r4   r4   -   sE        
T$r|*<%= 
 
 
 X
 
 
r   r4   r   c                  "   K   t                      S r6   )r4   r2   r   r   comfy_entrypointr=   5   s       """r   )
r$   typing_extensionsr   comfy_api.latestr   r   r;   r   r4   r=   r2   r   r   <module>r@      s     & & & & & & / / / / / / / /#( #( #( #( #(2< #( #( #(L
 
 
 
 
N 
 
 
# 5 # # # # # #r   