
    +j                        S SK Jr  S SKJr  S SKJrJrJr   " S S\\5      r	 " S S\\5      r
S&S jr " S	 S
\\5      r " S S\\5      r " S S\\5      r " S S\5      r " S S\5      r " S S\5      r " S S\5      r " S S\5      r " S S\5      r " S S\5      r " S S\5      r " S S \5      r " S! S"\5      r " S# S$\5      rg%)'    )Enum)Optional)	BaseModelFieldconfloatc                        \ rS rSrSrSrSrSrg)StabilityFormat   pngjpegwebp N)__name__
__module____qualname____firstlineno__r   r   r   __static_attributes__r       >/home/wildlama/comfy/ComfyUI/comfy_api_nodes/apis/stability.pyr	   r	      s    
CDDr   r	   c                   8    \ rS rSrSrSrSrSrSrSr	Sr
S	rS
rSrg)StabilityAspectRatio   z1:1z16:9z9:16z3:2z2:3z5:4z4:5z21:9z9:21r   N)r   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   r      s/    IJJIIIIJJr   r   c                     / nU (       a  UR                  S5        U[         Vs/ s H  o"R                  PM     sn-   $ s  snf )NNone)appendStabilityStylePresetvalue)include_nonepresetsxs      r   get_stability_style_presetsr*      s8    Gv';<';!gg';<<<<s   >c                   X    \ rS rSrSrSrSrSrSrSr	Sr
S	rS
rSrSrSrSrSrSrSrSrSrg)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-texturer   N)r   r   r   r   	_3d_modelanalog_filmr-   r.   
comic_bookdigital_artr/   fantasy_artr0   line_artlow_polymodeling_compound	neon_punkr1   r2   	pixel_arttile_texturer   r   r   r   r%   r%       sX    IKEIJKGKIHH+IG!LI!Lr   r%   c                       \ rS rSrSrSrSrg)Stability_SD3_5_Model4   zsd3.5-largezsd3.5-mediumr   N)r   r   r   r   sd3_5_largesd3_5_mediumr   r   r   r   r?   r?   4   s    K!Lr   r?   c                       \ rS rSrSrSrSrg)Stability_SD3_5_GenerationMode:   ztext-to-imagezimage-to-imager   N)r   r   r   r   text_to_imageimage_to_imager   r   r   r   rD   rD   :   s    #M%Nr   rD   c                      \ rS rSr% \" S5      r\\S'   \" S5      r\\S'   \" S5      r	\\S'   \" S5      r
\\   \S'   \" S5      r\\   \S'   \" S5      r\\   \S	'   \" \R                   R"                  5      r\\   \S
'   \" S5      r\\   \S'   \" S5      r\\   \S'   \" S5      r\\S'   \" S5      r\\" SSS9   \S'   Srg)StabilityStable3_5Request?   .modelmodepromptNnegative_promptaspect_ratioseedoutput_formatimagestyle_preset	cfg_scale              ?gelestrengthr   )r   r   r   r   r   rK   str__annotations__rL   rM   rN   r   rO   rP   intr	   r   r&   rQ   rR   rS   rT   floatrZ   r   r   r   r   r   rI   rI   ?   s    sE3c
D#*FC%*4[OXc]0"'+L(3--+D(3-%#()<)<)B)B#CM8C=C ;E8C=&"'+L(3--SzIu!38;Hhx33/0>r   rI   c                      \ rS rSr% \" S5      r\\S'   \" S5      r\	\   \S'   \" S5      r
\	\   \S'   \" \R                  R                  5      r\	\   \S'   \" S5      r\	\   \S'   \" S5      r\	\" S	S
S9   \S'   Srg)#StabilityUpscaleConservativeRequestM   .rM   NrN   rP   rQ   rR   g?      ?rW   
creativityr   )r   r   r   r   r   rM   r[   r\   rN   r   rP   r]   r	   r   r&   rQ   rR   rc   r   r   r   r   r   r`   r`   M   s    *FC%*4[OXc]0+D(3-%#()<)<)B)B#CM8C=C ;E8C=&5:4[JSS12@r   r`   c                   "   \ rS rSr% \" S5      r\\S'   \" S5      r\	\   \S'   \" S5      r
\	\   \S'   \" \R                  R                  5      r\	\   \S'   \" S5      r\	\   \S'   \" S5      r\	\" S	S
S9   \S'   \" S5      r\	\   \S'   Srg)StabilityUpscaleCreativeRequestV   .rM   NrN   rP   rQ   rR   g?rb   rW   rc   rS   r   )r   r   r   r   r   rM   r[   r\   rN   r   rP   r]   r	   r   r&   rQ   rR   rc   r   rS   r   r   r   r   re   re   V   s    *FC%*4[OXc]0+D(3-%#()<)<)B)B#CM8C=C ;E8C=&5:4[JSS12@"'+L(3--r   re   c                   B   \ rS rSr% \" S5      r\\S'   \" S5      r\	\   \S'   \" S5      r
\	\   \S'   \" S5      r\	\   \S'   \" \R                  R                  5      r\	\   \S'   \" S5      r\	\   \S	'   \" S5      r\	\   \S
'   \" S5      r\	\" SSS9   \S'   Srg)StabilityStableUltraRequest`   .rM   NrN   rO   rP   rQ   rR   rS   rU   rV   rW   rZ   r   )r   r   r   r   r   rM   r[   r\   rN   r   rO   rP   r]   r	   r   r&   rQ   rR   rS   rZ   r   r   r   r   r   rh   rh   `   s    *FC%*4[OXc]0"'+L(3--+D(3-%#()<)<)B)B#CM8C=C ;E8C=&"'+L(3--38;Hhx33/0>r   rh   c                   v    \ rS rSr% \" S5      r\\   \S'   \" S5      r	\\   \S'   \" S5      r
\\   \S'   Srg)StabilityStableUltraResponsek   NrR   finish_reasonrP   r   )r   r   r   r   r   rR   r   r[   r\   rm   rP   r]   r   r   r   r   rk   rk   k   s8     ;E8C=&#(;M8C=.+D(3-%r   rk   c                      \ rS rSr% \" S5      r\\   \S'   \" S5      r	\\   \S'   \" S5      r
\\   \S'   \" S5      r\\   \S'   \" S5      r\\   \S'   \" S5      r\\\      \S'   \" S5      r\\   \S	'   \" S5      r\\   \S
'   Srg)StabilityResultsGetResponseq   NrR   rm   rP   idnameerrorsstatusresultr   )r   r   r   r   r   rR   r   r[   r\   rm   rP   r]   rq   rr   rs   listrt   ru   r   r   r   r   ro   ro   q   s     ;E8C=&#(;M8C=.+D(3-%dB#+D(3-%"'+FHT#Y-!$KFHSM'!$KFHSM'r   ro   c                   6    \ rS rSr% \" S5      r\\   \S'   Sr	g)StabilityAsyncResponse|   Nrq   r   )
r   r   r   r   r   rq   r   r[   r\   r   r   r   r   rx   rx   |   s    dB#r   rx   c                       \ rS rSr% \" S5      r\\S'   \" S5      r\\S'   \" SSSS9r	\
\S'   \" S	S	S
S9r\
\S'   \" SSSS9r\
\S'   \" S5      r\\S'   Srg)StabilityTextToAudioRequest   .rK   rM         rW   durationr   l    rP         stepswavrQ   r   N)r   r   r   r   r   rK   r[   r\   rM   r   r]   rP   r   rQ   r   r   r   r   r{   r{      se    sE3*FC#!,Hc,aA*-D#-qQ1%E3%uM3%r   r{   c                   0    \ rS rSr% \" SSSS9r\\S'   Srg)StabilityAudioToAudioRequest   g{Gz?rV   rW   rZ   r   N)	r   r   r   r   r   rZ   r^   r\   r   r   r   r   r   r      s    DTc2He2r   r   c                   J    \ rS rSr% \" SSSS9r\\S'   \" SSSS9r\\S'   Sr	g	)
StabilityAudioInpaintRequest      r   r}   rW   
mask_startmask_endr   N)
r   r   r   r   r   r   r]   r\   r   r   r   r   r   r   r      s(    B1-J-#!,Hc,r   r   c                   6    \ rS rSr% \" S5      r\\   \S'   Sr	g)StabilityAudioResponse   Naudior   )
r   r   r   r   r   r   r   r[   r\   r   r   r   r   r   r      s     ;E8C=&r   r   N)T)enumr   typingr   pydanticr   r   r   r[   r	   r   r*   r%   r?   rD   rI   r`   re   rh   rk   ro   rx   r{   r   r   r   r   r   r   <module>r      s      / /c4 	3 	="3 "("C "&S$ &
?	 ?A) A.i .?) ?&9 &() ($Y $&) &3#> 3-#> -
'Y 'r   