
    +jb                     d   d dl mZ d dlmZ d dlmZmZ  G d de          Z G d de          Z G d d	e          Z	 G d
 de          Z
 G d de          Z G d de          Z G d de          Z G d de          Z G d de          Z G d dee          Z G d de          ZdS )    )Enum)Any)	BaseModelFieldc                      e Zd ZU  ed          Zeed<    ed          Zedz  ed<    ed          Z	e
dz  ed<    ed          Ze
ed<    ed          Ze
ed<    ed          Ze
ed<    ed          Ze
ed	<    ed          Ze
ed
<    ed          Zeed<    ed          Ze
ed<    ed          Zeed<    edd          Zeed<   dS )BFLFluxExpandImageRequest.promptNprompt_upsamplingseedtopbottomleftrightstepsguidance   safety_tolerancepngoutput_formatzAA Base64-encoded string representing the image you wish to expanddescriptionimage)__name__
__module____qualname__r   r	   str__annotations__r
   boolr   intr   r   r   r   r   r   floatr   r   r        8/home/wildlama/comfy/ComfyUI/comfy_api_nodes/apis/bfl.pyr   r      sD        %**FC%*U4[[td{000uT{{D#*"""uSzzC%**FCc

D#sE3sE3eCjjHe   !E!HHc$$$uM3%%%t)lmmmE3mmmmmr"   r   c                   D   e Zd ZU  ed          Zeed<    ed          Zedz  ed<    ed          Z	e
dz  ed<    ed          Ze
ed<    ed          Zeed<    ed          Ze
ed	<    ed
          Zeed<    edd          Zeed<    edd          Zeed<   dS )BFLFluxFillImageRequest.r	   Nr
   r   r   r   r   r   r   r   zZBase64-encoded string representing the image to modify. Can contain alpha mask if desired.r   r   zLBase64-encoded string representing the mask of the areas you wish to modify.mask)r   r   r   r   r	   r   r   r
   r   r   r   r   r   r    r   r   r   r&   r!   r"   r#   r%   r%      s$        %**FC%*U4[[td{000uT{{D#*"""sE3eCjjHe   !E!HHc$$$uM3%%%v  E3    h  D#     r"   r%   c                       e Zd ZU  edd          Zeed<    edd          Zeed<    ed          Ze	ed<    ed	          Z
e	d	z  ed
<    ed          Zeed<   d	S )BFLFluxEraseRequest.z=A Base64-encoded string representing the image to erase from.r   r   zA Base64-encoded black/white mask matching the input dimensions; white (255) marks areas to remove, black (0) marks areas to preserve.r&   
   dilate_pixelsNr   r   r   )r   r   r   r   r   r   r   r&   r*   r   r   r   r!   r"   r#   r(   r(   &   s         s(ghhhE3hhhP  D#   
 rM3"""uT{{D#*"""uM3%%%%%r"   r(   c                       e Zd ZU  edd          Zeed<    edd          Zeed<    edd          Zeed<    ed	          Z	e
d	z  ed
<    ed          Ze
ed<    ed          Zeed<   d	S )BFLFluxVTORequest.zQNatural-language styling instruction. Required field, but may be an empty string.r   r	   z6A Base64-encoded string representing the person image.personzAA Base64-encoded string representing the garment reference image.garmentNr      r   r   r   )r   r   r   r   r	   r   r   r-   r.   r   r   r   r   r!   r"   r#   r,   r,   2   s         %l  FC    %)abbbFCbbb5*mnnnGSnnnuT{{D#*"""!E!HHc$$$uM3%%%%%r"   r,   c                   .   e Zd ZU  ed          Zeed<    ed          Zedz  ed<    ed          Z	e
dz  ed<    edd          Ze
ed	<    ed
d          Ze
ed<    ed          Ze
ed<    ed          Zeed<    edd          Zedz  ed<   dS )BFLFluxProGenerateRequest.r	   Nr
   r      Must be a multiple of 32.r   width   heightr   r   r   r   (Optional image to remix in base64 formatimage_prompt)r   r   r   r   r	   r   r   r
   r   r   r   r4   r6   r   r   r8   r!   r"   r#   r1   r1   =   s         %**FC%*U4[[td{000uT{{D#*"""t)DEEEE3EEE%)DEEEFCEEE!E!HHc$$$uM3%%%$uT7abbbL#*bbbbbr"   r1   c                   ~   e Zd ZU  ed          Zeed<    edd          Zeed<    edd          Z	eed<    ed	          Z
