o iQ@sddlZddlmZddlmZmZmZddlmZm Z m Z m Z m Z ddl mZmZmZmZmZmZmZmZmZmZdZdZdZd Zd Zd Zd Zd Zd Z d Z!d Z"d Z#d Z$d Z%dZ&dede'fddZ(d8ddZ)dej*dej*fddZ+dej*de,e-e-ffddZ.de-de-ddfddZ/dej*dej*fddZ0d e1ddfd!d"Z2dej*d e1dej*fd#d$Z3d%e4fd&d'Z5d(d)Z6d*e7ej8d+e4defd,d-Z9Gd.d/d/ej8Z:Gd0d1d1ej8Z;Gd2d3d3ej8ZdS)9N)override)IOComfyExtensionInput)MoonvalleyPromptResponse$MoonvalleyTextToVideoInferenceParamsMoonvalleyTextToVideoRequest%MoonvalleyVideoToVideoInferenceParamsMoonvalleyVideoToVideoRequest) ApiEndpointdownload_url_to_video_outputpoll_opsync_op trim_videoupload_images_to_comfyapiupload_video_to_comfyapi validate_container_format_is_mp4validate_image_dimensionsvalidate_stringz/proxy/moonvalley/uploadsz/proxy/moonvalley/promptsz(/proxy/moonvalley/prompts/video-to-videoz'/proxy/moonvalley/prompts/text-to-videoz(/proxy/moonvalley/prompts/image-to-video,i'i@iresponsereturncCs t|jS)z6Verifies that the initial response contains a task ID.)boolid)rr>/mnt/c/Users/fbmor/ComfyUI/comfy_api_nodes/nodes_moonvalley.pyis_valid_task_creation_response1s rcCs8t|sd|jd|jd|}t|t|dS)Nz4Moonvalley Marey API: Initial request failed. Code: z , Message: z, Data: )rcodemessageloggingerror RuntimeError)r error_msgrrrvalidate_task_creation_response6s  r#videocCs&t|\}}t||t|t|S)aF Validates and processes video input for Moonvalley Video-to-Video generation. Args: video: Input video to validate Returns: Validated and potentially trimmed video Raises: ValueError: If video doesn't meet requirements MoonvalleyApiError: If video duration is too short )_get_video_dimensions_validate_video_dimensionsr_validate_and_trim_duration)r$widthheightrrrvalidate_video_to_video_input=s  r*c Cs@z|WSty}ztd|td||d}~ww)z.Extracts video dimensions with error handling.z%Error getting dimensions of video: %szCannot get video dimensions: N)get_dimensions Exceptionrr ValueError)r$errrr%Rs  r%r(r)cCsJhd}||f|vr#dddt|D}td|d|d|dS) z8r1r1r1r0r4r3z, cSsg|] \}}|d|qS)xr).0whrrr fsz._validate_video_dimensions..z Resolution r8z not supported. Supported: N)joinsortedr-)r(r)supported_resolutionssupported_listrrrr&[s  r&cCs|}t|t||S)z:Validates video duration and trims to 5 seconds if needed.) get_duration_validate_minimum_duration_trim_if_too_longr$durationrrrr'js r'rEcCs|dkrtddS)z)Ensures video is at least 5 seconds long.z,Input video must be at least 5 seconds long.N)r-)rErrrrBqsrBcCs|dkr t|dS|S)z#Trims video to 5 seconds if longer.rF)rrDrrrrCws rC resolutioncCs@dddddddddddddddd}||dddS)Nr4r3)r(r)r1r016:9 (1920 x 1080)9:16 (1080 x 1920)1:1 (1152 x 1152)4:3 (1536 x 1152)3:4 (1152 x 1536)get)rGres_maprrrparse_width_height_from_res~srQcCsddddd}|||dS)Nmotion_control canny_control pose_control depth_control)Motion TransferCanny Pose TransferDepthrVrN)value control_maprrrparse_control_parameters r\clstask_idcs0t|ttd|dtdddddIdHS)N/)pathcSs|r|jr|jSdSN)status)rrrrszget_response..g0@)response_modelstatus_extractor poll_intervalmax_poll_attempts)r r API_PROMPTS_ENDPOINTr)r]r^rrr get_responses rkc@sReZdZedejfddZedejde de de de d e d e dej fd d Z d S)MoonvalleyImg2VideoNodercCstjddddtjjdddtjjdd d tjjd d d d dtjjdgddddtjjdddddddtjjdddddtjj dd d tjjd!d"d#d$dd%dgtj gtj j tj jtj jgd tjtd&d'd( S))NrlzMoonvalley Marey Image to Videoapi node/video/Moonvalley Mareyz$Moonvalley Marey Image to Video Nodeimagez.The reference image used to generate the videotooltippromptT multilinenegative_prompt gopro, bright, contrast, static, overexposed, vignette, artifacts, still, noise, texture, scanlines, videogame, 360 camera, VR, transition, flare, saturation, distorted, warped, wide angle, saturated, vibrant, glowing, cross dissolve, cheesy, ugly hands, mutated hands, mutant, disfigured, extra fingers, blown out, horrible, blurry, worst quality, bad, dissolve, melt, fade in, fade out, wobbly, weird, low quality, plastic, stock footage, video camera, boringNegative prompt textrsdefaultrprGrHrIResolution of the output videooptionsrxrpprompt_adherence@?4@%Guidance scale for generation controlrxminmaxsteprpseed rRandom seed valuerxrrr display_moderpcontrol_after_generatestepsPKdzNumber of denoising steps{"type":"usd","usd": 1.5} depends_onexpr node_id display_namecategory descriptioninputsoutputshidden is_api_node price_badge)rSchemaImagerStringComboFloatInt NumberDisplaynumberVideoOutputHiddenauth_token_comfy_orgapi_key_comfy_org unique_id PriceBadgePriceBadgeDependsr]rrr define_schemas~    <z%MoonvalleyImg2VideoNode.define_schemarnrqrtrGr|rrc st|ddttdt|dtdt|dtdt|}t|||||d|dd d } d } t||d| d IdHd } t|t t ddt t | || ddIdH} t | t|| jIdH} t| jIdH}t|S)Nr) min_width min_height max_height max_widthr min_length max_lengthrt field_namerr(r)T)rtrrguidance_scaler(r)use_negative_promptsz image/png) max_images mime_typerPOSTr`method) image_url prompt_textinference_paramsendpointrfdata)r MAX_HEIGHT MAX_WIDTHr"MOONVALLEY_MAREY_MAX_PROMPT_LENGTHrQrrrr API_IMG2VIDEO_ENDPOINTrrr#rkrr output_urlr NodeOutput)r]rnrqrtrGr|rr width_heightrrrtask_creation_responsefinal_responser$rrrexecutes8     zMoonvalleyImg2VideoNode.executeN)__name__ __module__ __qualname__ classmethodrrrrrstrfloatintrrrrrrrls*O rlc@sbeZdZedejfddZe     dd ed ed ed e j dBd ededBdej fddZ dS)MoonvalleyVideo2VideoNodercCstjddddtjjddddtjjd dd d d tjjd ddddtjjdddtjjdddtjjdddgdddtjjddddddddtjjd d!d!ddtjjd"d#gtj gtj j tj j tj j gdtjtd$d%d& S)'NrzMoonvalley Marey Video to VideormrqTzDescribes the video to generate)rsrprtrurvrwrrrrrrFrr$zThe reference video used to generate the output video. Must be at least 5 seconds long. Videos longer than 5s will be automatically trimmed. Only MP4 format supported.ro control_typerVrX)r{rxoptionalmotion_intensityrz.Only used if control_type is 'Motion Transfer')rxrrrrprr<zNumber of inference steps)rxrrrrrpz{"type":"usd","usd": 2.25}rr)rrrrrrrrrrrrrrrrrrrrrs    9z'MoonvalleyVideo2VideoNode.define_schemaNrVrrr}rqrtrr$rrc  st|} t|| IdH} t|dtdt|dtdi} |dkr)|dur)|| d<t||| ||d} t|ttdd tt t || || d d IdH} t | t || j IdH}tt|jIdHS) NrrrtrrVr)rtrcontrol_paramsrrrr)r video_urlrrr)r*rrrr rr API_VIDEO2VIDEO_ENDPOINTrr r\r#rkrrrr r)r]rqrtrr$rrrr|validated_videorrrrrrrrrls:    z!MoonvalleyVideo2VideoNode.execute)NrVrrr}) rrrrrrrrrrrrrrrrrrs0L rc@sLeZdZedejfddZededededede d e dej fd d Z d S) MoonvalleyTxt2VideoNodercCstjddddtjjdddtjjddd d d tjjd gd dddtjjdddddddtjjdddddtjjdddtjjddd d!dd"dgtj gtj j tj j tj jgdtjtd#d$d% S)&NrzMoonvalley Marey Text to VideormrrqTrrrtrurvrwrG)rIrJrKrLrMz21:9 (2560 x 1080)rIryrzr|g@r~rrrrrrrrr)rxrrrrrrprrrrzInference stepsrrr)rrrrrrrrrrrrrrrrrrrrrrsv    8z%MoonvalleyTxt2VideoNode.define_schemarqrtrGr|rrc st|dtdt|dtdt|}t||||d|d|dd}t|ttd d tt||d d IdH} t | t || j IdH} t t| jIdHS) Nrrrtrr(r))rtrrr num_framesr(r)rr)rrr)rrrQrrr API_TXT2VIDEO_ENDPOINTrrr#rkrrrr r) r]rqrtrGr|rrrrrrrrrrs,     zMoonvalleyTxt2VideoNode.executeN) rrrrrrrrrrrrrrrrrs&Krc@s(eZdZedeeejfddZdS)MoonvalleyExtensionrcs tttgSra)rlrr)selfrrr get_node_list s z!MoonvalleyExtension.get_node_listN) rrrrlisttyper ComfyNoderrrrrr srcstSra)rrrrrcomfy_entrypointsr)rN)?rtyping_extensionsrcomfy_api.latestrrrcomfy_api_nodes.apis.moonvalleyrrrr r comfy_api_nodes.utilr r r rrrrrrrAPI_UPLOADS_ENDPOINTrjrrr MIN_WIDTH MIN_HEIGHTrr MIN_VID_WIDTHMIN_VID_HEIGHT MAX_VID_WIDTHMAX_VID_HEIGHTMAX_VIDEO_SIZErrrr#rr*tuplerr%r&r'rrBrCrrQr\rrrkrlrrrrrrrrsH 0      }}q