
    +jW              	           d dl mZ d dlmZmZ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 G d	 d
e          Z G d de
eeee         f                            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e          Z G d de          Z G d dee          Z G d de          Z G d dee          Z G d d e          Z G d! d"e          Zd#S )$    )Enum)OptionalListUnion)datetime)	BaseModelField	RootModelc                   .    e Zd ZdZdZdZdZdZdZdZ	dZ
d	S )
RunwayAspectRatioEnumz1280:720z720:1280z1104:832z832:1104z960:960z1584:672z1280:768z768:1280N)__name__
__module____qualname__field_1280_720field_720_1280field_1104_832field_832_1104field_960_960field_1584_672field_1280_768field_768_1280     ;/home/wildlama/comfy/ComfyUI/comfy_api_nodes/apis/runway.pyr   r      s6        NNNNMNNNNNr   r   c                       e Zd ZdZdZdS )PositionfirstlastN)r   r   r   r   r   r   r   r   r   r      s        EDDDr   r   c                   X    e Zd ZU  edd          Zeed<    edd          Zeed<   dS )RunwayPromptImageDetailedObject.zbThe position of the image in the output video. 'last' is currently supported for gen3a_turbo only.descriptionpositionz4A HTTPS URL or data URI containing an encoded image.uriN)	r   r   r   r	   r#   r   __annotations__r$   strr   r   r   r    r       sr         x  Hh    uO  C     r   r    c                   P    e Zd ZU  edd          Zeeee         f         e	d<   dS )RunwayPromptImageObject.zjImage(s) to use for the video generation. Can be a single URI or an array of image objects with positions.r!   rootN)
r   r   r   r	   r)   r   r&   r   r    r%   r   r   r   r(   r(   "   sY          ?De A? ? ?D%T9::
;     r   r(   c                       e Zd ZdZdZdS )RunwayModelEnum
gen4_turbogen3a_turboN)r   r   r   r,   r-   r   r   r   r+   r+   +   s        JKKKr   r+   c                       e Zd ZdZdZdS )RunwayDurationEnum   
   N)r   r   r   	integer_5
integer_10r   r   r   r/   r/   0   s        IJJJr   r/   c                       e Zd ZU eed<   eed<   eed<    eddd          Ze	e
         ed<   eed	<    ed
ddd          Zeed<   dS )RunwayImageToVideoRequestdurationmodelpromptImageNzText prompt for the generation  r"   
max_length
promptTextratio.zRandom seed for generationr   l    r"   geleseed)r   r   r   r/   r%   r+   r(   r	   r<   r   r&   r   rA   intr   r   r   r5   r5   5   s             (((( %:t! ! !J    !   5!
  D#     r   r5   c                   @    e Zd ZU  edd          Zee         ed<   dS )RunwayImageToVideoResponseNTask IDr!   idr   r   r   r	   rF   r   r&   r%   r   r   r   rD   rD   B   5         d	:::B:::::r   rD   c                   &    e Zd ZdZdZdZdZdZdZdS )RunwayTaskStatusEnum	SUCCEEDEDRUNNINGFAILEDPENDING	CANCELLED	THROTTLEDN)	r   r   r   rK   rL   rM   rN   rO   rP   r   r   r   rJ   rJ   F   s,        IGFGIIIIr   rJ   c                       e Zd ZU  edd          Zeed<    edd          Zeed<    edd          Z	e
ee                  ed	<    edd
dd          Ze
e         ed<   eed<   dS )RunwayTaskStatusResponse.zTask creation timestampr!   	createdAtrE   rF   NzArray of output video URLsoutputzgFloat value between 0 and 1 representing the progress of the task. Only available if status is RUNNING.g        g      ?r>   progressstatus)r   r   r   r	   rS   r   r%   rF   r&   rT   r   r   rU   floatrJ   r   r   r   rR   rR   O   s         %1JKKKIxKKKeCY///B///"'%:V"W"W"WFHT#YWWW %}	! ! !Hhuo    !     r   rR   c                       e Zd ZdZdS )Model4
gen4_imageN)r   r   r   rZ   r   r   r   rY   rY   \   s        JJJr   rY   c                   @    e Zd ZU  edd          Zee         ed<   dS )ReferenceImageNz3A HTTPS URL or data URI containing an encoded imager!   r$   )r   r   r   r	   r$   r   r&   r%   r   r   r   r\   r\   `   sG         O  C#     r   r\   c                   6    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S ) RunwayTextToImageAspectRatioEnumz	1920:1080z	1080:1920z	1024:1024z1360:768z	1080:1080z1168:880z	1440:1080z	1080:1440z1808:768z2112:912N)r   r   r   field_1920_1080field_1080_1920field_1024_1024field_1360_768field_1080_1080field_1168_880field_1440_1080field_1080_1440field_1808_768field_2112_912r   r   r   r^   r^   f   s@        !O!O!ON!ON!O!ONNNNr   r^   c                       e Zd ZU  edd          Zeed<    eddd          Zeed<   e	ed	<    ed
d          Z
eee                  ed<   d
S )RunwayTextToImageRequest.zModel to use for generationr!   r7   z$Text prompt for the image generationr9   r:   r<   r=   Nz1Array of reference images to guide the generationreferenceImages)r   r   r   r	   r7   rY   r%   r<   r&   r^   rk   r   r   r\   r   r   r   rj   rj   s   s         E#+HIIIE6IIIe?D  J    ,+++6;eM7 7 7OXd>23     r   rj   c                   @    e Zd ZU  edd          Zee         ed<   dS )RunwayTextToImageResponseNrE   r!   rF   rG   r   r   r   rm   rm   ~   rH   r   rm   N)enumr   typingr   r   r   r   pydanticr   r	   r
   r&   r   r   r    r(   r+   rB   r/   r5   rD   rJ   rR   rY   r\   r^   rj   rm   r   r   r   <module>rq      s	         ( ( ( ( ( ( ( ( ( (       0 0 0 0 0 0 0 0 0 0         C          sD   
    i       eC=>>?@           c4      
    d   

 
 
 
 
	 
 
 
; ; ; ; ; ; ; ;    3   
! 
! 
! 
! 
!y 
! 
! 
!    S$       Y   
  
  
  
  
 sD 
  
  
     y   ; ; ; ; ;	 ; ; ; ; ;r   