o iA@sddlmZddlmZmZddlmZddlmZm Z m Z m Z ddl m Z mZmZmZmZdddZGd d d eZejejgd d d dZGdddejZGdddejZGddde ZdefddZdS))BytesIO) BaseModelField)override)IOComfyExtensionInput InputImpl) ApiEndpointget_number_of_images sync_op_rawupload_images_to_comfyapivalidate_stringz ltx-2-proz ltx-2-fast)z LTX-2 (Pro) LTX-2 (Fast)c@seZdZUedZeed<edZeed<edZe ed<edZ eed<edZ e dBed<ed Z e dBed <edZedBed <dS) ExecuteTaskRequest.promptmodelduration resolutionNfpsTgenerate_audio image_uri)__name__ __module__ __qualname__rrstr__annotations__rrintrrrboolrr r 8/mnt/c/Users/fbmor/ComfyUI/comfy_api_nodes/nodes_ltxv.pyrs r)rrr)widgetsaw ( $prices := { "ltx-2 (pro)": {"1920x1080":0.06,"2560x1440":0.12,"3840x2160":0.24}, "ltx-2 (fast)": {"1920x1080":0.04,"2560x1440":0.08,"3840x2160":0.16} }; $modelPrices := $lookup($prices, $lowercase(widgets.model)); $pps := $lookup($modelPrices, widgets.resolution); {"type":"usd","usd": $pps * widgets.duration} ) ) depends_onexprc@sJeZdZeddZe  ddedededed ed ed ej fd d Z dS)TextToVideoNodecCstjddddtjjdttdtjjddd d tjjd gd d dtjjdgddtjjdddgddtjjddddddgtj gtj j tj j tj jgdtd S)NLtxvApiTextToVideozLTXV Text To Videoapi node/video/LTXVzFProfessional-quality videos with customizable duration and resolution.roptionsrT multilinedefaultr r0r)r-r 1920x1080 2560x1440 3840x2160rr2rFRWhen true, the generated video will include AI-generated audio matching the scene.r-optionaltooltipadvanced node_id display_namecategory descriptioninputsoutputshidden is_api_node price_badge)rSchemaComborlist MODELS_MAPkeysStringBooleanVideoOutputHiddenauth_token_comfy_orgapi_key_comfy_org unique_id PRICE_BADGEclsr r r! define_schema0sDzTextToVideoNode.define_schemarFrrrrrrreturnc s~t|ddd|dkr|dks|dks|dkrtdt|td d t|t|||||d d dd IdH}tt t |S)N' min_length max_lengthr1rr9r\Durations over 10s are only available for the Fast model at 1920x1080 resolution and 25 FPS.z/proxy/ltx/v1/text-to-videoPOST)rrrrrrTdata as_binary max_retries) r ValueErrorr r rrOr NodeOutputr VideoFromFiler)r[rrrrrrresponser r r!execute\s*  zTextToVideoNode.executeNrF) rrr classmethodr\rrrrrjrmr r r r!r%/s* +r%c@sPeZdZeddZe  ddejdededed ed ed e d e j fd dZ dS)ImageToVideoNodecCstjddddtjjdddtjjdttd tjjd d d d tjjdgdddtjjdgdd tjjdddgddtj jddd dd dgtj gtj j tj jtj jgd td S)NLtxvApiImageToVideozLTXV Image To Videor'z[Professional-quality videos with customizable duration and resolution based on start image.imagez%First frame to be used for the video.)r@rr(rTr*r+rr.r0r7rr8rrr<rFr=r>rB)rrLImagerrMrNrOrPrQrRrSrTrUrVrWrXrYrZr r r!r\}sFzImageToVideoNode.define_schemarFrrrrrrrrr]c st|ddd|dkr|dks|dks|dkrtdt|dkr&td t|td d tt||dd d IdHd|t|||||ddddIdH}t t t |S)Nr^r_r`r1rr9rrcz,Currently only one input image is supported.z/proxy/ltx/v1/image-to-videordz image/png) max_images mime_typer)rrrrrrrTre) rrir r r rr rOrrjr rkr) r[rrrrrrrrrlr r r!rms0    zImageToVideoNode.executeNrn) rrrror\rrsrrrrrjrmr r r r!rp|s. , rpc@s(eZdZedeeejfddZdS)LtxvApiExtensionr]cs ttgSN)r%rp)selfr r r! get_node_listszLtxvApiExtension.get_node_listN) rrrrrNtyper ComfyNoderyr r r r!rvsrvr]cstSrw)rvr r r r!comfy_entrypointsr|N)iorpydanticrrtyping_extensionsrcomfy_api.latestrrrr comfy_api_nodes.utilr r r r rrOr PriceBadgePriceBadgeDependsrYr{r%rprvr|r r r r!s"    MR