o i@sRddlZddlmZmZddlmZddlmZmZm Z ddl m Z m Z m Z mZmZmZmZmZmZmZmZGdddeZGdd d eZGd d d eZGd d d eZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZ GdddeZ!GdddeZ"Gd d!d!eZ#Gd"d#d#eZ$Gd$d%d%eZ%Gd&d'd'eZ&Gd(d)d)eZ'Gd*d+d+e%Z(Gd,d-d-e%Z)Gd.d/d/eZ*Gd0d1d1eZ+e,d2Z-Gd3d4d4ej.Z/Gd5d6d6ej.Z0Gd7d8d8ej.Z1Gd9d:d:ej.Z2Gd;d<dd>eZ4d?e4fd@dAZ5dS)BN) BaseModelField)override)IOComfyExtensionInput) ApiEndpointaudio_to_base64_stringdownload_url_to_image_tensordownload_url_to_video_outputget_number_of_imagespoll_opsync_optensor_to_base64_stringupload_video_to_comfyapivalidate_audio_durationvalidate_video_durationc@s2eZdZUedZeed<edZedBed<dS)Text2ImageInputField.promptNnegative_prompt)__name__ __module__ __qualname__rrstr__annotations__rrr7/mnt/c/Users/fbmor/ComfyUI/comfy_api_nodes/nodes_wan.pyrs rc@sLeZdZUedZeed<edZedBed<eddddZe eed<dS) Image2ImageInputField.rNr) min_length max_lengthimages) rrrrrrrrr"listrrrrrs rc@sFeZdZUedZeed<edZedBed<edZedBed<dS)Text2VideoInputField.rNr audio_url) rrrrrrrrr%rrrrr$! r$c@sVeZdZUedZeed<edZedBed<edZeed<edZ edBed<dS)Image2VideoInputField.rNrimg_urlr%) rrrrrrrrr(r%rrrrr''s r'c@sFeZdZUedZeed<edZedBed<edZe eed<dS)Reference2VideoInputField.rNrreference_video_urls) rrrrrrrrr*r#rrrrr).r&r)c@sheZdZUedZeed<edddZeed<edddd Z eed <ed Z e ed <ed Z e ed<dS)Txt2ImageParametersField.sizerNumber of images to generate. descriptionnrgeleseedT prompt_extendF watermarkN) rrrrr,rrr0intr5r6boolr7rrrrr+4s r+c@s\eZdZUedZedBed<edddZeed<eddd d Z eed <ed Z e ed <dS)Image2ImageParametersFieldNr,rr-r.r0.rr1r2r5Fr7) rrrrr,rrr0r8r5r7r9rrrrr:<s r:c@eZdZUedZeed<eddddZeed<eddddZ eed <ed Z e ed <ed Z e ed <ed ddZ e ed<edZeed<dS)Text2VideoParametersField.r,rr1r2r5durationTr6Fr7(Whether to generate audio automatically.r.audiosingle shot_typeN)rrrrr,rrr5r8r?r6r9r7rArCrrrrr<C r<c@r;)Image2VideoParametersField. resolutionrr1r2r5r=r>r?Tr6Fr7r@r.rArBrCN)rrrrrFrrr5r8r?r6r9r7rArCrrrrrEMrDrEc@sjeZdZUedZeed<eddddZeed<edZ eed<edd d dZ eed <ed Z e ed <dS)Reference2VideoParametersField.r,r=r>r2r?rBrCrr1r5Fr7N) rrrrr,rrr?r8rCr5r7r9rrrrrGWs rGc@>eZdZUedZeed<edZeed<edZ e ed<dS)Text2ImageTaskCreationRequest.modelinput parametersN) rrrrrJrrrKrrLr+rrrrrI_ rIc@rH)Image2ImageTaskCreationRequest.rJrKrLN) rrrrrJrrrKrrLr:rrrrrNerMrNc@rH)Text2VideoTaskCreationRequest.rJrKrLN) rrrrrJrrrKr$rLr<rrrrrOkrMrOc@rH)Image2VideoTaskCreationRequest.rJrKrLN) rrrrrJrrrKr'rLrErrrrrPqrMrPc@rH)"Reference2VideoTaskCreationRequest.rJrKrLN) rrrrrJrrrKr)rLrGrrrrrQwrMrQc@s.eZdZUedZeed<edZeed<dS)TaskCreationOutputField.task_id task_statusN)rrrrrSrrrTrrrrrR}s rRc@sbeZdZUedZedBed<edZeed<edddZ edBed<edddZ edBed <dS) TaskCreationResponseNoutput. request_idz"Error code for the failed request.r.codez!Details about the failed request.message) rrrrrVrRrrWrrXrYrrrrrUs rUc@sJeZdZUedZedBed<edZedBed<edZedBed<dS) TaskResultNurlrXrY) rrrrr[rrrXrYrrrrrZs rZc@sFeZdZUedZeed<edZeed<edZe e dBed<dS)ImageTaskStatusOutputField.rSrTNresults) rrrrrSrrrTr]r#rZrrrrr\s r\c@sjeZdZUedZeed<edZeed<edZedBed<edZ edBed<edZ edBed<dS)VideoTaskStatusOutputField.rSrTN video_urlrXrY) rrrrrSrrrTr_rXrYrrrrr^s r^c@2eZdZUedZedBed<edZeed<dS)ImageTaskStatusResponseNrV.rW) rrrrrVr\rrWrrrrrra rac@r`)VideoTaskStatusResponseNrV.rW) rrrrrVr^rrWrrrrrrcrbrcu\((\d+)\s*[x×]\s*(\d+)\)c@sTeZdZeddZe      dded ed ed ed ed ededefddZdS)WanTextToImageApicCstjddddtjjddgdddtjjd d d d d tjjdd d dd dtjjdddddd dtjjdddddd dtjjdddddtjjd dd d tjjdd dd d d tjjd!d"d#d d d gtj gtj j tj j tj jgd tjd$d%d& S)'NrdzWan Text to Imageapi node/image/Wanz*Generates an image based on a text prompt.rJzwan2.5-t2i-preview Model to use.optionsdefaulttooltiprTQPrompt describing the elements and visual features. Supports English and Chinese. multilinerirjr)Negative prompt describing what to avoid.rnrirjoptionalwidthii )riminmaxsteprqheightr5rr1rSeed to use for generation.rirurvrw display_modecontrol_after_generaterjrqr61Whether to enhance the prompt with AI assistance.rirjrqadvancedr7F7Whether to add an AI-generated watermark to the result.{"type":"usd","usd":0.03}expr node_id display_namecategoryr/inputsoutputshidden is_api_node price_badge)rSchemaComborStringInt NumberDisplaynumberBooleanImageOutputHiddenauth_token_comfy_orgapi_key_comfy_org unique_id PriceBadgeclsrrr define_schemas ?zWanTextToImageApi.define_schemarkrsrTFrJrrrrrxr5r6r7c st|tdddtt|t||dt|d||||dddIdH} | js3td | jd | j t |td | jj d t d ddddIdH} t tt| jjdjIdHS)Nz:/proxy/wan/api/v1/services/aigc/text2image/image-synthesisPOSTpathmethod)rr*)r,r5r6r7rJrKrLresponse_modeldataAn unknown error occurred:  - /proxy/wan/api/v1/tasks/rcS|jjSNrVrTxrrrz+WanTextToImageApi.execute.. rstatus_extractorestimated_duration poll_intervalr)rrrUrIrr+rV ExceptionrXrYr rSrar NodeOutputr rr]r[) rrJrrrrrxr5r6r7initial_responseresponserrrexecutes6      "zWanTextToImageApi.executeN)rkrsrsrTF) rrr classmethodrrr8r9rrrrrrds6 R rdc@sHeZdZeddZe   ddedejded ed ed e f d d Z dS)WanImageToImageApicCstjddddtjjddgdddtjjd d d tjjd d dddtjjdd ddd dtjjdddddtjjd dd d tj jdddd d dgtj gtj j tj j tj jgd tjddd S)!NrzWan Image to ImagerezGenerates an image from one or two input images and a text prompt. The output image is currently fixed at 1.6 MP, and its aspect ratio matches the input image(s).rJzwan2.5-i2i-previewrfrgimagez=Single-image editing or multi-image fusion. Maximum 2 images.)rjrTrkrlrmrrorpr5rr1rryrzr7Frr~rrr)rrrrrrrrrrrrrrrrrrrrr'st =z WanImageToImageApi.define_schemarkrFrJrrrr5r7c st|}|dvrtd|dg}|D] } |dt| ddqt|tddd tt|t|||d t ||d d d IdH} | j sOt d| j d| j t|td| j jdtdddddIdH} ttt| j jdjIdHS)N)rrz&Expected 1 or 2 input images, but got .data:image/png;base64,i total_pixelsz;/proxy/wan/api/v1/services/aigc/image2image/image-synthesisrr)rrr")r5r7rrrrrrcSrrrrrrrrrz,WanImageToImageApi.execute..*rr)r ValueErrorappendrrrrUrNrr:rVrrXrYr rSrarrr rr]r[) rrJrrrr5r7n_imagesr"irrrrrrys>     "zWanImageToImageApi.executeN)rkrF) rrrrrrrrr8r9rrrrrr&s( Q rc@sleZdZeddZe        dd ed ed edededejdBdede de de defddZ dS)WanTextToVideoApicCs@tjddddtjjdddgddd tjjd d d d dtjjdd d dd dtjjdgddd dtjjdddddtjjdd dtjjdd ddtjjdddd d!tjjd d"d d# tj jd$d%d d&d d'tj jd(d d)d d d*tj jd+d%d,d d d*tjjd-d.d/gd0d d d1g tj gtj j tj jtj jgd tjtjddgd2d3d4d5 S)6NrzWan Text to Videoapi node/video/Wanz)Generates a video based on a text prompt.rJwan2.5-t2v-preview wan2.6-t2vrfrgrTrkrlrmrrorpr,) z480p: 1:1 (624x624)z480p: 16:9 (832x480)z480p: 9:16 (480x832)720p: 1:1 (960x960)720p: 16:9 (1280x720)720p: 9:16 (720x1280)720p: 4:3 (1088x832)720p: 3:4 (832x1088)1080p: 1:1 (1440x1440)1080p: 16:9 (1920x1080)1080p: 9:16 (1080x1920)1080p: 4:3 (1632x1248)1080p: 3:4 (1248x1632)rrhrirqr?r=r>z=A 15-second duration is available only for the Wan 2.6 model.rirurvrwr{rjrqrAUAudio must contain a clear, loud voice, without extraneous noise or background music.rqrjr5rr1rryrzgenerate_audioFr.xr=r)r RES_IN_PARENSsearchgroupsrr rrrUrOr$r<rVrrXrYr rSrcr8rrr r_)rrJrrr,r?rAr5rr6r7rCrrrxr%rrrrrr%sN      zWanTextToVideoApi.execute) rkrr=NrFTFrB) rrrrrrr8rrr9rrrrrrsH }    rc@sreZdZeddZe        dd ed ejd ededededej dBdede de de defddZ dS)WanImageToVideoApicCsJtjddddtjjdddgddd tjd tjjd d d ddtjjdd d dd dtjjdgddd dtjjdddddtjjdd dtj jdd ddtjjdd d d!d"tjjd d#d d$ tj jd%d&d d'd d(tj jd)d d*d d d+tj jd,d&d-d d d+tjjd.d/d0gd1d d d2g tj gtj jtj jtj jgd tjtjddgd3d4d5d6 S)7NrzWan Image to Videorz9Generates a video from the first frame and a text prompt.rJwan2.5-i2v-preview wan2.6-i2vrfrgrrTrkrlrmrrorprF)480P720P1080Prrr?r=r>z,Duration 15 available only for WAN2.6 model.rrArrr5rr1rryrzrFrrr6r}r~r7rrCrBrrrra ( $ppsTable := { "480p": 0.05, "720p": 0.1, "1080p": 0.15 }; $pps := $lookup($ppsTable, widgets.resolution); { "type": "usd", "usd": $round($pps * widgets.duration, 2) } ) rr)rrrrrrrrrrrrrrrrrrrrrrrr^s   [z WanImageToVideoApi.define_schemarkrr=NrFTrBrJrrrrFr?rAr5rr6r7rCc s4t|dkr tdd|vr|dkrtd|dkr#|dkr#tdd t|d d } d}|dur?t|d d dt|dd}t|tdddtt|t ||| |dt |||| | | | dddIdH}|j sqt d|j d|jt|td|j jdtdddt|dd d!IdH}tt|j jIdHS)"Nrz$Exactly one input image is required.rrz(The Wan 2.6 model does not support 480P.r>rrri =rrrrrrrrr)rrr(r%)rFr?r5rAr6r7rCrrrrrrcSrrrrrrrr rz,WanImageToVideoApi.execute..rr=rr)r rrrr rrrUrPr'rErVrrXrYr rSrcr8rrr r_)rrJrrrrFr?rAr5rr6r7rC image_urlr%rrrrrrsV     zWanImageToVideoApi.execute) rkrr=NrFTFrB) rrrrrrrrr8rr9rrrrrr]sL u     rc@sNeZdZeddZededededejjdede d e d ed e fd d Z dS)WanReferenceVideoApicCstjddddtjjddgdtjjdd d d d tjjd d d dd tjjdtjjtjdgddddtjjdgddtjjdddddtj j dtjjdddddtj j d dtjjdd d!gd"d d#tj jd$d%d&d d'g tj gtjjtjjtjjgd tjtjddgd(d)d*d+ S),NrzWan Reference to VideorzUse the character and voice from input videos, combined with a prompt, to generate a new video that maintains character consistency.rJz wan2.6-r2v)rhrTrkzPrompt describing the elements and visual features. Supports English and Chinese. Use identifiers such as `character1` and `character2` to refer to the reference characters.rmrroreference_videosreference_video) character1 character2 character3r)namesru)templater,) rrrrrrrrrrr?r= )rirurvrwr{r5rr1)rirurvrwr{r|rCrBrzSpecifies the shot type for the generated video, that is, whether the video is a single continuous shot or multiple shots with cuts.)rhrjrr7Fr)rirjrra ( $rate := $contains(widgets.size, "1080p") ? 0.15 : 0.10; $inputMin := 2 * $rate; $inputMax := 5 * $rate; $outputPrice := widgets.duration * $rate; { "type": "range_usd", "min_usd": $inputMin + $outputPrice, "max_usd": $inputMax + $outputPrice } ) rr)rrrrrAutogrow TemplateNamesrrrsliderrrrrrrrrrrrrrrs  Fz"WanReferenceVideoApi.define_schemarJrrrr,r?r5rCr7c sg} |D] } t|| dddq|D]} | t||| IdHqt|\} } t|tdddtt |t ||| dt | d| ||| |d d d IdH}|j s`t d |jd |jt|td|j jdtdddddIdH}tt|j jIdHS)Nr) min_duration max_durationrrr)rrr*r)r,r?rCr7r5rrrrrrcSrrrrrrrrrz.WanReferenceVideoApi.execute..ri)rrrmax_poll_attempts)rrrrrrrrrUrQr)rGrVrrXrYr rSrcrrr r_)rrJrrrr,r?r5rCr7r*rrrrxrrrrrrysH     zWanReferenceVideoApi.executeN) rrrrrrrrTyper8r9rrrrrrs. g  rc@s(eZdZedeeejfddZdS)WanApiExtensionreturncstttttgSr)rdrrrr)selfrrr get_node_listszWanApiExtension.get_node_listN) rrrrr#typer ComfyNoderrrrrrsrrcstSr)rrrrrcomfy_entrypointsr)6repydanticrrtyping_extensionsrcomfy_api.latestrrrcomfy_api_nodes.utilrr r r r r rrrrrrrr$r'r)r+r:r<rErGrIrNrOrPrQrRrUrZr\r^rarccompilerrrdrrrrrrrrrrsN 4   |84