o i>@s ddlmZddlmZddlmZmZmZddlm Z m Z m Z m Z ddl mZmZmZmZmZdededee d Bfd d Zd d Zdeded BfddZdedd fddZdeedeefddZGdddejZGdddejZGdddejZ GdddeZ!de!fdd Z"d S)!)BytesIO)override)IOComfyExtensionInput)ReveImageCreateRequestReveImageEditRequestReveImageRemixRequestRevePostprocessingOperation) ApiEndpointbytesio_to_image_tensor sync_op_rawtensor_to_base64_stringvalidate_stringupscaleremove_backgroundreturnNcCsBg}|ddkr|td|dd|r|tdd|p dS)Nrenabledupscale_factor)processrr)r)appendr )rropsr8/mnt/c/Users/fbmor/ComfyUI/comfy_api_nodes/nodes_reve.py_build_postprocessings rcCsRtjjdtjdgtjdtjjddddddd ggd d tjjd d ddgS)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.)rr#)r DynamicComborOptionIntBooleanrrrr_postprocessing_inputs#s0 r*headerscCs"|d}|durt|dSdS)Nzx-reve-credits-usedgp= c@)getfloat)r+ credits_usedrrr_reve_price_extractorAs  r/cCs<|d}|rtd||dddkrtddS)Nzx-reve-error-codezReve API error: zx-reve-content-violationtruez=The generated image was flagged for content policy violation.)r, ValueErrorlower)r+ error_coderrr_reve_response_header_validatorHs r5versions aspect_ratioscsfdd|DS)NcsBg|]}tj|tjjdddtjjddddddddgqS) aspect_ratioz!Aspect ratio of the output image.r$test_time_scalingrz:Higher values produce better images but cost more credits.T)rr r!r"r#advanced)rr&r'Comborr().0versionr7rr Qs*z!_model_inputs..r)r6r7rr?r _model_inputsPs rAc @s@eZdZeddZedededededede j f d d Z d S) ReveImageCreateNodecCstjddddtjjddddd tjjd td ggd d ddgttjjdddddddtj gtj j tj j tj j gdtjtjddgdddd S)NrBzReve Image Createapi node/image/Revez2Generate images from text descriptions using Reve.promptTr0z?Text description of the desired image. Maximum 2560 characters. multilinerr#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.rr 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&rAr*r(ImageOutputHiddenauth_token_comfy_orgapi_key_comfy_org unique_id PriceBadgePriceBadgeDependsclsrrr define_schemaksX z!ReveImageCreateNode.define_schemarDrGrrrOrcsjt|dddt|tddddidd ttt||d |d |d t||d dIdH}tt t |S)Nr  min_length max_lengthz/proxy/reve/v1/image/createPOSTAccept image/webppathmethodr+Tr8rGr9)rDr8r>r9postprocessing as_binaryprice_extractorresponse_header_validatordata) rr r r/r5rrr NodeOutputr r)rorDrGrrrOresponserrrexecutes*  zReveImageCreateNode.executeN) __name__ __module__ __qualname__ classmethodrpstrdictboolintrrrrrrrrBis" 9rBc@sFeZdZeddZedejdededede de d e j fd d Z d S) ReveImageEditNodecCstjddddtjjdddtjjdd d d d tjjd tddggddddgttjjddddd ddtj gtj j tj j tj j gd tjtjgddddd S)NrzReve Image EditrCz:Edit images using natural language instructions with Reve.imagezThe image to edit.)r#edit_instructionTr0zCText description of how to edit the image. Maximum 2560 characters.rErGzreve-edit@20250915zreve-edit-fast@20251030autorIrJrHrKrLrMrNr?z!Model version to use for editing.r$rOrrPrQrRrGrrTrU ( $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} ) rWrZ)rrdrfrrer&rAr*r(rgrhrirjrkrlrmrnrrrrpsZ zReveImageEditNode.define_schemarrrGrrrOrc st|ddd|d}|d}t|tdddd id d ttt|t||d kr(|nd|d |r4|dkr4|ndt||ddIdH} t t t | S)Nrrqrrr9r8z/proxy/reve/v1/image/editrurvrwrxTrrG)rreference_imager8r>r9r{r|) rr r r/r5rrrrrr r) rorrrGrrrOttsarrrrrrs0  zReveImageEditNode.executeN)rrrrrprrfrrrrrrrrrrrrs& >rc@sHeZdZeddZedejjdede de de de d ej fd d Z d S) ReveImageRemixNodecCstjddddtjjdtjjtjdddd d d tjjd d dddtjjdtddggddddgt tj jddddd ddtj gtj j tj jtj jgd tjtjgdddd d! S)"NrzReve Image RemixrCzKCombine reference images with text prompts to create new images using Reve.reference_imagesrimage_r)prefixr r!)templaterDTr0zText description of the desired image. May include XML img tags to reference specific images by index, e.g. 0, 1, etc.rErGzreve-remix@20250915zreve-remix-fast@20251030rr?z"Model version to use for remixing.r$rOrrPrQrRrrUrrWrZ)rrdAutogrowrTemplatePrefixrfrer&rAr*r(rgrhrirjrkrlrmrnrrrrp*sj   %z ReveImageRemixNode.define_schemarrDrGrrrOrc st|ddd|stdg}|D] }|t||qt|dkr(td|d} |d} t|td d d d id dttt ||| dkrF| nd|d| rR| dkrR| ndt ||ddIdH} t t t| S)Nrrqrrz)At least one reference image is required.rz'Maximum 6 reference images are allowed.r9r8z/proxy/reve/v1/image/remixrurvrwrxTrrG)rDrr8r>r9r{r|)rr2rrlenr r r/r5r rrrr r) rorrDrGrrrOref_base64_listkeyrrrrrrrss>   zReveImageRemixNode.executeN)rrrrrprrTyperrrrrrrrrrr(s& Hrc@s(eZdZedeeejfddZdS) ReveExtensionrcs tttgSN)rBrr)selfrrr get_node_lists zReveExtension.get_node_listN) rrrrlisttyper ComfyNoderrrrrrsrcstSr)rrrrrcomfy_entrypointsr)#iortyping_extensionsrcomfy_api.latestrrrcomfy_api_nodes.apis.reverrr r comfy_api_nodes.utilr r r rrrrrrr*r-r/r5rrArrBrrrrrrrrs  [du