+j1=zddlmZddlmZddlmZmZmZddlm Z m Z m Z m Z ddl mZmZmZmZmZdededee d zfd Zd Zd eded zfd Zd edd fdZdeedeefdZGddejZGddejZGddejZ GddeZ!de!fdZ"d S))BytesIO)override)IOComfyExtensionInput)ReveImageCreateRequestReveImageEditRequestReveImageRemixRequestRevePostprocessingOperation) ApiEndpointbytesio_to_image_tensor sync_op_rawtensor_to_base64_stringvalidate_stringupscaleremove_backgroundreturnNcg}|ddkr*|td|d|r#|td|pdS)Nrenabledupscale_factor)processrr)r)appendr )rropss :/home/wildlama/comfy/ComfyUI/comfy_api_nodes/nodes_reve.py_build_postprocessingrs CyY&& '!&'78      M .7JKKKLLL ;$cNtjdtjdgtjdtjdddddd ggd tjd 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  &&z266&& ,$% ! !!"$E %    L%   (  ^   + rheaderscX|d}|t|dz SdS)Nzx-reve-credits-usedgp= c@)getfloat)r/ credits_useds r_reve_price_extractorr4As2;;455L\""V++ 4rc|d}|rtd||dddkrtddS)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;Hsq011J:8J88999{{-r2288::fDDXYYYEDrversions aspect_ratiosc fd|DS)Ncg|]h}tj|tjddtjdddddddgiS) 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+).0versionr=s r z!_model_inputs..Qs   ( '  ")?  'X!       rr-)r<r=s `r _model_inputsrHPs0    ( )   rc ^eZdZedZedededededede j f dZ d S) ReveImageCreateNodec8tjddddtjdddd tjd t d ggd  dgt tjddddddtj gtj j tj j tj j gdtjtjddgd S)NrJzReve 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)rHr.r+ImageOutputHiddenauth_token_comfy_orgapi_key_comfy_org unique_id PriceBadgePriceBadgeDependsclss r define_schemaz!ReveImageCreateNode.define_schemaksHy),)L "]   %%)/0&Y&Y&YC &())  "+/H !4X__&&' . + #  /&(@A  M6 6 6 6 rrMrPrrrYrcXKt|ddt|tddddid ttt ||d |d |d t || d{V}tjtt|S)Nr!  min_length max_lengthz/proxy/reve/v1/image/createPOSTAccept image/webppathmethodr/Tr@rPrA)rMr@rFrApostprocessing as_binaryprice_extractorresponse_header_validatordata) rrr r4r;rrr NodeOutputr r)ryrMrPrrrYresponses rexecutezReveImageCreateNode.executes 1>>>>$  2!<0    1&E'">2g"'(;"<4W>OPP          $}4WX5F5FGGHHHrN) __name__ __module__ __qualname__ classmethodrzstrdictboolintrrrr-rrrJrJis7 7 [7 rIII I  I  I III[IIIrrJcleZdZedZedejdededede de de j fd Z d S) ReveImageEditNodecztjddddtjddtjdd d d tjd t ddggddgttjddddd dtj gtj j tj j tj j gd tjtjgdd S)NrzReve Image EditrLz: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.rNrPzreve-edit@20250915zreve-edit-fast@20251030autorRrSrQrTrUrVrWrXz!Model version to use for editing.r'rYrrZr[r\rPrr^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} ) rard)rrnrprror)rHr.r+rqrrrsrtrurvrwrxs rrzzReveImageEditNode.define_schemasay'*)Tw0DEE &"a   %%)-/HI&a&a&a@ & ())!" "+/H #6X__&&' . + #  /JJJ O; ; ; ; rrrrPrrrYrcKt|dd|d}|d}t|tdddd i d ttt |t ||d kr|nd|d |r|dkr|ndt||d{V} tj tt| S)Nr!r|r}rAr@z/proxy/reve/v1/image/editrrrrTrrP)rreference_imager@rFrArr) rrr r4r;r rrrrr r) ryrrrPrrrYttsarrs rrzReveImageEditNode.executes (Q4HHHH'( > "$  0!<0    1&E%!1 7 > >#%<OPP          &}4WX5F5FGGHHHrN)rrrrrzrrprrrrrrrr-rrrrs< < [< |I{II I  I  II III[IIIrrcveZdZedZedejjdede de de de dej fd Z d S) ReveImageRemixNodectjddddtjdtjtjdddd  tjd d ddtjdtddggddgttj ddddd dtj gtj j tj jtj jgd tjtjgdd ! S)"NrzReve Image RemixrLzKCombine reference images with text prompts to create new images using Reve.reference_imagesrimage_r!)prefixr#r$)templaterMTr6zText description of the desired image. May include XML img tags to reference specific images by index, e.g. 0, 1, etc.rNrPzreve-remix@20250915zreve-remix-fast@20251030rrXz"Model version to use for remixing.r'rYrrZr[r\rr_rrard)rrnAutogrowrTemplatePrefixrpror)rHr.r+rqrrrsrtrurvrwrxs rrzz ReveImageRemixNode.define_schema*sy(+)e !!&[77w//' 8" "<  %%).0JK&a&a&aA &%$4())5$6 "+/H 7$JX__&&' . + #  /JJJ cE E E E rrrMrPrrrYrcJKt|dd|stdg}|D]*}|t||+t |dkrtd|d} |d} t |t d d d d i dttt||| dkr| nd|d| r| dkr| ndt||d{V} tj tt| S)Nr!r|r}z)At least one reference image is required.rz'Maximum 6 reference images are allowed.rAr@z/proxy/reve/v1/image/remixrrrrTrrP)rMrr@rFrArr)rr8rrlenrr r4r;r rrrr r) ryrrMrPrrrYref_base64_listkeyrrrs rrzReveImageRemixNode.executess{ 1>>>> JHII I# S SC  " "#:;KC;P#Q#Q R R R R   ! # #FGG G'( > "$  1!<0    1&E&!0#%<OPP          &}4WX5F5FGGHHHrN)rrrrrzrrTyperrrrrrr-rrrr(sF F [F P&I+*&I&I &I  &I  &I&I &I&I&I[&I&I&IrrcLeZdZedeeejfdZdS) ReveExtensionrc.KtttgSN)rJrr)selfs r get_node_listzReveExtension.get_node_lists    rN) rrrrlisttyper ComfyNoderr-rrrrsE  T$r|*<%=   X   rrc"KtSr)rr-rrcomfy_entrypointrs ??r)#iortyping_extensionsrcomfy_api.latestrrrcomfy_api_nodes.apis.reverr r r comfy_api_nodes.utilr r rrrrrrrr.r2r4r;rrHrrJrrrrr-rrrs&&&&&&6666666666  4 D TJeEfimEm    <4EDLZTZdZZZZDId3i2XIXIXIXIXI",XIXIXIvaIaIaIaIaI aIaIaIHrIrIrIrIrIrIrIrIj     N    r