
    +j                        d dl mZmZ  G d de          Z G d de          Z G d de          Z G d d	e          Z G d
 de          Z G d de          Z G d de          Z	 G d de          Z
 G d de          Z G d de          Z G d de          Z G d de          Z G d de          Z G d de          Z G d de          Z G d  d!e          Z G d" d#e          Z G d$ d%e          Z G d& d'e          Z G d( d)e          Z G d* d+e          Z G d, d-e          Z G d. d/e          Z G d0 d1e          Z G d2 d3e          Z G d4 d5e          Z G d6 d7e          Z G d8 d9e          Z G d: d;e          Z G d< d=e          Z  G d> d?e          Z! G d@ dAe          Z" G dB dCe          Z# G dD dEe          Z$dFS )G    )	BaseModelFieldc                   V    e Zd ZU  ed          Zeed<    ed          Zedz  ed<   dS )Text2ImageInputField.promptNnegative_prompt)__name__
__module____qualname__r   r   str__annotations__r        8/home/wildlama/comfy/ComfyUI/comfy_api_nodes/apis/wan.pyr   r      sF         %**FC"'%++OS4Z-----r   r   c                       e Zd ZU  ed          Zeed<    ed          Zedz  ed<    eddd          Ze	e         ed<   dS )	Image2ImageInputField.r   Nr         )
min_length
max_lengthimages)
r	   r
   r   r   r   r   r   r   r   listr   r   r   r   r   	   sk         %**FC"'%++OS4Z---caA>>>FDI>>>>>r   r   c                   |    e Zd ZU  ed          Zeed<    ed          Zedz  ed<    ed          Zedz  ed<   dS )Text2VideoInputField.r   Nr   	audio_url)	r	   r
   r   r   r   r   r   r   r   r   r   r   r   r      sb         %**FC"'%++OS4Z---!E$KKIsTz'''''r   r   c                       e Zd ZU  ed          Zeed<    ed          Zedz  ed<    ed          Zeed<    ed          Z	edz  ed<   dS )Image2VideoInputField.r   Nr   img_urlr   )
r	   r
   r   r   r   r   r   r   r   r   r   r   r   r   r      sz         %**FC"'%++OS4Z---5::GS!E$KKIsTz'''''r   r   c                       e Zd ZU  ed          Zeed<    ed          Zedz  ed<    ed          Ze	e         ed<   dS )Reference2VideoInputField.r   Nr   reference_video_urls)
r	   r
   r   r   r   r   r   r   r!   r   r   r   r   r    r       sc         %**FC"'%++OS4Z---&+eCjj$s)00000r   r    c                       e Zd ZU  ed          Zeed<    e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<   dS )Txt2ImageParametersField.sizer   Number of images to generate.descriptionnr   geleseedTprompt_extendF	watermarkN)r	   r
   r   r   r$   r   r   r(   intr-   r.   boolr/   r   r   r   r#   r#   "   s         c

D#U1"ABBBAsBBBcaJ///D#///%++M4%%%eEllIt"""""r   r#   c                       e Zd ZU  ed          Zedz  ed<    edd          Zeed<    eddd	
          Z	eed<    ed          Z
eed<   dS )Image2ImageParametersFieldNr$   r   r%   r&   r(   .r   r)   r*   r-   Fr/   )r	   r
   r   r   r$   r   r   r(   r0   r-   r/   r1   r   r   r   r3   r3   *   s         uT{{D#*"""U1"ABBBAsBBBcaJ///D#///eEllIt"""""r   r3   c                       e Zd ZU  e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d          Zeed<    ed          Zeed<   dS )Text2VideoParametersField.r$   r   r)   r*   r-         durationTr.   Fr/   (Whether to generate audio automatically.r&   audiosingle	shot_typeN)r	   r
   r   r   r$   r   r   r-   r0   r8   r.   r1   r/   r:   r<   r   r   r   r5   r5   1   s         c

D#caJ///D#///E!b)))Hc)))%++M4%%%eEllIt"""%+UVVVE4VVVU8__Is$$$$$r   r5   c                       e Zd ZU  e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d          Zeed<    ed          Zeed<   dS )Image2VideoParametersField.
resolutionr   r)   r*   r-   r6   r7   r8   Tr.   Fr/   r9   r&   r:   r;   r<   N)r	   r
   r   r   r?   r   r   r-   r0   r8   r.   r1   r/   r:   r<   r   r   r   r>   r>   ;   s         eCjjJ   caJ///D#///E!b)))Hc)))%++M4%%%eEllIt"""%+UVVVE4VVVU8__Is$$$$$r   r>   c                       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d	d
          Z
eed<    ed          Zeed<   dS )Reference2VideoParametersField.r$   r6   r7   r*   r8   r;   r<   r   r)   r-   Fr/   N)r	   r
   r   r   r$   r   r   r8   r0   r<   r-   r/   r1   r   r   r   rA   rA   E   s         c

