o igT@sddlZddlmZddlmZddlmZmZmZm Z ddl m Z m Z m Z mZmZmZmZddlmZmZmZmZmZdZdd d d d d ZGdddejZGdddeZGdddejZGdddeZdefddZdS)N)BytesIO)override)IOComfyExtensionInput InputImpl)VeoGenVidPollRequestVeoGenVidPollResponseVeoGenVidRequestVeoGenVidResponseVeoRequestInstanceVeoRequestInstanceImageVeoRequestParameters) ApiEndpointdownload_url_to_video_outputpoll_opsync_optensor_to_base64_string veo-2.0-generate-001zveo-3.1-generate-previewzveo-3.1-fast-generate-previewveo-3.0-generate-001veo-3.0-fast-generate-001)rveo-3.1-generateveo-3.1-fast-generaterrc@s<eZdZdZeddZe      dd dZd S)VeoVideoGenerationNodez Generates videos from text prompts using Google's Veo API. This node can create videos from text descriptions and optional image inputs, with control over parameters like aspect ratio, duration, and more. cCstjddddtjjddddd tjjd d d gd d dtjjddddddtjjdddddtjjdddtjjddddddtjjdddgdddddtjjd d!d!d"dtjjdd#dd$ tj jd%d&dd'tjjd(d)gd)d*dd+g tj gtj j tj jtj jgdtjtjdgd,d-d.d/ S)0NrzGoogle Veo 2 Video Generationapi node/video/Veoz;Generates videos from text prompts using Google's Veo 2 APIpromptTText description of the video multilinedefaulttooltip aspect_ratio16:99:16 Aspect ratio of the output videooptionsr!r"negative_prompt8Negative text prompt to guide what to avoid in the videor r!r"optionalduration_seconds'Duration of the output video in secondsr!minmaxstep display_moder"r,enhance_promptz0Whether to enhance the prompt with AI assistancer!r"r,advancedperson_generationALLOWBLOCK/Whether to allow generating people in the videor(r!r"r,r9seedr(Seed for video generation (0 for random)r!r3r4r5r6control_after_generater"r,image2Optional reference image to guide video generationr"r,modelrz'Veo 2 model to use for video generationr(r!r"r,widgetsz4{"type":"usd","usd": 0.5 * widgets.duration_seconds} depends_onexpr node_id display_namecategory descriptioninputsoutputshidden is_api_node price_badgerSchemaStringrComboInt NumberDisplaynumberBooleanImageVideoOutputHiddenauth_token_comfy_orgapi_key_comfy_org unique_id PriceBadgePriceBadgeDependsclsrk8/mnt/c/Users/fbmor/ComfyUI/comfy_api_nodes/nodes_veo2.py define_schema*s  F z$VeoVideoGenerationNode.define_schemar$rr.Tr;rNrFc  s*t| } g} d|i} |durt|} | r| dd| d<| | ||||d}|r.||d<|dkr6||d<| d d krE| |d <d |d <t|td| dddtt| |ddIdH}dd}t|td| dddt |t |j ddt dIdH}|j rtd|j jd|j jdt|jdr|jjdkrt|jdr|jjr|jjd}d|d |jjd!}t|d"|jjd!}t||jrt|jd#r|jjrt|jjdkr|jjd}t|d$r|jrtttt|jSt|d%r |jr tt |jIdHStd&td')(Nr image/pngbytesBase64EncodedmimeTyperD) aspectRatiopersonGenerationdurationSeconds enhancePromptnegativePromptrr?zveo-2.0 generateAudioTru /proxy/veo/ /generatePOSTpathmethod instances parametersresponse_modeldatacS|jrdSdSN completedpendingdone)responserkrkrlstatus_extractorsz8VeoVideoGenerationNode.execute..status_extractor/poll operationName@rrr poll_intervalestimated_durationVeo API error:  (code: )raiMediaFilteredCountraiMediaFilteredReasonsz7Content filtered by Google's Responsible AI practices:  (z videos filtered.)z7Content filtered by Google's Responsible AI practices (videosrpgcsUri.Video returned but no data or URL was provided4Video generation completed but no video was returned)! MODELS_MAPrappendfindrrr r rr rnameAVERAGE_DURATION_VIDEO_GENerror Exceptionmessagecodehasattrrrrrlenrpr NodeOutputr VideoFromFilerbase64 b64decoderr)rjrr#r)r-r7r:r?rDrGgenerate_audiorinstance image_base64rinitial_responser poll_responsereason error_messagevideorkrkrlexecutes          zVeoVideoGenerationNode.execute) r$rr.Tr;rNrF)__name__ __module__ __qualname____doc__ classmethodrmrrkrkrkrlr"s Zrc@seZdZdZeddZdS)Veo3VideoGenerationNodea Generates videos from text prompts using Google's Veo 3 API. Supported models: - veo-3.0-generate-001 - veo-3.0-fast-generate-001 This node extends the base Veo node with Veo 3 specific features including audio generation and fixed 8-second duration. cCs.tjddddtjjddddd tjjd d d gd d dtjjddddddtjjdddddtjjdddtjjddddddtjjdddgdddddtjjdd d d!dtjjdd"dd# tj jd$d%dd&tjjd'gd(d)d*dd+tjjd,d-d.dd/g tj gtj j tj jtj jgdtjtjd'd,gd0d1d2d3 S)4NrzGoogle Veo 3 Video Generationrz;Generates videos from text prompts using Google's Veo 3 APIrTrrrr#r$r%r&r'r)r*r+r-r/r0zGDuration of the output video in seconds (Veo 3 only supports 8 seconds)r2r7z)This parameter is deprecated and ignored.r8r:r;r<r=r>r?rr@rArBrDrErFrG)rrrrrz'Veo 3 model to use for video generationrHrFzsz0Veo3FirstLastFrameNode.execute..rrrrrrz: rrz2Content blocked by Google's Responsible AI filtersrz videor0sz filtered).rr) rrrr r r r rrrr rrrrrrrrrrrrprrrrrrrrr)rjrr)rr#rr?rrrGrrrrfiltered_countreasons reason_partrrkrkrlrs|      zVeo3FirstLastFrameNode.executeN) rrrrrmstrintrr`boolrrkrkrkrlrws2 ]   rc@s(eZdZedeeejfddZdS) VeoExtensionreturncs tttgSN)rrr)selfrkrkrl get_node_list&s zVeoExtension.get_node_listN) rrrrlisttyper ComfyNoderrkrkrkrlr%srrcstSr)rrkrkrkrlcomfy_entrypoint/sr) riortyping_extensionsrcomfy_api.latestrrrrcomfy_api_nodes.apis.veorr r r r r rcomfy_api_nodes.utilrrrrrrrrrrrrrrkrkrkrls(  $  Y}/