^(j1=lddlmZddlmZddlmZmZmZddlm Z m Z m Z m Z ddl mZmZmZmZmZdededee d zfd Zd Zd eded zfd Zd edd fdZdeedeefdZGddej:ZGddej:ZGddej:Z GddeZ!de!fdZ"y ))BytesIO)override)IOComfyExtensionInput)ReveImageCreateRequestReveImageEditRequestReveImageRemixRequestRevePostprocessingOperation) ApiEndpointbytesio_to_image_tensor sync_op_rawtensor_to_base64_stringvalidate_stringupscaleremove_backgroundreturnNcg}|ddk(r|jtd|d|r|jtd|xsdS)Nrenabledupscale_factor)processrr)r)appendr )rropss D/Users/danicosta/Desktop/Flux2/ComfyUI/comfy_api_nodes/nodes_reve.py_build_postprocessingrsV CyY& '!&'78   .7JKL ;$cNtjjdtjjdgtjjdtjjdddddd ggd tj jd d dgS)NrdisabledrrzUpscale factor (2x, 3x, or 4x).)defaultminmaxsteptooltipz5Upscale the generated image. May add additional cost.optionsr&rFzHRemove the background from the generated image. May add additional cost.)r"r&)r DynamicComborOptionIntBooleanrr_postprocessing_inputsr.#s  &&z26&& ,$% ! !!"$E %   L%  (  ^  + rheaderscF|jd}|t|dz Sy)Nzx-reve-credits-usedgp= c@)getfloat)r/ credits_useds r_reve_price_extractorr4As*;;45L\"V++ rc|jd}|rtd||jddjdk(r tdy)Nzx-reve-error-codezReve API error: zx-reve-content-violationtruez=The generated image was flagged for content policy violation.)r1 ValueErrorlower)r/ error_codes r_reve_response_header_validatorr;HsV01J+J<899{{-r288:fDXYYErversions aspect_ratiosc|Dcgc]h}tjj|tjj d|dtj j ddddddd gjc}Scc}w) N aspect_ratioz!Aspect ratio of the output image.r'test_time_scalingr!z:Higher values produce better images but cost more credits.T)r"r#r$r%r&advanced)rr)r*Comborr+)r<r=versions r _model_inputsrEPs* ) ( G'  ")?  'X!   & )  sA-A5c ZeZdZedZedededededede jf dZ y ) ReveImageCreateNodecdtjddddtjjdddd tjjd t d ggd  dgt tjjddddddtjjgtjjtjjtjjgdtjtjddgd S)NrGzReve Image Createpartner/image/Revez2Generate images from text descriptions using Reve.promptTr6z?Text description of the desired image. Maximum 2560 characters. multiliner"r&modelzreve-create@20250915)3:216:99:162:34:33:41:1r=z$Model version to use for generation.r'seedr_Seed controls whether the node should re-run; results are non-deterministic regardless of seed.r"r#r$control_after_generater&rupscale.upscale_factorwidgetsa6 ( $factor := $lookup(widgets, "upscale.upscale_factor"); $fmt := {"approximate": true, "note": "(base)"}; widgets.upscale = "enabled" ? ( $factor = 4 ? {"type": "usd", "usd": 0.0762, "format": $fmt} : $factor = 3 ? {"type": "usd", "usd": 0.0591, "format": $fmt} : {"type": "usd", "usd": 0.0457, "format": $fmt} ) : {"type": "usd", "usd": 0.03432, "format": $fmt} )  depends_onexpr node_id display_namecategory descriptioninputsoutputshidden is_api_node price_badge)rSchemaStringrr)rEr.r+ImageOutputHiddenauth_token_comfy_orgapi_key_comfy_org unique_id PriceBadgePriceBadgeDependsclss r define_schemaz!ReveImageCreateNode.define_schemaks#yy),)L "]   %%)/0&YC &()  "+/H !4XX__&' .. ++ ##  //&(@A  M6 6 rrJrMrrrVrcKt|ddt|tddddid ttt ||d |d |d t || d{}tjtt|S7+w)Nr!  min_length max_lengthz/proxy/reve/v1/image/createPOSTAccept image/webppathmethodr/Tr?rMr@)rJr?rDr@postprocessing as_binaryprice_extractorresponse_header_validatordata) rrr r4r;rrr NodeOutputr r)rvrJrMrrrVresponses rexecutezReveImageCreateNode.executes 1>$  2!<0  1&E'">2g"'(;"<4W>OP   $}}4WX5FGHH% sAB B,B N) __name__ __module__ __qualname__ classmethodrwstrdictboolintrrrr-rrrGrGiso7 7 rIII I  I  I IIrrGcreZdZedZedej dededede de de jfd Z y ) ReveImageEditNodectjddddtjjddtjjdd d d tj jd t ddggddgttjjddddd dtjjgtjjtjjtjjgd tjtjgdd S)NrzReve Image EditrIz:Edit images using natural language instructions with Reve.imagezThe image to edit.)r&edit_instructionTr6zCText description of how to edit the image. Maximum 2560 characters.rKrMzreve-edit@20250915zreve-edit-fast@20251030autorOrPrNrQrRrSrTrUz!Model version to use for editing.r'rVrrWrXrYrMrr[r\ ( $fmt := {"approximate": true, "note": "(base)"}; $isFast := $contains(widgets.model, "fast"); $enabled := widgets.upscale = "enabled"; $factor := $lookup(widgets, "upscale.upscale_factor"); $isFast ? {"type": "usd", "usd": 0.01001, "format": $fmt} : $enabled ? ( $factor = 4 ? {"type": "usd", "usd": 0.0991, "format": $fmt} : $factor = 3 ? {"type": "usd", "usd": 0.0819, "format": $fmt} : {"type": "usd", "usd": 0.0686, "format": $fmt} ) : {"type": "usd", "usd": 0.0572, "format": $fmt} ) r^ra)rrkrmrrlr)rEr.r+rnrorprqrrrsrtrus rrwzReveImageEditNode.define_schemas5yy'*)Tw0DE &"a   %%)-/HI&a@ & ()!" "+/H #6XX__&' .. ++ ##  //J O; ; rrrrMrrrVrcTKt|dd|d}|d}t|tdddd i d ttt |t ||d k7r|nd|d |r|dkDr|ndt||d{} tjtt| S7+w)Nr!ryrzr@r?z/proxy/reve/v1/image/editr}r~rrTrrM)rreference_imager?rDr@rr) rrr r4r;r rrrrr r) rvrrrMrrrVttsarrs rrzReveImageEditNode.executes (Q4H'( > "$  0!<0  1&E%!1 7 >#%OP   &}}4WX5FGHH' sA8B(:B&;,B(N)rrrrrwrrmrrrrrrrr-rrrrs~< < |I{{II I  I  II IIrrceZdZedZedej jdede de de de dejfd Z y ) ReveImageRemixNodec"tjddddtjjdtjj tj jdddd  tj jd d ddtjjdtddggddgttjjddddd dtj jgtjjtjjtjjgd tj tj"gdd ! S)"NrzReve Image RemixrIzKCombine reference images with text prompts to create new images using Reve.reference_imagesrimage_r!)prefixr#r$)templaterJTr6zText description of the desired image. May include XML img tags to reference specific images by index, e.g. 0, 1, etc.rKrMzreve-remix@20250915zreve-remix-fast@20251030rrUz"Model version to use for remixing.r'rVrrWrXrYrr\rr^ra)rrkAutogrowrTemplatePrefixrmrlr)rEr.r+rnrorprqrrrsrtrus rrwz ReveImageRemixNode.define_schema*sqyy(+)e !!&[[77w/' 8" "<  %%).0JK&aA &%$4()5$6 "+/H 7$JXX__&' .. ++ ##  //J cE E rrrJrMrrrVrcKt|dd|s tdg}|D]}|jt||!t |dkDr td|d} |d} t |t d d d d i dttt||| dk7r| nd|d| r| dkDr| ndt||d{} tjtt| S7+w)Nr!ryrzz)At least one reference image is required.rz'Maximum 6 reference images are allowed.r@r?z/proxy/reve/v1/image/remixr}r~rrTrrM)rJrr?rDr@rr)rr8rrlenrr r4r;r rrrr r) rvrrJrMrrrVref_base64_listkeyrrrs rrzReveImageRemixNode.executess 1>HI I#C  " "#:;KC;P#Q R$  ! #FG G'( > "$  1!<0  1&E&!0#%OP   &}}4WX5FGHH' sB;C+=C)>,C+N)rrrrrwrrTyperrrrrrr-rrrr(sF F P&I++**&I&I &I  &I  &I&I &I&IrrcBeZdZedeeejfdZy) ReveExtensionrc,KtttgSwN)rGrr)selfs r get_node_listzReveExtension.get_node_lists    sN) rrrrlisttyper ComfyNoderr-rrrrs)  T$r||*<%=  rrcKtSwr)rr-rrcomfy_entrypointrs ?s )#iortyping_extensionsrcomfy_api.latestrrrcomfy_api_nodes.apis.reverr r r comfy_api_nodes.utilr r rrrrrrrr.r2r4r;rrErrGrrrrr-rrrs&66  4 D TJeEfimEm <4EDLZTZdZDId3i2XI",,XIvaI aIHrIrIj N  r