
    +j                        d dl mZ d dlmZ d dlmZmZmZ  G d dee          Z	 G d dee          Z
d'd	Z G d
 dee          Z G d dee          Z G d de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 )(    )Enum)Optional)	BaseModelFieldconfloatc                       e Zd ZdZdZdZdS )StabilityFormatpngjpegwebpN)__name__
__module____qualname__r
   r   r        >/home/wildlama/comfy/ComfyUI/comfy_api_nodes/apis/stability.pyr	   r	      s        
CDDDDr   r	   c                   2    e Zd ZdZdZdZdZdZdZdZ	dZ
d	Zd
S )StabilityAspectRatioz1:1z16:9z9:16z3:2z2:3z5:4z4:5z21:9z9:21N)r   r   r   	ratio_1_1
ratio_16_9
ratio_9_16	ratio_3_2	ratio_2_3	ratio_5_4	ratio_4_5
ratio_21_9
ratio_9_21r   r   r   r   r      s;        IJJIIIIJJJJr   r   Tc                 \    g }| r|                     d           |d t          D             z   S )NNonec                     g | ]	}|j         
S r   )value).0xs     r   
<listcomp>z/get_stability_style_presets.<locals>.<listcomp>   s    <<<!ag<<<r   )appendStabilityStylePreset)include_nonepresetss     r   get_stability_style_presetsr)      s<    G v<<';<<<<<r   c                   R    e Zd ZdZdZdZdZdZdZdZ	dZ
d	Zd
ZdZdZdZdZdZdZdZdS )r&   z3d-modelzanalog-filmanime	cinematicz
comic-bookzdigital-artenhancezfantasy-art	isometriczline-artzlow-polyzmodeling-compoundz	neon-punkorigamiphotographicz	pixel-artztile-textureN)r   r   r   	_3d_modelanalog_filmr+   r,   
comic_bookdigital_artr-   fantasy_artr.   line_artlow_polymodeling_compound	neon_punkr/   r0   	pixel_arttile_texturer   r   r   r&   r&       sd        IKEIJKGKIHH+IG!LI!LLLr   r&   c                       e Zd ZdZdZdS )Stability_SD3_5_Modelzsd3.5-largezsd3.5-mediumN)r   r   r   sd3_5_largesd3_5_mediumr   r   r   r=   r=   4   s        K!LLLr   r=   c                       e Zd ZdZdZdS )Stability_SD3_5_GenerationModeztext-to-imagezimage-to-imageN)r   r   r   text_to_imageimage_to_imager   r   r   rA   rA   :   s        #M%NNNr   rA   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         ed<    ed          Ze
e         ed<    ed          Ze
e         ed<    eej        j                  Ze
e         ed	<    ed          Ze
e         ed
<    ed          Ze
e         ed<    ed          Zeed<    ed          Ze
 edd                   ed<   dS )StabilityStable3_5Request.modelmodepromptNnegative_promptaspect_ratioseedoutput_formatimagestyle_preset	cfg_scale              ?gelestrength)r   r   r   r   rF   str__annotations__rG   rH   rI   r   rJ   rK   intr	   r
   r!   rL   rM   rN   rO   floatrU   r   r   r   r   rE   rE   ?   sN        sE3c

D#%**FC%*U4[[OXc]000"'%++L(3----%++D(3-%%%#(5)<)B#C#CM8C=CCC 5;;E8C=&&&"'%++L(3----uSzzIu!!!385;;Hhxx33///0>>>>>r   rE   c                   6   e Zd ZU  ed          Zeed<    ed          Zee         ed<    ed          Z	ee
         ed<    eej        j                  Zee         ed<    ed          Zee         ed<    ed          Ze edd	
                   ed<   dS )#StabilityUpscaleConservativeRequest.rH   NrI   rK   rL   rM   g?      ?rR   