D#E!b)))Hc)))U8__Is$$$caJ///D#///eEllIt"""""r   rA   c                   p    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 )Text2ImageTaskCreationRequest.modelinput
parametersN)r	   r
   r   r   rD   r   r   rE   r   rF   r#   r   r   r   rC   rC   M   s\         sE3"'%**E,,,+05::J(55555r   rC   c                   p    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 )Image2ImageTaskCreationRequest.rD   rE   rF   N)r	   r
   r   r   rD   r   r   rE   r   rF   r3   r   r   r   rH   rH   S   \         sE3#(5::E ----2U3ZZJ*77777r   rH   c                   p    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 )Text2VideoTaskCreationRequest.rD   rE   rF   N)r	   r
   r   r   rD   r   r   rE   r   rF   r5   r   r   r   rK   rK   Y   s\         sE3"'%**E,,,,1E#JJJ)66666r   rK   c                   p    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 )Image2VideoTaskCreationRequest.rD   rE   rF   N)r	   r
   r   r   rD   r   r   rE   r   rF   r>   r   r   r   rM   rM   _   rI   r   rM   c                   p    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 )"Reference2VideoTaskCreationRequest.rD   rE   rF   N)r	   r
   r   r   rD   r   r   rE   r    rF   rA   r   r   r   rO   rO   e   s\         sE3',uSzzE$11116sJ.;;;;;r   rO   c                   P    e Zd ZU  ed          Zeed<    ed          Zeed<   dS )Wan27MediaItem.typeurlN)r	   r
   r   r   rR   r   r   rS   r   r   r   rQ   rQ   k   sB         c

D#uSzzCr   rQ   c                       e Zd ZU  ed          Zeed<    ed          Zedz  ed<    ed          Ze	e
         ed<   dS )Wan27ReferenceVideoInputField.r   Nr   mediar	   r
   r   r   r   r   r   r   rV   r   rQ   r   r   r   rU   rU   p   sc         %**FC"'%++OS4Z---"'%**E4,,,,,r   rU   c                       e Zd ZU  ed          Zeed<    ed          Zedz  ed<    eddd          Ze	ed	<    ed
          Z
eed<    eddd          Ze	ed<   dS )"Wan27ReferenceVideoParametersField.r?   Nratior6   r   r7   r*   r8   Fr/   r   r)   r-   )r	   r
   r   r   r?   r   r   rZ   r8   r0   r/   r1   r-   r   r   r   rY   rY   v   s         eCjjJ   dE3:###E!b)))Hc)))eEllIt"""caJ///D#/////r   rY   c                   p    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 )&Wan27ReferenceVideoTaskCreationRequest.rD   rE   rF   N)r	   r
   r   r   rD   r   r   rE   rU   rF   rY   r   r   r   r\   r\   ~   s\         sE3+05::E(5555:U3ZZJ2?????r   r\   c                       e Zd ZU  ed          Zedz  ed<    ed          Zedz  ed<    ed          Ze	e
         ed<   dS )Wan27ImageToVideoInputFieldNr   r   .rV   rW   r   r   r   r^   r^      sg         tFC$J$$$"'%++OS4Z---"'%**E4,,,,,r   r^   c                       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
ed<    eddd          Zeed<   dS ) Wan27ImageToVideoParametersField.r?   r6   r   r7   r*   r8   Tr.   Fr/   r   r)   r-   N)r	   r
   r   r   r?   r   r   r8   r0   r.   r1   r/   r-   r   r   r   r`   r`      s         eCjjJ   E!b)))Hc)))%++M4%%%eEllIt"""caJ///D#/////r   r`   c                   p    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 )$Wan27ImageToVideoTaskCreationRequest.rD   rE   rF   N)r	   r
   r   r   rD   r   r   rE   r^   rF   r`   r   r   r   rb   rb      s\         sE3).sE&333385::J0=====r   rb   c                   \    e Zd ZU  ed          Zeed<    ed          Zee	         ed<   dS )Wan27VideoEditInputField.r   rV   N)
r	   r
   r   r   r   r   r   rV   r   rQ   r   r   r   rd   rd      sG         %**FC"'%**E4,,,,,r   rd   c                       e Zd ZU  ed          Zeed<    ed          Zedz  ed<    ed          Ze	dz  ed<    ed          Z
eed<    ed	          Zeed
<    eddd          Ze	ed<   dS )Wan27VideoEditParametersField.r?   NrZ   r   r8   autoaudio_settingFr/   r)   r*   r-   )r	   r
   r   r   r?   r   r   rZ   r8   r0   rh   r/   r1   r-   r   r   r   rf   rf      s         eCjjJ   dE3:### 588HcDj###vM3&&&eEllIt"""caJ///D#/////r   rf   c                   p    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 )!Wan27VideoEditTaskCreationRequest.rD   rE   rF   N)r	   r
   r   r   rD   r   r   rE   rd   rF   rf   r   r   r   rj   rj      s\         sE3&+eCjjE#00005c

