
    +j^                     z   d dl mZ d dlmZmZ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          Z G d de          Z G d de          ZdS )    )	TypedDict)AliasChoices	BaseModelFieldmodel_validatorc                   .    e Zd ZU eed<   eed<   eed<   dS )InputPortraitModeportrait_modeportrait_styleportrait_beautifierN)__name__
__module____qualname__str__annotations__     =/home/wildlama/comfy/ComfyUI/comfy_api_nodes/apis/magnific.pyr	   r	      s6         r   r	   c                   t    e Zd ZU eed<   eed<   eed<   eed<   eed<   eed<   eed<   eed<   eed	<   eed
<   dS )InputAdvancedSettingsadvanced_settingswhitesblacks
brightnesscontrast
saturationenginetransfer_light_atransfer_light_bfixed_generationN)r   r   r   r   r   intboolr   r   r   r   r      sx         KKKKKKOOOMMMOOOKKKr   r   c                   .    e Zd ZU eed<   eed<   eed<   dS )InputSkinEnhancerModemodeskin_detailoptimized_forN)r   r   r   r   r   r!   r   r   r   r$   r$      s3         
IIIr   r$   c                   6   e Zd ZU  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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<   dS )ImageUpscalerCreativeRequest.imagescale_factorr'   Nprompt
creativityhdrresemblance
fractalityr   )r   r   r   r   r*   r   r   r+   r'   r,   r-   r!   r.   r/   r0   r   r   r   r   r)   r)      s         sE3c

L#"""sM3###tFC$J$$$eCjjJ   uSzzCuSzzK!!!eCjjJ   %**FCr   r)   c                       e Zd ZU  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<   dS )	ImageUpscalerPrecisionV2Request.r*   sharpensmart_grainultra_detailflavorr+   N)r   r   r   r   r*   r   r   r3   r!   r4   r5   r6   r+   r   r   r   r2   r2   +   s         sE35::GSuSzzK!!!c

L#"""%**FCc

L#"""""r   r2   c                   0   e Zd ZU  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          Zeed
<   dS )#ImageRelightAdvancedSettingsRequest.r   r   r   r   r   r   r   r   r    N)r   r   r   r   r   r!   r   r   r   r   r   r   r   r   r   r    r"   r   r   r   r8   r8   4   s         %**FC%**FCeCjjJ   E#JJHceCjjJ   %**FC!E#JJc&&&!E#JJc&&&"U3ZZd'''''r   r8   c                   B   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dz  ed<   dS )ImageRelightRequest.r*   Nr,   #transfer_light_from_reference_imagelight_transfer_strengthinterpolate_from_originalchange_backgroundstylepreserve_detailsr   )r   r   r   r   r*   r   r   r,   r;   r<   r!   r=   r"   r>   r?   r@   r   r8   r   r   r   r:   r:   @   s         sE3tFC$J$$$6;eDkk'tAAA#(5::S---&+eCjjt000#eCjjt(((sE3"U3ZZd'''DIE#JJ:TANNNNNr   r:   c                      e Zd ZU  e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<    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
<    ed          Zeed<    ed          Zeed<    ed          Zeed<   dS )ImageStyleTransferRequest.r*   reference_imageNr,   style_strengthstructure_strengthis_portraitr   r   r6   r   r    )r   r   r   r   r*   r   r   rC   r,   rD   r!   rE   rF   r"   r   r   r6   r   r    r   r   r   rB   rB   L   s)        sE3 5::OS%%%tFC$J$$$%**NC$$$#eCjj(((c

K"""!&sNC$J+++&+eCjjt000%**FC%**FC"U3ZZd'''''r   rB   c                   p    e Zd ZU  ed          Zeed<    ed          Zeed<    ed          Z	eed<   dS ) ImageSkinEnhancerCreativeRequest.r*   r3   r4   N)
r   r   r   r   r*   r   r   r3   r!   r4   r   r   r   rH   rH   Z   sZ         sE35::GSuSzzK!!!!!r   rH   c                       e Zd ZU  ed          Zeed<    ed          Zeed<    ed          Z	eed<    ed          Z
eed<   dS ) ImageSkinEnhancerFaithfulRequest.r*   r3   r4   r&   N)r   r   r   r   r*   r   r   r3   r!   r4   r&   r   r   r   rJ   rJ   `   sr         sE35::GSuSzzK!!!uSzzK!!!!!r   rJ   c                       e Zd ZU  ed          Zeed<    ed          Zeed<    ed          Z	eed<    ed          Z
eed<   dS ) ImageSkinEnhancerFlexibleRequest.r*   r3   r4   r'   N)r   r   r   r   r*   r   r   r3   r!   r4   r'   r   r   r   rL   rL   g   sr         sE35::GSuSzzK!!!sM3#####r   rL   c                       e Zd ZU dZ ed          Zeed<    e edd                    Z	eed<    ed          Z
ee         dz  ed<    ed	
          ededefd                        ZdS )TaskResponsezMUnified response model that handles both wrapped and unwrapped API responses..task_idstatustask_status)validation_aliasN	generatedbefore)r%   valuesreturnc                 T    d|v r#t          |d         t                    r|d         S |S )Ndata)
isinstancedict)clsrU   s     r   unwrap_datazTaskResponse.unwrap_datau   s1     V
6&>4 @ @&>!r   )r   r   r   __doc__r   rO   r   r   r   rP   rS   listr   classmethodrZ   r\   r   r   r   rN   rN   n   s         WW5::GS%h)N)NOOOFCOOO"'%++ItCy4---_(### $    [ $#  r   rN   N)typingr   pydanticr   r   r   r   r	   r   r$   r)   r2   r8   r:   rB   rH   rJ   rL   rN   r   r   r   <module>rb      sE         D D D D D D D D D D D D    	   
 
 
 
 
I 
 
 
    I   	 	 	 	 	9 	 	 	# # # # #i # # #	( 	( 	( 	( 	() 	( 	( 	(	O 	O 	O 	O 	O) 	O 	O 	O( ( ( ( (	 ( ( (" " " " "y " " "" " " " "y " " "$ $ $ $ $y $ $ $    9     r   