creativity)r   r   r   r   rH   rV   rW   rI   r   rK   rX   r	   r
   r!   rL   rM   r]   r   r   r   r   r[   r[   M   s         %**FC%*U4[[OXc]000%++D(3-%%%#(5)<)B#C#CM8C=CCC 5;;E8C=&&&5:U4[[JSS1112@@@@@r   r[   c                   b   e Zd ZU  ed          Zeed<    ed          Zee         ed<    ed          Z	ee
         ed<    eej        j                  Zee         ed<    ed          Zee         ed<    ed          Ze edd	
                   ed<    ed          Zee         ed<   dS )StabilityUpscaleCreativeRequest.rH   NrI   rK   rL   rM   g?r\   rR   r]   rN   )r   r   r   r   rH   rV   rW   rI   r   rK   rX   r	   r
   r!   rL   rM   r]   r   rN   r   r   r   r_   r_   V   s         %**FC%*U4[[OXc]000%++D(3-%%%#(5)<)B#C#CM8C=CCC 5;;E8C=&&&5:U4[[JSS1112@@@"'%++L(3------r   r_   c                      e Zd ZU  ed          Zeed<    ed          Zee         ed<    ed          Z	ee         ed<    ed          Z
ee         ed<    eej        j                  Zee         ed<    ed          Zee         ed<    ed          Zee         ed	<    ed          Ze ed
d                   ed<   dS )StabilityStableUltraRequest.rH   NrI   rJ   rK   rL   rM   rN   rP   rQ   rR   rU   )r   r   r   r   rH   rV   rW   rI   r   rJ   rK   rX   r	   r
   r!   rL   rM   rN   rU   r   r   r   r   ra   ra   `   s        %**FC%*U4[[OXc]000"'%++L(3----%++D(3-%%%#(5)<)B#C#CM8C=CCC 5;;E8C=&&&"'%++L(3----385;;Hhxx33///0>>>>>r   ra   c                       e Zd ZU  ed          Zee         ed<    ed          Zee         ed<    ed          Z	ee
         ed<   dS )StabilityStableUltraResponseNrM   finish_reasonrK   )r   r   r   r   rM   r   rV   rW   rd   rK   rX   r   r   r   rc   rc   k   sf          5;;E8C=&&&#(5;;M8C=...%++D(3-%%%%%r   rc   c                   |   e Zd ZU  ed          Zee         ed<    ed          Zee         ed<    ed          Z	ee
         ed<    ed          Zee         ed<    ed          Zee         ed<    ed          Zeee                  ed<    ed          Zee         ed<    ed          Zee         ed	<   dS )
StabilityResultsGetResponseNrM   rd   rK   idnameerrorsstatusresult)r   r   r   r   rM   r   rV   rW   rd   rK   rX   rg   rh   ri   listrj   rk   r   r   r   rf   rf   q   s          5;;E8C=&&&#(5;;M8C=...%++D(3-%%%dB###%++D(3-%%%"'%++FHT#Y---!E$KKFHSM'''!E$KKFHSM'''''r   rf   c                   <    e Zd ZU  ed          Zee         ed<   dS )StabilityAsyncResponseNrg   )r   r   r   r   rg   r   rV   rW   r   r   r   rn   rn   |   s.         dB#####r   rn   c                       e Zd ZU  ed          Zeed<    ed          Zeed<    eddd          Ze	ed<    eddd	          Z
e	ed
<    eddd          Ze	ed<    ed          Zeed<   dS )StabilityTextToAudioRequest.rF   rH         rR   durationr   l    rK         stepswavrL   N)r   r   r   r   rF   rV   rW   rH   rs   rX   rK   rv   rL   r   r   r   rp   rp      s         sE3%**FCE#!,,,Hc,,,aA*---D#---qQ1%%%E3%%%uM3%%%%%r   rp   c                   6    e Zd ZU  eddd          Zeed<   dS )StabilityAudioToAudioRequestg{Gz?rQ   rR   rU   N)r   r   r   r   rU   rY   rW   r   r   r   ry   ry      s3         eDTc222He22222r   ry   c                   \    e Zd ZU  eddd          Zeed<    eddd          Zeed<   dS )StabilityAudioInpaintRequest   r   rq   rR   
mask_startmask_endN)r   r   r   r   r}   rX   rW   r~   r   r   r   r{   r{      sT         eB1---J---E#!,,,Hc,,,,,r   r{   c                   <    e Zd ZU  ed          Zee         ed<   dS )StabilityAudioResponseNaudio)r   r   r   r   r   r   rV   rW   r   r   r   r   r      s.          5;;E8C=&&&&&r   r   N)T)enumr   typingr   pydanticr   r   r   rV   r	   r   r)   r&   r=   rA   rE   r[   r_   ra   rc   rf   rn   rp   ry   r{   r   r   r   r   <module>r      s               / / / / / / / / / /    c4   	 	 	 	 	3 	 	 	= = = =" " " " "3 " " "(" " " " "C " " "& & & & &S$ & & &
? ? ? ? ?	 ? ? ?A A A A A) A A A. . . . .i . . .? ? ? ? ?) ? ? ?& & & & &9 & & &( ( ( ( () ( ( ($ $ $ $ $Y $ $ $& & & & &) & & &3 3 3 3 3#> 3 3 3- - - - -#> - - -
' ' ' ' 'Y ' ' ' ' 'r   