o i*_@sTddlmZddlZddlmZddlmZmZddlm Z m Z m Z m Z m Z mZmZmZmZmZmZmZmZmZmZmZmZmZddlmZmZmZmZm Z m!Z!m"Z"dZ#dZ$Gd d d ej%Z&Gd d d ej%Z'Gd ddej%Z(Gdddej%Z)Gdddej%Z*Gdddej%Z+ej,ej-gddddZ.GdddeZ/de/fddZ0dS))OptionalN)override)IOComfyExtension)LumaAspectRatioLumaCharacterRefLumaConceptChainLumaGenerationLumaGenerationRequestLumaImageGenerationRequestLumaImageIdentityLumaImageModelLumaImageReferenceLumaIO LumaKeyframesLumaModifyImageRef LumaReferenceLumaReferenceChainLumaVideoModelLumaVideoModelOutputDurationLumaVideoOutputResolutionget_luma_concepts) ApiEndpointdownload_url_to_image_tensordownload_url_to_video_outputpoll_opsync_opupload_images_to_comfyapivalidate_stringidc @sDeZdZedejfddZed dejde de dej fdd Z dS) LumaReferenceNodereturncCs`tjddddtjjdddtjjdd d d d d d ttjjdddgttjjddgdS)Nr!zLuma Referenceapi node/image/Lumaz@Holds an image and weight for use with Luma Generate Image node.imagezImage to use as reference.)tooltipweight?{Gz?zWeight of image reference.defaultminmaxstepr%luma_refT)optional display_namenode_idr2category descriptioninputsoutputs) rSchemaImageInputFloatCustomrLUMA_REFOutputclsrB8/mnt/c/Users/fbmor/ComfyUI/comfy_api_nodes/nodes_luma.py define_schema*s0 zLumaReferenceNode.define_schemaNr$r&r/cCs:|dur |}nt}|t|t|ddt|S)Nr$r&)cloneraddrroundr NodeOutput)rAr$r&r/rBrBrCexecuteFs   zLumaReferenceNode.executeN) __name__ __module__ __qualname__ classmethodrr9rDtorchTensorfloatrrJrKrBrBrBrCr!)s &r!c@sLeZdZedejfddZe d dedededed edej f d d Z dS) LumaConceptsNoder"cCstjddddtjjdtdddtjjd tdddtjjd tdddtjjd tdddttjjd d ddgttjjd dgdS)NrTz Luma Conceptsapi node/video/LumazNCamera Concepts for use with Luma Text to Video and Luma Image to Video nodes.concept1T) include_noneoptionsconcept2concept3concept4 luma_conceptsz8Optional Camera Concepts to add to the ones chosen here.r%r0r1r3) rr9Combor;rr=r LUMA_CONCEPTSr?r@rBrBrCrDQs: zLumaConceptsNode.define_schemaNrVrZr[r\r]cCs.t||||gd}|dur||}t|S)N)str_list)rclone_and_mergerrJ)rArVrZr[r\r]chainrBrBrCrKrs   zLumaConceptsNode.executerL) rMrNrOrPrr9rDstrrrJrKrBrBrBrCrTPs$ rTc@seZdZedejfddZe   ddedededed e e d e e j d e e j dej fd d Zede defddZed e j defddZdS)LumaImageGenerationNoder"cCstjddddtjjddddd tjjd td tjjd ttjd tjjdddddddtj jdddddddt t j jddddtj jddddtj jddddgtj gtjjtjjtjjgdtjtjd gd d!d"d# S)$N LumaImageNodezLuma Text to Imager#z@Generates images synchronously based on prompt and aspect ratio.promptTPrompt for the image generation multiliner+r%modelrX aspect_ratiorYr+seedr`Seed to determine if node should re-run; actual results are nondeterministic regardless of seed.r+r,r-control_after_generater%style_image_weightr'r(r)z:Weight of style image. Ignored if no style_image provided.r*image_luma_refzkLuma Reference node connection to influence generation with input images; up to 4 images can be considered.r^ style_imagez1Style reference image; only 1 image will be used.character_imagezYCharacter reference images; can be a batch of multiple, up to 4 images can be considered.widgetsV ( $m := widgets.model; $contains($m,"photon-flash-1") ? {"type":"usd","usd":0.0027} : $contains($m,"photon-1") ? {"type":"usd","usd":0.0104} : {"type":"usd","usd":0.0246} )  depends_onexpr r4r2r5r6r7r8hidden is_api_node price_badge)rr9Stringr;r_r r ratio_16_9Intr<r=rr>r:r?Hiddenauth_token_comfy_orgapi_key_comfy_org unique_id PriceBadgePriceBadgeDependsr@rBrBrCrDs~  0 z%LumaImageGenerationNode.define_schemaNrgrlrmrtrurvrwc  st|dddd} |dur|j|ddIdH} d} |dur(|j||dIdH} d} |dur@t||ddIdH} tt| dd } t|td d d tt |||| | | d dIdH} t |td| j dtdddIdH}t t|jjIdHS)NTstrip_whitespace min_lengthmax_refs)r& max_images)images) identity0/proxy/luma/generations/imagePOSTpathmethod)rgrlrm image_ref style_ref character_refresponse_modeldata/proxy/luma/generations/rcS|jSrLstatexrBrBrCz1LumaImageGenerationNode.execute..rstatus_extractor)r_convert_luma_refs_convert_style_imagerrr rrr r ridrrJrassetsr$)rArgrlrmrortrurvrw api_image_ref api_style_refr download_urls response_api response_pollrBrBrCrKs@     zLumaImageGenerationNode.executer/rcs\g}d}|jD]}t||jddIdH}||d|d7}||kr&nq|j||dS)Nrr)rr)refsrr$appendcreate_api_model)rAr/r luma_urls ref_countrefrrBrBrCrs z*LumaImageGenerationNode._convert_luma_refsr&cs(tt||dd}|j|ddIdHS)NrF) first_refrr)rrr)rArvr&rcrBrBrCr sz,LumaImageGenerationNode._convert_style_imageNNN)rMrNrOrPrr9rDrdrSrrrQrRrJrKintrrrBrBrBrCres8L  / rec @sFeZdZedejfddZedededej de dej f dd Z d S) LumaImageModifyNoder"cCstjddddtjdtjjdddd d tjjd d d ddddtjjdtdtjjdddddddgtj gtj j tj j tj j gdtjtjdgdddd S)NrzLuma Image to Imager#z?Modifies images synchronously based on prompt and aspect ratio.r$rgTrhrirj image_weightg?r(\(\?r)zLWeight of the image; the closer to 1.0, the less the image will be modified.r*rlrXrorrprqrrrxrzr{r~)rr9r:r;rr<r_r rr?rrrrrrr@rBrBrCrDs\  z!LumaImageModifyNode.define_schemargrlr$rc st||ddIdH}|d}t|tdddtt||t|tttd|dd d d d d IdH}t |td|j dtdddIdH} t t | jjIdHS)Nrrrrrrr'rr(rE)urlr&)rgrlmodify_image_refrrrcSrrLrrrBrBrCrirz-LumaImageModifyNode.execute..r)rrrr r rrIr-r,rrrrJrrr$) rArgrlr$rror image_urlrrrBrBrCrKNs,     zLumaImageModifyNode.executeN) rMrNrOrPrr9rDrdrQrRrSrJrKrBrBrBrCrs;rc@sXeZdZedejfddZe ddedededed ed ed e e dej fd d Z dS)LumaTextToVideoGenerationNoder"cCstjddddtjjddddd tjjd td tjjd ttjd tjjdttj d tjjdt d tj jdddtj jdddddddt tjjddddgtjgtjjtjjtjjgdtd S)N LumaVideoNodezLuma Text to VideorUz?Generates videos synchronously based on prompt and output_size.rgTrhPrompt for the video generationrjrlrXrmrn resolutiondurationloopFr+rorrprqrrr]MOptional Camera Concepts to dictate camera motion via the Luma Concepts node.r^r~)rr9rr;r_rrrrres_540prBooleanrr=rr`Videor?rrrrPRICE_BADGE_VIDEOr@rBrBrCrDosn  +z+LumaTextToVideoGenerationNode.define_schemaNrgrlrmrrrr]c st|ddd|tjkr|nd}|tjkr|nd}t|tdddtt|||||||r/|ndddIdH} t|td | j d td d t d IdH} t t | jjIdHS)NFrr/proxy/luma/generationsrr)rgrlrrmrrconceptsrrrcSrrLrrrBrBrCrrz7LumaTextToVideoGenerationNode.execute..)rrestimated_duration)rrray_1_6rrr r rrrLUMA_T2V_AVERAGE_DURATIONrrJrrvideo) rArgrlrmrrrror]rrrBrBrCrKs4    z%LumaTextToVideoGenerationNode.executerL) rMrNrOrPrr9rDrdboolrrrJrKrBrBrBrCrns,;   rc@seZdZedejfddZe   ddedededed ed e j d e j d e dej fd dZ e  dd e j d e j fddZdS)LumaImageToVideoGenerationNoder"cCstjddddtjjddddd tjjd td tjjd ttjd tjjdddtDd tj jdddtj jdddddddtj jddddtj jddddt t jjddddg tjgtjjtjjtjjgdtd S)!NLumaImageToVideoNodezLuma Image to VideorUzNGenerates videos synchronously based on prompt, input images, and output_size.rgTrhrrjrlrXrrnrcSsg|]}|jqSrB)value).0durrBrBrC sz@LumaImageToVideoGenerationNode.define_schema..rFrrorrprqrr first_imagezFirst frame of generated video.r^ last_imagezLast frame of generated video.r]rr~)rr9rr;r_rrrrrrr:r=rr`rr?rrrrrr@rBrBrCrDsx    5z,LumaImageToVideoGenerationNode.define_schemaNrgrlrrrrrr]c s|dur |dur td|||IdH} |tjkr|nd}|tjkr&|nd}t|tdddtt||tj |||| | r?| ndddIdH} t |td| j dtd d t d IdH} tt| jjIdHS) Nz=At least one of first_image and last_image requires an input.rrr)rgrlrmrrr keyframesrrrrcSrrLrrrBrBrCr@rz8LumaImageToVideoGenerationNode.execute..) poll_endpointrrr) Exception_convert_to_keyframesrrrrr r rrrrrLUMA_I2V_AVERAGE_DURATIONrrJrrr) rArgrlrrrrorrr]rrrrBrBrCrKs:    z&LumaImageToVideoGenerationNode.executecs|dur |dur dSd}d}|dur%t||ddIdH}td|dd}|dur;t||ddIdH}td|dd}t||dS)Nrrr$r)typer)frame0frame1)rrr)rArrrrrrBrBrCrEs z4LumaImageToVideoGenerationNode._convert_to_keyframesr)NN)rMrNrOrPrr9rDrdrrQrRrrJrKrrBrBrBrCrsDF    )r)rlrrrxa ( $p := { "ray-flash-2": { "5s": {"4k":3.13,"1080p":0.79,"720p":0.34,"540p":0.2}, "9s": {"4k":5.65,"1080p":1.42,"720p":0.61,"540p":0.36} }, "ray-2": { "5s": {"4k":9.11,"1080p":2.27,"720p":1.02,"540p":0.57}, "9s": {"4k":16.4,"1080p":4.1,"720p":1.83,"540p":1.03} } }; $m := widgets.model; $d := widgets.duration; $r := widgets.resolution; $modelKey := $contains($m,"ray-flash-2") ? "ray-flash-2" : $contains($m,"ray-2") ? "ray-2" : $contains($m,"ray-1-6") ? "ray-1-6" : "other"; $durKey := $contains($d,"5s") ? "5s" : $contains($d,"9s") ? "9s" : ""; $resKey := $contains($r,"4k") ? "4k" : $contains($r,"1080p") ? "1080p" : $contains($r,"720p") ? "720p" : $contains($r,"540p") ? "540p" : ""; $modelPrices := $lookup($p, $modelKey); $durPrices := $lookup($modelPrices, $durKey); $v := $lookup($durPrices, $resKey); $price := ($modelKey = "ray-1-6") ? 0.5 : ($modelKey = "other") ? 0.79 : ($exists($v) ? $v : 0.79); {"type":"usd","usd": $price} ) r{c@s(eZdZedeeejfddZdS) LumaExtensionr"csttttttgSrL)rerrrr!rT)selfrBrBrC get_node_listszLumaExtension.get_node_listN) rMrNrOrlistrr ComfyNoderrBrBrBrCrsrr"cstSrL)rrBrBrBrCcomfy_entrypointsr)1typingrrQtyping_extensionsrcomfy_api.latestrrcomfy_api_nodes.apis.lumarrrr r r r r rrrrrrrrrrcomfy_api_nodes.utilrrrrrrrrrrr!rTrerrrrrrrrrBrBrBrCs,  P$ '1]e/