o ih@sddlZddlZddlZddlmZddlmZmZmZddl m Z m Z m Z m Z mZmZmZmZmZmZmZmZmZmZddlmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#dZ$ddd d Z%d Z&d Z'd d hZ(e)e*Z+dede,fddZ-Gdddej.Z/Gdddej.Z0Gdddej.Z1Gdddej.Z2Gdddej.Z3Gdddej.Z4de5ej.dee Bd e6dBdej7fd!d"Z8d#e,d$e9e,ddfd%d&Z:ej;ejde>fd-d.Z?dS)/N)override)IOComfyExtensionInput)RECOMMENDED_PRESETSRECOMMENDED_PRESETS_SEEDREAM_4VIDEO_TASKS_EXECUTION_TIMEImage2VideoTaskCreationRequestImageTaskCreationResponseSeedream4OptionsSeedream4TaskCreationRequestTaskCreationResponseTaskImageContentTaskImageContentUrlTaskStatusResponseTaskTextContentText2ImageTaskCreationRequestText2VideoTaskCreationRequest) ApiEndpointdownload_url_to_image_tensordownload_url_to_video_outputget_number_of_imagesimage_tensor_pair_to_batchpoll_opsync_opupload_images_to_comfyapivalidate_image_aspect_ratiovalidate_image_dimensionsvalidate_stringz)/proxy/byteplus/api/v3/images/generationsseedream-5-0-260128seedream-4-5-251128seedream-4-0-250828)zseedream 5.0 liter r!z1/proxy/byteplus/api/v3/contents/generations/tasksseedance-1-0-lite-t2v-250428seedance-1-0-lite-i2v-250428responsereturncCsX|jrd|jdd|jd}t|t|td|jdd|jddS)Nz ByteDance request failed. Code: codez , message: messagez'ByteDance task succeeded, image URL: %srurl)errorlogginginfo RuntimeErrordata)r$ error_msgr/=/mnt/c/Users/fbmor/ComfyUI/comfy_api_nodes/nodes_bytedance.pyget_image_url_from_response7s  r1c@sLeZdZeddZededededededed ed ed e j fd d Z dS)ByteDanceImageNodecCstjddddtjjddgdtjjdd d d tjjd d dtDddtjjdddddddtjjdddddddtjjdddddtjjd dd d tj jd d!d"d#d$tjjd%d d&tj jd'd(d)d d d*gtj gtj jtj jtj jgd tjd+d,d d- S).Nr2zByteDance Imageapi node/image/ByteDancez>Generate images using ByteDance models via api based on promptmodelzseedream-3-0-t2i-250415optionspromptTz*The text prompt used to generate the image multilinetooltip size_presetcSg|]\}}}|qSr/r/.0label_r/r/r0 Rz4ByteDanceImageNode.define_schema..zHPick a recommended size. Select Custom to use the width and height belowr6r:width@QCustom width for image. Value is working only if `size_preset` is set to `Custom`)defaultminmaxstepr:heightRCustom height for image. Value is working only if `size_preset` is set to `Custom`seedrzSeed to use for generationrJrKrLrM display_modecontrol_after_generater:optionalguidance_scaleg@g?$@g{Gz?z;Higher value makes the image follow the prompt more closelyrJrKrLrMrTr:rV watermarkFz7Whether to add an "AI generated" watermark to the imagerJr:rVadvancedz{"type":"usd","usd":0.03})expr) node_id display_namecategory descriptioninputsoutputshidden is_api_node price_badge is_deprecated)rSchemaComborStringrInt NumberDisplaynumberFloatBooleanImageOutputHiddenauth_token_comfy_orgapi_key_comfy_org unique_id PriceBadgeclsr/r/r0 define_schemaBs   :z ByteDanceImageNode.define_schemar4r7r;rDrNrPrWrZr%c st|dddd} } tD]\} } } | |kr| | } } nq| dus'| durN||} } d| kr6dkrCnn d| krBdksNntd| d| dt||| d| |||d }t|ttd d |td IdH}t t t |IdHS) NTrRstrip_whitespace min_lengthrFrGzCustom size out of range: xz<. Both width and height must be between 512 and 2048 pixels.)r4r7sizerPrWrZPOSTpathmethodr-response_model) rr ValueErrorrrrBYTEPLUS_IMAGE_ENDPOINTr r NodeOutputrr1)rxr4r7r;rDrNrPrWrZwhr?twthpayloadr$r/r/r0executes:   .   zByteDanceImageNode.executeN) __name__ __module__ __qualname__ classmethodrystrintfloatboolrrrr/r/r/r0r2@s. N  r2c@szeZdZeddZededdddddddd f d ed ed ejdBd ede de dede de de de de j fddZ dS)ByteDanceSeedreamNodecCsHtjddddtjjdttdtjjddd d d tjjd d ddtjjdddt Dddtj jddddddddtj jddddddddtjjddd gd!dd"tj jd#d$d$d%d$tj j d&dd'tj jd(d)d)d*d$tj j dd+dd, tj jd-d.d/ddd0tj jd1dd2ddd0g tjgtjjtjjtjjgdtjtjdgd3d4d5d6 S)7NrzByteDance Seedream 4.5 & 5.0r3z\Unified text-to-image generation and precise single-sentence editing at up to 4K resolution.r4r5r7Tz-Text prompt for creating or editing an image.)r9rJr:imagezjInput image(s) for image-to-image generation. Reference image(s) for single or multi-reference generation.)r:rVr;cSr<r/r/r=r/r/r0rArBz7ByteDanceSeedreamNode.define_schema..zIPick a recommended size. Select Custom to use the width and height below.rCrDrGrEi`rI)rJrKrLrMr:rVrNirOsequential_image_generationdisabledautozGroup image generation mode. 'disabled' generates a single image. 'auto' lets the model decide whether to generate multiple related images (e.g., story scenes, character variations).)r6r:rV max_imagesrRzMaximum number of images to generate when sequential_image_generation='auto'. Total images (input + generated) cannot exceed 15.rYrPrrQSeed to use for generation.rSrZFz8Whether to add an "AI generated" watermark to the image.r[fail_on_partialzSIf enabled, abort execution if any requested images are missing or return an error.widgetsa ( $price := $contains(widgets.model, "5.0 lite") ? 0.035 : $contains(widgets.model, "4-5") ? 0.04 : 0.03; { "type":"usd", "usd": $price, "format": { "suffix":" x images/Run", "approximate": true } } )  depends_onr] r^r_r`rarbrcrdrerf)rrhrirlistSEEDREAM_MODELSkeysrjrprrkrlrmrorqrrrsrtrurvPriceBadgeDependsrwr/r/r0rys       W z#ByteDanceSeedreamNode.define_schemaNrrGrrRFTr4r7rr;rDrNrrrPrZrr%c sdt|}t|dddd} } tD]\}}}||kr"||} } nq| dus+| dur0||} } | | }|d}d|vs@d|vrM|dkrMtd|d d d |vr^|d kr^td |d d d|vrddnd}||krytd|dd d|d d |durt|nd}|dkrdnd}||krtd|d|d|dkr||dkrtdg}|r|D]}t|ddqt|||dd IdH}t|tt d!d"t t |||| d#| | |t |d$| |dkrd%ndd& d'IdH}t |jdkrttt|IdHSd(d)|jD}| r"t |t |jkr"td*t |d+t |jd,ttd-d)|DIdHS).NTrRrzg.Az seedream-4-5z seedream-5-0i@8z?Minimum image resolution for the selected model is 3.68MP, but z.2fz MP provided.z seedream-4-0izMMinimum image resolution that the selected model can generate is 0.92MP, but ižiz3Maximum image resolution for the selected model is i@BzMP, but rr z Maximum of z% reference images are supported, but z received.rrz\The maximum number of generated images plus the number of reference images cannot exceed 15.)rR)rrR image/pngr mime_typerrr}rpng) r4r7rr~rPr#sequential_image_generation_optionsrZ output_format)rr-cSs*g|]}t|trd|vrt|dqSr() isinstancedictr)r>dr/r/r0rAs*z1ByteDanceSeedreamNode.execute..zOnly z of z$ images were generated before error.csg|] }t|IdHqSN)rr>ir/r/r0rAs)rrrrrrrrrrr r r lenr-rrrr1r,torchcat)rxr4r7rr;rDrNrrrPrZrrrr?rrout_num_pixels mp_provided max_pixelsn_input_imagesmax_num_of_imagesreference_images_urlsrr$urlsr/r/r0r4s        zByteDanceSeedreamNode.execute)rrrrryrrrrprrrrrr/r/r/r0rsL u      rc@sTeZdZeddZe ddededededed ed ed ed ed ej fddZ dS)ByteDanceTextToVideoNodecCstjddddtjjdgdddtjjd d d d tjjd gdddtjjdgdddtjjddddddtjjdtjjdddddtjjd dd d tj jd d!d"d d d#tj jd$d!d%d d d#tj jd&d!d'd d d#g tj gtj j tj jtj jgd td( S))NrzByteDance Text to Videoapi node/video/ByteDancez=Generate video using ByteDance models via api based on promptr4)seedance-1-5-pro-251215seedance-1-0-pro-250528r"seedance-1-0-pro-fast-251015rr6rJr7T+The text prompt used to generate the video.r8 resolution480p720p1080p#The resolution of the output video.rC aspect_ratio)16:94:31:13:49:1621:9%The aspect ratio of the output video.durationr rR,The duration of the output video in seconds.rJrKrLrMr:rTrPrrQrrS camera_fixedFSpecifies whether to fix the camera. The platform appends an instruction to fix the camera to your prompt, but does not guarantee the actual effect.r[rZ8Whether to add an "AI generated" watermark to the video.generate_audio@This parameter is ignored for any model except seedance-1-5-pro.r)rrhrirrjrkrlsliderrmroVideorqrrrsrtruPRICE_BADGE_VIDEOrwr/r/r0rys   Fz&ByteDanceTextToVideoNode.define_schemaFr4r7rrrrPrrZrr%c s|dkr |dkr tdt|dddt|gd|d|d |d |d |d t|d t| }t|t|t|dg|dkrH| nddtdt t |||ddIdHS)Nr=Minimum supported duration for Seedance 1.5 Pro is 4 seconds.TrRrzrratiorrP camerafixedrZ --resolution --ratio --duration  --seed  --camerafixed --watermark textr4contentrrXrestimated_duration) rrraise_if_text_paramsrlowerprocess_video_taskrrrLmathceilr) rxr4r7rrrrPrrZrr/r/r0rs8      z ByteDanceTextToVideoNode.executeNF) rrrrryrrrrrrr/r/r/r0rs4 W    rc@sZeZdZeddZe ddededejdeded ed ed e d e d e de j fddZ dS)ByteDanceImageToVideoNodecCstjddddtjjdgdddtjjd d d d tjjd ddtjjdgdddtjjdgdddtjjddddddtjjdtjjdddd dtjj d d!d d" tj jd#d$d%d d d&tj jd'd$d(d d d&tj jd)d$d*d d d&g tj gtj jtj jtj jgd td+ S),NrzByteDance Image to VideorzGGenerate video using ByteDance models via api based on image and promptr4)rrr#rrrr7Trr8r%First frame to be used for the video.r:rrrrCradaptiverrrrrrrrrrrrRrrrPrrQrrSrFrr[rZrrrrrrhrirrjrprkrlrrmrorrqrrrsrtrurrwr/r/r0rys   Jz'ByteDanceImageToVideoNode.define_schemaFr4r7rrrrrPrrZrr%c s|dkr |dkr tdt|dddt|gdt|ddd d d t|d d d dt||ddIdHd} |d|d|d|d|dt|dt|  }t|t |t |dt t | ddg|dkrl| nddt dtt|||ddIdHS)NrrrTrRrzr,p min_width min_height max_width max_heightrrrrFstrictrrrrrrrrrr) image_urlrrXr)rrrrrrrrrr rrrrLrrr) rxr4r7rrrrrPrrZrrr/r/r0rjs>    z!ByteDanceImageToVideoNode.executeNr rrrrryrrrprrrrrr/r/r/r0r s8 [     rc@s`eZdZeddZe ddededejdejded ed ed ed e d e de de j fddZ dS)ByteDanceFirstLastFrameNodecCstjddddtjjdgdddtjjd d d d tjjd ddtjjdddtjjdgdddtjjdgdddtjjddddddtjjdtjjd d!d!d"dtjj d d#d d$ tj jd%d&d'd d d(tj jd)d&d*d d d(tj jd+d&d,d d d(g tj gtj jtj jtj jgd td- S).Nrz#ByteDance First-Last-Frame to Videorz6Generate video using prompt and first and last frames.r4)rrr#r#rr7Trr8 first_framerr last_framez$Last frame to be used for the video.rrrrCrrrrrrrrRrrrPrrQrrSrFrr[rZrrrrrrwr/r/r0rys  Iz)ByteDanceFirstLastFrameNode.define_schemaFr4r7rrrrrrPrrZrr%c s<|dkr |dkr tdt|dddt|gd||fD]} t| ddd d d t| d d d dqt|t||dddIdH} |d|d|d|d|dt| dt|  }t |t |t |dt t t| ddddt t t| ddddg|dkr| nddtdtt|||dd IdHS)!NrrrTrRrzrrrrr r Fr rrrrrrrrrrrrrrrolerrrXr)rrrrrrrrrrr rrrrLrrr)rxr4r7rrrrrrPrrZrr download_urlsr/r/r0rsP      z#ByteDanceFirstLastFrameNode.executeNrrr/r/r/r0rs< Z      rc@sNeZdZeddZedededejdededed ed e d e j fd d Z dS)ByteDanceImageReferenceNodecCstjddddtjjdddgddtjjd d d d tjjd ddtjjdddgddtjjdgdddtjjddddddtjjdtjjdd d d!dtjj d d"d d# tj jd$d%d&d d d'gtj gtj jtj jtj jgd tjtjgd(d)d*d+d, S)-Nrz#ByteDance Reference Images to Videorz1Generate video using prompt and reference images.r4rr#rr7Trr8imageszOne to four images.rrrrrrCrrrrrrrrRrrrPrrQrrSrZFrr[)r4rrra( ( $priceByModel := { "seedance-1-0-pro": { "480p":[0.23,0.24], "720p":[0.51,0.56] }, "seedance-1-0-lite": { "480p":[0.17,0.18], "720p":[0.37,0.41] } }; $model := widgets.model; $modelKey := $contains($model, "seedance-1-0-pro") ? "seedance-1-0-pro" : "seedance-1-0-lite"; $resolution := widgets.resolution; $resKey := $contains($resolution, "720") ? "720p" : "480p"; $modelPrices := $lookup($priceByModel, $modelKey); $baseRange := $lookup($modelPrices, $resKey); $min10s := $baseRange[0]; $max10s := $baseRange[1]; $scale := widgets.duration / 10; $minCost := $min10s * $scale; $maxCost := $max10s * $scale; ($minCost = $maxCost) ? {"type":"usd","usd": $minCost} : {"type":"range_usd","min_usd": $minCost, "max_usd": $maxCost} ) rr)rrhrirrjrprkrlrrmrorrqrrrsrtrurvrrwr/r/r0ry+s  6z)ByteDanceImageReferenceNode.define_schemar4r7rrrrrPrZr%c st|dddt|gd|D]} t| dddddt| dd d d qt||d d dIdH} |d|d|d|d|dt| }t|dgdd| D} t|t || ddt dt t |||ddIdHS)NTrRrz)rrrrPrZrrrr r Fr rrrrrrrrrcSs"g|] }ttt|dddqS)rreference_imager)rrrrr/r/r0rAs"z7ByteDanceImageReferenceNode.execute..rrXr)rrrrrrrrrr rLrrr) rxr4r7rrrrrPrZr image_urlsr}r/r/r0rs8      z#ByteDanceImageReferenceNode.executeNrr/r/r/r0r)s. i  rrxrrcs~|jtvr td|jt|ttdd|tdIdH}t|tt d|j ddd|t d IdH}t t|jjIdHS) NzModel '%s' is deprecated and will be deactivated on May 13, 2026. Please switch to a newer model. Recommended: seedance-1-0-pro-fast-251015.rrr/)rcSs|jSr)status)rr/r/r0sz$process_video_task..)status_extractorrr)r4DEPRECATED_MODELSloggerwarningrrBYTEPLUS_TASK_ENDPOINTr rBYTEPLUS_TASK_STATUS_ENDPOINTidrrrrr video_url)rxrrinitial_responser$r/r/r0rs(    rr7 text_paramscCs.|D]}d|d|vrtd|dqdS)Nz-- zV is not allowed in the prompt, use the appropriated widget input to change this value.)r)r7r(rr/r/r0rs r)r4rrrra ( $priceByModel := { "seedance-1-5-pro": { "480p":[0.12,0.12], "720p":[0.26,0.26], "1080p":[0.58,0.59] }, "seedance-1-0-pro": { "480p":[0.23,0.24], "720p":[0.51,0.56], "1080p":[1.18,1.22] }, "seedance-1-0-pro-fast": { "480p":[0.09,0.1], "720p":[0.21,0.23], "1080p":[0.47,0.49] }, "seedance-1-0-lite": { "480p":[0.17,0.18], "720p":[0.37,0.41], "1080p":[0.85,0.88] } }; $model := widgets.model; $modelKey := $contains($model, "seedance-1-5-pro") ? "seedance-1-5-pro" : $contains($model, "seedance-1-0-pro-fast") ? "seedance-1-0-pro-fast" : $contains($model, "seedance-1-0-pro") ? "seedance-1-0-pro" : "seedance-1-0-lite"; $resolution := widgets.resolution; $resKey := $contains($resolution, "1080") ? "1080p" : $contains($resolution, "720") ? "720p" : "480p"; $modelPrices := $lookup($priceByModel, $modelKey); $baseRange := $lookup($modelPrices, $resKey); $min10s := $baseRange[0]; $max10s := $baseRange[1]; $scale := widgets.duration / 10; $audioMultiplier := ($modelKey = "seedance-1-5-pro" and widgets.generate_audio) ? 2 : 1; $minCost := $min10s * $scale * $audioMultiplier; $maxCost := $max10s * $scale * $audioMultiplier; ($minCost = $maxCost) ? {"type":"usd","usd": $minCost, "format": { "approximate": true }} : {"type":"range_usd","min_usd": $minCost, "max_usd": $maxCost, "format": { "approximate": true }} ) rc@s(eZdZedeeejfddZdS)ByteDanceExtensionr%csttttttgSr)r2rrrrr)selfr/r/r0 get_node_listsz ByteDanceExtension.get_node_listN) rrrrrtyper ComfyNoder,r/r/r/r0r*sr*cstSr)r*r/r/r/r0comfy_entrypoint sr/)@r*rrtyping_extensionsrcomfy_api.latestrrrcomfy_api_nodes.apis.bytedancerrrr r r r r rrrrrrcomfy_api_nodes.utilrrrrrrrrrrrrrr#r$r  getLoggerrr!rr1r.r2rrrrrr-rrrrrrvrrr*r/r/r/r/r0sX @4  |Q  5