o
    i                     @   s   d dl mZ d dlmZmZmZ d dlmZ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eZG dd deZdS )    )Enum)OptionalDictAny)	BaseModelFieldStrictBytesc                   @   s   e Zd ZU dZeeeef  ed< dZ	eeeef  ed< dZ
ee ed< dZeeeef  ed< dZeeeef  ed< dZeeeef  ed< dZee ed< dZee ed	< dZee ed
< dS )MoonvalleyPromptResponseNerrorframe_conditioningidinference_paramsmetamodel_params
output_urlprompt_textstatus)__name__
__module____qualname__r
   r   r   strr   __annotations__r   r   r   r   r   r   r   r    r   r   =/mnt/c/Users/fbmor/ComfyUI/comfy_api_nodes/apis/moonvalley.pyr	      s   
 r	   c                   @   s  e Zd ZU edddZee ed< edddZee	 ed< edd	dZ
ee ed
< edddZee ed< edddZee	 ed< edddZee ed< edddZee ed< edddZee ed< edddZee	 ed< edddZee ed< edddZee ed < ed!d"dZee ed#< edd$dZee ed%< edd&dZee	 ed'< ed(d)dZee ed*< edd+dZee ed,< ed-d.dZee ed/< edd0dZee ed1< edd2dZee ed3< ed4d5dZee ed6< dS )7$MoonvalleyTextToVideoInferenceParamsTWhether to add quality guidancedescriptionadd_quality_guidance333333?$Caching coefficient for optimizationcaching_coefficient    Number of caching cooldown stepscaching_cooldownNumber of caching warmup stepscaching_warmup!CLIP value for generation control
clip_valuer   Index of the conditioning frameconditioning_frame_indexK   9Number of cooldown steps (calculated based on num_frames)cooldown_steps   z(Frames per second of the generated videofps
   %Guidance scale for generation controlguidance_scalei8  z'Height of the generated video in pixelsheightNNegative prompt textnegative_prompt@   zNumber of frames to generate
num_frames,Random seed for generation (default: random)seed"Shift value for generation controlshift_valueP   Number of denoising stepssteps"Whether to use guidance schedulinguse_guidance_scheduleFWhether to use negative promptsuse_negative_prompts&Whether to use timestep transformationuse_timestep_transform7Number of warmup steps (calculated based on num_frames)warmup_stepsi  z&Width of the generated video in pixelswidth)r   r   r   r   r   r   boolr   r!   floatr$   intr&   r(   r*   r-   r/   r2   r3   r5   r   r7   r9   r;   r>   r@   rB   rD   rF   rG   r   r   r   r   r      sn   
 r   c                   @   sN   e Zd ZU dZee ed< dZee ed< dZ	ee ed< dZ
ee ed< dS )MoonvalleyTextToVideoRequestN	image_urlr   r   webhook_url)r   r   r   rL   r   r   r   r   r   r   rM   r   r   r   r   rK   L   s
   
 rK   c                   @      e Zd ZU dZee ed< dS )MoonvalleyUploadFileRequestNfile)r   r   r   rP   r   r   r   r   r   r   r   rO   S      
 rO   c                   @   rN   )MoonvalleyUploadFileResponseN
access_url)r   r   r   rS   r   r   r   r   r   r   r   rR   W   rQ   rR   c                   @   s  e Zd ZU edddZee ed< edddZee	 ed< edd	dZ
ee ed
< edddZee ed< edddZee	 ed< edddZee ed< edddZee ed< edddZee	 ed< edddZee ed< edddZee ed< edddZee	 ed< edd dZee ed!< edd"dZee ed#< ed$d%dZee ed&< edd'dZee ed(< ed)d*dZee ed+< dS ),%MoonvalleyVideoToVideoInferenceParamsTr   r   r   r   r    r!   r"   r#   r$   r%   r&   r'   r(   r   r)   r*   $   r,   r-      r1   r2   Nr4   r5   r8   r9   r:   r;   r<   r=   r>   r?   r@   FrA   rB   rC   rD   r.   rE   rF   )r   r   r   r   r   r   rH   r   r!   rI   r$   rJ   r&   r(   r*   r-   r2   r5   r   r9   r;   r>   r@   rB   rD   rF   r   r   r   r   rT   [   sZ   
 rT   c                   @   s   e Zd ZdZdZdS )ControlTypemotion_controlpose_controlN)r   r   r   rX   rY   r   r   r   r   rW      s    rW   c                   @   sr   e Zd ZU edddZeed< dZe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 ed< dS )MoonvalleyVideoToVideoRequest.z!Supported types for video controlr   control_typeNr   zDescribes the video to generater   zUrl to control video	video_urlz&Optional webhook URL for notificationsrM   )r   r   r   r   r[   rW   r   r   r   rT   r   r   r\   rM   r   r   r   r   rZ      s   
 rZ   N)enumr   typingr   r   r   pydanticr   r   r   r	   r   rK   rO   rR   rT   r   rW   rZ   r   r   r   r   <module>   s    9/