ed	z  ed
<    ed	          Zed	z  ed<    ed	d          Zed	z  ed<    ed	d          Zed	z  ed<    ed	d          Zed	z  ed<    ed	d          Zed	z  ed<    ed	d          Zed	z  ed<    ed	d          Zed	z  ed<    ed	d          Zed	z  ed<    ed	d          Zed	z  ed<    ed	d          Zed	z  ed<    ed          Zeed<    ed          Zeed<   d	S )Flux2ProGenerateRequest.r	   r2   r3   r   r4   r5   r6   Nr   r
   z2Base64 encoded image for image-to-image generationinput_imageinput_image_2input_image_3input_image_4input_image_5input_image_6input_image_7input_image_8input_image_9r/   r   r   r   )r   r   r   r   r	   r   r   r4   r   r6   r   r
   r   r;   r<   r=   r>   r?   r@   rA   rB   rC   r   r   r!   r"   r#   r:   r:   H   s        %**FCt)DEEEE3EEE%)DEEEFCEEEuT{{D#*"""%*U4[[td{000#eD6jkkkKtkkk %d8l m m mM3:mmm %d8l m m mM3:mmm %d8l m m mM3:mmm %d8l m m mM3:mmm %d8l m m mM3:mmm %d8l m m mM3:mmm %d8l m m mM3:mmm %d8l m m mM3:mmm!E!HHc$$$uM3%%%%%r"   r:   c                   L   e Zd ZU  ed          Zeed<    edd          Zedz  ed<    ed          Ze	dz  ed<    ed          Z
eed<    ed          Ze	ed	<    ed
          Ze	ed<    ed          Zeed<    ed          Zedz  ed<    ed          Zedz  ed<   dS ) BFLFluxKontextProGenerateRequest.r	   NzImage to edit in base64 formatr   r;   r   r   r      r   r   r   aspect_ratior
   )r   r   r   r   r	   r   r   r;   r   r   r   r    r   r   r   rG   r
   r   r!   r"   r#   rE   rE   [   s        %**FC#eD6VWWWKtWWWuT{{D#*"""eCjjHe   sE3!E!HHc$$$uM3%%%$uT{{L#****%*U4[[td{00000r"   rE   c                   X   e Zd ZU  ed          Zeed<    ed          Zedz  ed<    ed          Z	e
dz  ed<    ed          Zedz  ed<    ed          Ze
ed<    ed	          Zeed
<    ed          Zedz  ed<    edd          Zedz  ed<    ed          Zedz  ed<   dS )BFLFluxProUltraGenerateRequest.r	   Nr
   r   rG   r   r   r   r   rawr7   r   r8   image_prompt_strength)r   r   r   r   r	   r   r   r
   r   r   r   rG   r   r   rJ   r8   rK   r    r!   r"   r#   rI   rI   g   s        %**FC%*U4[[td{000uT{{D#*"""$uT{{L#****!E!HHc$$$uM3%%%uT{{C"""$uT7abbbL#*bbb*/%++54<55555r"   rI   c                   z    e Zd ZU  ed          Zeed<    ed          Zeed<    edd          Ze	dz  ed<   dS )BFLFluxProGenerateResponse.idpolling_urlNzPrice in centsr   cost)
r   r   r   r   rN   r   r   rO   rP   r    r!   r"   r#   rM   rM   s   sf         eCjjBuSzzK!!!t1ABBBD%$,BBBBBr"   rM   c                   &    e Zd ZdZdZdZdZdZdZdS )	BFLStatuszTask not foundPendingzRequest ModeratedzContent ModeratedReadyErrorN)	r   r   r   task_not_foundpendingrequest_moderatedcontent_moderatedreadyerrorr!   r"   r#   rR   rR   y   s.        %NG++EEEEr"   rR   c                       e Zd ZU  ed          Zeed<    ed          Zeed<    ed          Z	e
eef         dz  ed<    eddd          Zedz  ed	<   dS )
BFLFluxStatusResponse.rN   statusNresultg        g      ?)geleprogress)r   r   r   r   rN   r   r   r^   rR   r_   dictr   rb   r    r!   r"   r#   r]   r]      s         eCjjBc

FI"""$)E$KKFDcNT!///"U4CC888Hedl88888r"   r]   N)enumr   typingr   pydanticr   r   r   r%   r(   r,   r1   r:   rE   rI   rM   r   rR   r]   r!   r"   r#   <module>rg      s3               % % % % % % % %n n n n n	 n n n    i    	& 	& 	& 	& 	&) 	& 	& 	&& & & & &	 & & &c c c c c	 c c c& & & & &i & & &&	1 	1 	1 	1 	1y 	1 	1 	1	6 	6 	6 	6 	6Y 	6 	6 	6C C C C C C C C    T   9 9 9 9 9I 9 9 9 9 9r"   