o
    i                     @  sP  d dl mZ d dlmZ d dlmZ d dlmZmZm	Z	 G dd de
eZG dd de
eZd)d
dZG dd de
eZG dd de
eZG dd de
eZG dd deZG dd deZG dd deZG dd deZG dd deZG dd deZG dd deZG d d! d!eZG d"d# d#eZG d$d% d%eZG d&d' d'eZd(S )*    )annotations)Enum)Optional)	BaseModelFieldconfloatc                   @  s   e Zd ZdZdZdZdS )StabilityFormatpngjpegwebpN)__name__
__module____qualname__r	   r
   r    r   r   </mnt/c/Users/fbmor/ComfyUI/comfy_api_nodes/apis/stability.pyr   	   s    r   c                   @  s0   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    r   Tc                 C  s$   g }| r	| d |dd tD  S )NNonec                 S  s   g | ]}|j qS r   )value).0xr   r   r   
<listcomp>   s    z/get_stability_style_presets.<locals>.<listcomp>)appendStabilityStylePreset)include_nonepresetsr   r   r   get_stability_style_presets   s   
r$   c                   @  sP   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)   r*   	pixel_arttile_texturer   r   r   r   r!   "   s$    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   r7   6   s    r7   c                   @  r6   )Stability_SD3_5_GenerationModeztext-to-imagezimage-to-imageN)r   r   r   text_to_imageimage_to_imager   r   r   r   r:   <   s    r:   c                   @  s   e Zd ZU edZded< edZded< edZded< edZded< edZ	ded	< edZ
d
ed< eejjZded< edZded< edZded< edZded< edZded< dS )StabilityStable3_5Request.strmodelmodepromptNOptional[str]negative_promptaspect_ratioOptional[int]seedoutput_formatimagestyle_presetfloat	cfg_scale"Optional[confloat(ge=0.0, le=1.0)]strength)r   r   r   r   r?   __annotations__r@   rA   rC   rD   rF   r   r	   r   rG   rH   rI   rK   rM   r   r   r   r   r=   A   s   
 r=   c                   @  sr   e Zd ZU edZded< edZded< edZded< eej	j
Zded	< edZded
< edZded< dS )#StabilityUpscaleConservativeRequest.r>   rA   NrB   rC   rE   rF   rG   rH   z"Optional[confloat(ge=0.2, le=0.5)]
creativity)r   r   r   r   rA   rN   rC   rF   r   r	   r   rG   rH   rP   r   r   r   r   rO   O   s   
 rO   c                   @  s   e Zd ZU edZded< edZded< edZded< eej	j
Zded	< edZded
< edZded< edZded< dS )StabilityUpscaleCreativeRequest.r>   rA   NrB   rC   rE   rF   rG   rH   z"Optional[confloat(ge=0.1, le=0.5)]rP   rI   )r   r   r   r   rA   rN   rC   rF   r   r	   r   rG   rH   rP   rI   r   r   r   r   rQ   X   s   
 rQ   c                   @  s   e Zd ZU edZded< edZded< edZded< edZded	< ee	j
jZded
< edZded< edZded< edZded< dS )StabilityStableUltraRequest.r>   rA   NrB   rC   rD   rE   rF   rG   rH   rI   rL   rM   )r   r   r   r   rA   rN   rC   rD   rF   r   r	   r   rG   rH   rI   rM   r   r   r   r   rR   b   s   
 rR   c                   @  s>   e Zd ZU edZded< edZded< edZded< dS )StabilityStableUltraResponseNrB   rH   finish_reasonrE   rF   )r   r   r   r   rH   rN   rT   rF   r   r   r   r   rS   m   s   
 rS   c                   @  s   e Zd ZU edZded< edZded< edZded< edZded< edZ	ded< edZ
d	ed
< edZded< edZded< dS )StabilityResultsGetResponseNrB   rH   rT   rE   rF   idnamezOptional[list[str]]errorsstatusresult)r   r   r   r   rH   rN   rT   rF   rV   rW   rX   rY   rZ   r   r   r   r   rU   s   s   
 rU   c                   @     e Zd ZU edZded< dS )StabilityAsyncResponseNrB   rV   )r   r   r   r   rV   rN   r   r   r   r   r\   ~      
 r\   c                   @  s   e Zd ZU edZded< edZded< eddddZded	< ed
d
ddZded< eddddZ	ded< edZ
ded< dS )StabilityTextToAudioRequest.r>   r?   rA         geleintdurationr   l    rF         stepswavrG   N)r   r   r   r   r?   rN   rA   re   rF   rh   rG   r   r   r   r   r^      s   
 r^   c                   @  s$   e Zd ZU eddddZded< dS )StabilityAudioToAudioRequestg{Gz?g      ?ra   rJ   rM   N)r   r   r   r   rM   rN   r   r   r   r   rj      s   
 rj   c                   @  s:   e Zd ZU eddddZded< eddddZded< dS )	StabilityAudioInpaintRequest   r   r_   ra   rd   
mask_startmask_endN)r   r   r   r   rm   rN   rn   r   r   r   r   rk      s   
 rk   c                   @  r[   )StabilityAudioResponseNrB   audio)r   r   r   r   rp   rN   r   r   r   r   ro      r]   ro   N)T)
__future__r   enumr   typingr   pydanticr   r   r   r>   r   r   r$   r!   r7   r:   r=   rO   rQ   rR   rS   rU   r\   r^   rj   rk   ro   r   r   r   r   <module>   s*    
	
	