J-:::::r   rj   c                       e Zd ZU  ed          Zeed<    ed          Zedz  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	ed<   dS )Wan27Text2VideoParametersField.r?   NrZ   r6   r   r7   r*   r8   Tr.   Fr/   r   r)   r-   )r	   r
   r   r   r?   r   r   rZ   r8   r0   r.   r1   r/   r-   r   r   r   rl   rl      s         eCjjJ   dE3:###E!b)))Hc)))%++M4%%%eEllIt"""caJ///D#/////r   rl   c                   p    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 )"Wan27Text2VideoTaskCreationRequest.rD   rE   rF   N)r	   r
   r   r   rD   r   r   rE   r   rF   rl   r   r   r   rn   rn      s\         sE3"'%**E,,,16sJ.;;;;;r   rn   c                   P    e Zd ZU  ed          Zeed<    ed          Zeed<   dS )TaskCreationOutputField.task_idtask_statusN)r	   r
   r   r   rq   r   r   rr   r   r   r   rp   rp      sB         5::GSuSzzK!!!!!r   rp   c                       e Zd ZU  ed          Zedz  ed<    ed          Zeed<    edd          Z	edz  ed<    edd          Z
edz  ed	<   dS )
TaskCreationResponseNoutput.
request_idz"Error code for the failed request.r&   codez!Details about the failed request.message)r	   r
   r   r   ru   rp   r   rv   r   rw   rx   r   r   r   rt   rt      s         -2U4[[F#d*888eCjjJ   uT/STTTD#*TTT%2UVVVGS4ZVVVVVr   rt   c                       e Zd ZU  ed          Zedz  ed<    ed          Zedz  ed<    ed          Zedz  ed<   dS )
TaskResultNrS   rw   rx   )	r	   r
   r   r   rS   r   r   rw   rx   r   r   r   rz   rz      sf         eDkkCt!!!uT{{D#*"""%++GS4Z%%%%%r   rz   c                       e Zd ZU  ed          Zeed<    ed          Zeed<    ed          Ze	e
         dz  ed<   dS )ImageTaskStatusOutputField.rq   rr   Nresults)r	   r
   r   r   rq   r   r   rr   r}   r   rz   r   r   r   r|   r|      sd         5::GSuSzzK!!!',uT{{GT*$22222r   r|   c                       e Zd ZU  ed          Zeed<    ed          Zeed<    ed          Zedz  ed<    ed          Z	edz  ed<    ed          Z
edz  ed<   dS )VideoTaskStatusOutputField.rq   rr   N	video_urlrw   rx   )r	   r
   r   r   rq   r   r   rr   r   rw   rx   r   r   r   r   r      s         5::GSuSzzK!!!!E$KKIsTz'''uT{{D#*"""%++GS4Z%%%%%r   r   c                   V    e Zd ZU  ed          Zedz  ed<    ed          Zeed<   dS )ImageTaskStatusResponseNru   .rv   )	r	   r
   r   r   ru   r|   r   rv   r   r   r   r   r   r      H         05dF&-;;;eCjjJ     r   r   c                   V    e Zd ZU  ed          Zedz  ed<    ed          Zeed<   dS )VideoTaskStatusResponseNru   .rv   )	r	   r
   r   r   ru   r   r   rv   r   r   r   r   r   r      r   r   r   N)%pydanticr   r   r   r   r   r   r    r#   r3   r5   r>   rA   rC   rH   rK   rM   rO   rQ   rU   rY   r\   r^   r`   rb   rd   rf   rj   rl   rn   rp   rt   rz   r|   r   r   r   r   r   r   <module>r      s   % % % % % % % %. . . . .9 . . .
? ? ? ? ?I ? ? ?( ( ( ( (9 ( ( (( ( ( ( (I ( ( (1 1 1 1 1	 1 1 1# # # # #y # # ## # # # # # # #% % % % %	 % % %% % % % % % % %# # # # #Y # # #6 6 6 6 6I 6 6 68 8 8 8 8Y 8 8 87 7 7 7 7I 7 7 78 8 8 8 8Y 8 8 8< < < < < < < <    Y   
- - - - -I - - -0 0 0 0 0 0 0 0@ @ @ @ @Y @ @ @- - - - -) - - -0 0 0 0 0y 0 0 0> > > > >9 > > >- - - - -y - - -
0 0 0 0 0I 0 0 0; ; ; ; ;	 ; ; ;0 0 0 0 0Y 0 0 0< < < < < < < <" " " " "i " " "
W W W W W9 W W W& & & & & & & &3 3 3 3 3!8 3 3 3& & & & &!8 & & &! ! ! ! !i ! ! !
! ! ! ! !i ! ! ! ! !r   