o
    i                     @   s$  d dl mZm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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 )%    )	BaseModelFieldc                   @   s>   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 )MultiPromptEntry.indexpromptdurationN)
__name__
__module____qualname__r   r   int__annotations__r   strr    r   r   8/mnt/c/Users/fbmor/ComfyUI/comfy_api_nodes/apis/kling.pyr      s   
 r   c                   @   s   e Zd ZU 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	< ed
Z
eed< edZedB ed< edZee dB ed< edZedB ed< edddZeed< dS )OmniProText2VideoRequest.kling-video-o1description
model_name'16:9', '9:16' or '1:1'aspect_ratio'5' or '10'r   r   promodeN
multi_shotmulti_prompt	shot_type'on' or 'off'sound)r   r	   r
   r   r   r   r   r   r   r   r   r   boolr   listr   r   r   r   r   r   r   r   
   s   
 r   c                   @   s6   e Zd ZU edZeed< edddZedB ed< dS )OmniParamImage.	image_urlNz#Can be 'first_frame' or 'end_frame'r   type)r   r	   r
   r   r"   r   r   r#   r   r   r   r   r!      s   
 r!   c                   @   sJ   e Zd ZU edZeed< edddZedB ed< edddZeed< dS )	OmniParamVideo.	video_urlzCan be 'base' or 'feature'r   N
refer_typez'yes' or 'no'keep_original_sound)	r   r	   r
   r   r%   r   r   r&   r'   r   r   r   r   r$      s   
 r$   c                   @   s   e Zd ZU edddZeed< eddddZee	 ed< edd	dZ
eed
< edZeed< edZeed< edddZedB ed< edZedB ed< edZee dB ed< edZedB ed< dS )OmniProFirstLastFrameRequest.r   r   r         )
min_length
max_length
image_listr   r   r   r   r   Nr   r   r   r   r   )r   r	   r
   r   r   r   r   r-   r    r!   r   r   r   r   r   r   r   r   r   r   r   r   r   r(   !   s   
 r(   c                   @   s  e Zd ZU edddZeed< edddZedB ed< eddd	d
Ze	e
 dB ed< edddZe	e dB ed< edddZedB ed< edZeed< edZeed< edddZedB ed< edZedB ed< edZe	e dB ed< edZedB ed< dS )OmniProReferences2VideoRequest.r   r   r   r   Nr   r*   z#Max length 4 when video is present.)r,   r   r-   r)   r,   
video_listzFrom 3 to 10.r   r   r   r   r   r   r   r   r   )r   r	   r
   r   r   r   r   r   r-   r    r!   r0   r$   r   r   r   r   r   r   r   r   r   r   r   r   r   r.   -   s   
 r.   c                   @   sV   e Zd ZU edddZedB ed< edddZedB ed< edddZedB ed< dS )	TaskStatusVideoResultNzTotal video durationr   r   zGenerated video IDidzURL for generated videourl)	r   r	   r
   r   r   r   r   r2   r3   r   r   r   r   r1   =      
 r1   c                   @   s6   e Zd ZU edddZeed< edddZeed< dS )TaskStatusImageResult.u   Image Number，0-9r   r   zURL for generated imager3   N)	r   r	   r
   r   r   r   r   r3   r   r   r   r   r   r5   C   s   
 r5   c                   @   sV   e Zd ZU edZee dB ed< edZee	 dB ed< edZ
ee	 dB ed< dS )TaskStatusResultsNvideosimagesseries_images)r   r	   r
   r   r7   r    r1   r   r8   r5   r9   r   r   r   r   r6   H   r4   r6   c                   @   s   e Zd ZU edddZedB ed< edddZedB ed< dZe	dB ed< edddZ
e	dB ed	< edd
dZe	dB ed< edZedB ed< dS )TaskStatusResponseDataNzTask creation timer   
created_atzTask update time
updated_attask_statusz5Additional failure reason. Only for polling endpoint.task_status_msgzTask IDtask_idtask_result)r   r	   r
   r   r;   r   r   r<   r=   r   r>   r?   r@   r6   r   r   r   r   r:   N   s   
 r:   c                   @   sj   e Zd ZU edddZedB ed< edddZedB ed< edddZ	edB ed< edZ
edB ed	< dS )
TaskStatusResponseNz
Error coder   codezError messagemessagez
Request ID
request_iddata)r   r	   r
   r   rB   r   r   rC   r   rD   rE   r:   r   r   r   r   rA   W   s
   
 rA   c                   @   s   e Zd ZU edZeed< dS )OmniImageParamImage.imageN)r   r	   r
   r   rG   r   r   r   r   r   r   rF   ^   s   
 rF   c                   @   s   e Zd ZU edZeed< edZeed< edZedB ed< edZ	eed< edZ
eed< ed	d
dZedB ed< edddZee dB ed< edddZedB ed< eddd
ddZedB ed< dS )OmniProImageRequest.r   
resolutionNr   r   r   r   r)   	   )len
   r/   r-   z%Set to 'series' for series generationr   result_type   zNumber of images in a series)gerK   r   series_amount)r   r	   r
   r   r   r   r   rI   r   r   r   rL   r   r-   r    rF   rN   rQ   r   r   r   r   rH   b   s   
  rH   c                   @   s   e Zd ZU edZeed< edddZeed< edZeed< edZ	edB ed< edZ
edB ed	< ed
Zeed< edddZeed< edZedB ed< edZee dB ed< edZedB ed< dS )TextToVideoWithAudioRequest.r   r   r   r   r   Nr   negative_promptr   r   r   r   r   r   r   )r   r	   r
   r   r   r   r   r   r   r   rS   r   r   r   r   r   r    r   r   r   r   r   r   rR   n   s   
 rR   c                   @   s   e Zd ZU edZeed< edZeed< edZedB ed< edZ	eed< edZ
edB ed< edZedB ed< ed	Zeed
< edddZeed< edZedB ed< edZee dB ed< edZedB ed< dS )ImageToVideoWithAudioRequest.r   rG   N
image_tailr   r   rS   r   r   r   r   r   r   r   r   )r   r	   r
   r   r   r   r   rG   rU   r   r   rS   r   r   r   r   r   r    r   r   r   r   r   r   rT   {   s   
 rT   c                   @   sR   e Zd ZU edZeed< edZeed< edZedB ed< edZ	eed< dS )KlingAvatarRequest.rG   
sound_fileNr   r   )
r   r	   r
   r   rG   r   r   rW   r   r   r   r   r   r   rV      s
   
 rV   c                   @   s   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ddZeed	< edZeed
< dS )MotionControlRequest.r   r"   r%   r'   character_orientationz'pro' or 'std'r   r   r   N)r   r	   r
   r   r   r   r   r"   r%   r'   rY   r   r   r   r   r   r   rX      s   
 rX   N)pydanticr   r   r   r   r!   r$   r(   r.   r1   r5   r6   r:   rA   rF   rH   rR   rT   rV   rX   r   r   r   r   <module>   s$    	