+jddlZddlmZddlmZmZmZddlmZm Z m Z m Z m Z m Z mZmZmZmZmZmZddlmZmZmZmZmZmZmZmZmZmZdZde de fd Z!d e"d e"d e"de fd Z#Gddej$Z%Gddej$Z&Gddej$Z'Gddej$Z(Gddej$Z)GddeZ*de*fdZ+dS)N)override)IOComfyExtensionInput) #ImageRelightAdvancedSettingsRequestImageRelightRequest ImageSkinEnhancerCreativeRequest ImageSkinEnhancerFaithfulRequest ImageSkinEnhancerFlexibleRequestImageStyleTransferRequestImageUpscalerCreativeRequestImageUpscalerPrecisionV2RequestInputAdvancedSettingsInputPortraitModeInputSkinEnhancerMode TaskResponse) ApiEndpointdownload_url_to_image_tensordownscale_image_tensorget_image_dimensionsget_number_of_imagespoll_opsync_opupload_images_to_comfyapivalidate_image_aspect_ratiovalidate_image_dimensionsg ףp= ? megapixelsreturnc6|dkrdS|dkrdS|dkrdSdS)zLPrice in EUR for a single Magnific upscaling step based on input megapixels.g?gM?g@gM?g@g~jt?g~jt?)rs >/home/wildlama/comfy/ComfyUI/comfy_api_nodes/nodes_magnific.py_tier_price_eurr"$s7SuSuSu 5widthheightscalecttj|}d}||z}t|D]}|t |dz z }|dz}t |t zdS)zZCalculate total Magnific upscale price in USD for given input dimensions and scale factor.gi@B)intmathlog2ranger"round _EUR_TO_USD)r$r%r& num_steps total_eurpixels_s r!%_calculate_magnific_upscale_price_usdr4/suDIe$$%%II V^F 9  _Vi%7888 !  [(! , ,,r#c|eZdZedZedejdedededededed ed ed e d e j fd Z dS)!MagnificImageUpscalerCreativeNodecNtjddddtjdtjddd tjd gd  tjd gd tjddddtjjtjdddddtjjtjdddddtjjtjdddddtjjtjdgddtj ddd d!g tj gtj j tj j tj jgdtjtjd dg"d#$% S)&Nr6z!Magnific Image Upscale (Creative)partner/image/MagnificufPrompt‑guided enhancement, stylization, and 2x/4x/8x/16x upscaling. Maximum output: 25.3 megapixels.imagepromptT multilinedefault scale_factor2x4x8x16xoptions optimized_for) standardsoft_portraitshard_portraitsart_n_illustrationvideogame_assetsnature_n_landscapesfilms_n_photography 3d_rendersscience_fiction_n_horror creativityi r)minmaxr> display_modehdrz#The level of definition and detail.rSrTr>tooltiprU resemblancez/The level of resemblance to the original image. fractalityz:The strength of the prompt and intricacy per square pixel.engine) automaticmagnific_illusiomagnific_sharpymagnific_sparkle)rFadvancedauto_downscaleFzOAutomatically downscale input image if output would exceed maximum pixel limit.r>rXr`widgetsa ( $ad := widgets.auto_downscale; $mins := $ad ? {"2x": 0.172, "4x": 0.343, "8x": 0.515, "16x": 0.515} : {"2x": 0.172, "4x": 0.343, "8x": 0.515, "16x": 0.844}; $maxs := {"2x": 0.515, "4x": 0.844, "8x": 1.015, "16x": 1.187}; { "type": "range_usd", "min_usd": $lookup($mins, widgets.scale_factor), "max_usd": $lookup($maxs, widgets.scale_factor), "format": { "approximate": true } } )  depends_onexpr node_id display_namecategory descriptioninputsoutputshidden is_api_node price_badge)rSchemaImagerStringComboInt NumberDisplaysliderBooleanOutputHiddenauth_token_comfy_orgapi_key_comfy_org unique_id PriceBadgePriceBadgeDependsclss r! define_schemaz/MagnificImageUpscalerCreativeNode.define_schema;sy7<-/w'' D"EE~7P7P7PQQ#      \sATVTdTk ll A!#!1!8  !M!#!1!8   X!#!1!8 ddd!    $!m! !a6p!! . + #  /IY8Z[[[MX X X X r#r9r:r?rGrQrVrYrZr[rarc vKt|dkrtdt|dddt|ddd } t |\} } t |d }| | z|z|z}|| kr| rn| | z}d }| d z}d D]F}||kr ||z|z}|| kr|}d}n*tj|| z }|dkr |}| ||zz}nG|t||}|d }n$td| |zd | |zd|dd| dd t |\}}t |d }t|||t|tddttt||ddd{Vd||||||| |r|nd d{V}t!|td|jtdfdd !d{V}t%jt)|jdd{VS)"N$Exactly one input image is required.rrrFstrict min_height min_widthi xr)r(r(r)@ total_pixelsz Output size (z = ,z) pixels) exceeds maximum allowed size of 9 pixels. Use a smaller input image or lower scale factor.z#/proxy/freepik/v1/ai/image-upscalerPOSTpathmethod max_imagesrr) r9r?rGrQrVrYrZr[r:response_modeldataz$/proxy/freepik/v1/ai/image-upscaler/rc|jSNstatusrs r!z;MagnificImageUpscalerCreativeNode.execute..qxr#cSrr r3 price_usds r!rz;MagnificImageUpscalerCreativeNode.execute..ir#$@rstatus_extractorprice_extractor poll_interval)r ValueErrorrrrr*rstripr+sqrtrr4rrrr rrtask_idr NodeOutputr generated)rr9r:r?rGrQrVrYrZr[ramax_output_pixelsr%r$requested_scale output_pixels input_pixelsr&max_input_pixels candidatescale_output_pixelsdownscale_ratio final_height final_width actual_scale initial_resfinal_responsers @r!executez)MagnificImageUpscalerCreativeNode.executesm  & &! + +CDD D#E66%HHHH!%C3GGGG&,U33 l11#66778?J , , ,  %v~ #4#9 !.  I ?22 *6*BY*N'*.??? )+/(&*i0CFW0W&X&XO&#-- )+<YAV+W(. $/25GWXXXE"'{{{  HEO$;HHf>VHH[huHH7HUHHH %9$?$?! k<..s3344 9+|Uabb #  B6 R R R'-6sEa^bcccccccccefg)+%'%!'1vvT             '  YKDWYY Z Z Z'//////          }#?@XYZ@[#\#\\\\\\\]]]r#N __name__ __module__ __qualname__ classmethodrrrsstrr*boolrrrr r#r!r6r6:sY Y [Y vR^{R^R^ R^  R^  R^R^R^R^R^R^ R^R^R^[R^R^R^r#r6cpeZdZedZedejdededededede d e j fd Z d S) "MagnificImageUpscalerPreciseV2Nodecbtjddddtjdtjdgdtjd gd d tjd ddddtjjtjdddddtjjtjdddddtjjtjddddgtj gtj j tj j tj j gdtjtjdgd S)!Nrz#Magnific Image Upscale (Precise V2)r8uqHigh-fidelity upscaling with fine control over sharpness, grain, and detail. Maximum output: 10060×10060 pixels.r9r?r@rEflavor)sublimephotophoto_denoiserzbProcessing style: sublime for general use, photo for photographs, photo_denoiser for noisy photos.rFrXsharpenrdzNImage sharpness intensity. Higher values increase edge definition and clarity.rW smart_grainzaIntelligent grain/texture enhancement to prevent the image from looking too smooth or artificial. ultra_detailzIControls fine detail, textures, and micro-details added during upscaling.raFzNAutomatically downscale input image if output would exceed maximum resolution.Trbrca ( $mins := {"2x": 0.172, "4x": 0.343, "8x": 0.515, "16x": 0.844}; $maxs := {"2x": 2.045, "4x": 2.545, "8x": 2.889, "16x": 3.06}; { "type": "range_usd", "min_usd": $lookup($mins, widgets.scale_factor), "max_usd": $lookup($maxs, widgets.scale_factor), "format": { "approximate": true } } ) rerh)rrrrsrrurvrwrxryrzr{r|r}r~rrrs r!rz0MagnificImageUpscalerPreciseV2Node.define_schemasy8>-3w''~7P7P7PQQBBBg  l!#!1!8  !8!#!1!8 "g!#!1!8    $!l! !E(T!! . + #  /8HIII qG G G G r#r9r?rrrrrarc rKt|dkrtdt|dddt|ddd }t |\} } t |d } | | z} | | z} | |ks| |kr|rt| | }d }|d z}||z }t | | z|z|z}d D]K}|| kr ||z}||kr|}d}n2||z }|d kr&|}||z}||z }t | | z|z|z}nL|t||}|} ntd| d | d|d |d t |\}}t||| t|tddttt||ddd{Vd| ||||d{V}t|td|jtdfddd{V}t#jt'|jdd{VS)NrrrrFrrriL'rr)rrrzOutput dimensions (z') exceed maximum allowed resolution of rz0/proxy/freepik/v1/ai/image-upscaler-precision-v2rrrr)r9r?rrrrrz1/proxy/freepik/v1/ai/image-upscaler-precision-v2/rc|jSrrrs r!rz.rr#cSrr rs r!rz.rr#rr)rrrrrr*striprTrr4rrrrrrrrrrr)rr9r?rrrrramax_output_dimensionr%r$r output_width output_heightmax_dimr& max_input_dim scale_ratiorr output_dimrrrrrrs @r!rz*MagnificImageUpscalerPreciseV2Node.execute9sr  & &! + +CDD D#E66%HHHH!%C3GGGG$,U33 l005566. 0 . . .-BV2V2V eV,, 4 9 +g5 #&uv~ 'Ck'Q#R#R !.I ?22 !(9!4J!%999 )+/(&03G&GO&#-- )(< (I &3g&= +.uv~ /Kk/Y+Z+Z( .$/25GWXXXE"' H,HHHH%9HHTransfer the style from a reference image to your input image.r9z%The image to apply style transfer to.rXreference_imagez*The reference image to extract style from.r:Tr;r<style_strengthrrzPercentage of style strength.rWstructure_strength2z.Maintains the structure of the original image.r)faithfulgen_z psychedeliadetailyclear donotstyledonotstyle_sharpzStyle transfer flavor.rr[) balanceddefinioillusio 3d_cartooncolorful_anime caricaturereal super_realsoftyProcessing engine selection.)rFrXr` portrait_modedisabledenabledportrait_style)rHpop super_popz(Visual style applied to portrait images.portrait_beautifier)none beautify_facebeautify_face_maxz-Facial beautification intensity on portraits.z-Enable portrait mode for facial enhancements.fixed_generationzlWhen disabled, expect each generation to introduce a degree of randomness, leading to more diverse outcomes.rb{"type":"usd","usd":0.11}rgrh)rrrrsrrtrvrwrxru DynamicComboOptionryrzr{r|r}r~rrs r!rz,MagnificImageStyleTransferNode.define_schemas^y48-Xw0WXX0:fgg D"EE $;!#!1!8  (L!#!1!8 vvv4    ;! %%#..z2>>..% "$4,L,L,L,V!/!"!" !#$9,Z,Z,Z,[!/!"!" $L)&,   & 8! !FP!! . + #  4kX X X X r#r9rr:rrrr[rr rc \Kt|dkrtdt|dkrtdt|dddt|dddt|dd t|dd |d d k} |d d } |dd} t |||gdd{V} t |tddtt| d| d|r|nd||| | r| nd| r| dkr| nd|||  d{V}t|td|j tddd{V}tj t|jdd{VS)Nrr(Exactly one reference image is required.rrFrrrrrrrHrrr)rz)/proxy/freepik/v1/ai/image-style-transferrrr) r9rr:rr is_portraitrrrr[r rz*/proxy/freepik/v1/ai/image-style-transfer/rc|jSrrrs r!rz8MagnificImageStyleTransferNode.execute..rr#rrrr)rrrrgetrrrrr rrrrrr)rr9rr:rrrr[rr rrr uploaded_urlsrrs r!rz&MagnificImageStyleTransferNode.executesP  & &! + +CDD D  0 0A 5 5GHH H#E66%HHHH#OVVERRRR!%C3GGGG!/cSQQQQ#O4 A &**+;ZHH+//0EvNN7e_=Ubcddddddddd #  HQW X X X'*#A& -a 0!'1vvT-#5'1<F~~$;F$rK^bhKhKh$7$7nr!1             $ '  _+J]__ ` ` `'//          }#?@XYZ@[#\#\\\\\\\]]]r#N)rrrrrrrsrr*rrrrrr r#r!rrsY Y [Y v4^{4^4^ 4^  4^  4^4^4^)4^4^ 4^4^4^[4^4^4^r#rceZdZedZe ddejdededede de d e d e d ejdzd e j fd Z dS)MagnificImageRelightNodec! tjddddtjddtjdd d d tjd ddddtjjtjdgddtj dddd tj dd dd tj dd dd tj dtj dgtj d tjd!ddd"d#tjjtjd$ddd"d%tjjtjd&ddd"d'tjjtjd(ddd"d)tjjtjd*ddd"d+tjjtjd,gd-d.tjd/gd0d1tjd2gd3d4tj d5d d67g gd8tjd9d d:;g tj gtj jtj jtj jgd tjd<=> S)?NrzMagnific Image Relightr8zWRelight an image with lighting adjustments and optional reference-based light transfer.r9zThe image to relight.rr:Tr;zFDescriptive guidance for lighting. Supports emphasis notation (1-1.4).)r=r>rXlight_transfer_strengthrrz(Intensity of light transfer application.rWstyle)rHdarker_but_realisticcleansmoothbrightercontrasted_n_hdrjust_compositionzStylistic output preference.rinterpolate_from_originalFzrXz2Fine-tuning options for advanced lighting control.rz3Optional reference image to transfer lighting from.)optionalrXr r rh)rrrrsrrtrvrwrxruryr rrzr{r|r}r~rrs r!rz&MagnificImageRelightNode.define_schema#sy.1-qw0GHH "d    -F!#!1!8 ;     /!Z! !    ' L! !    & O! ! %%'..z2>>..% " $,()(+,.,W131A1H !-!"!"!# $,()(+,.,U131A1H !-!"!"!# $0()(+,.,L131A1H !-!"!"!# $.()(+,.,B131A1H !-!"!"!# $0()(+,.,J131A1H !-!"!"!#$, -& -& -&-K!/!"!"!#$6,m,m,m,N!/!"!" !#$6-&-&-&-d#!/!"!"(!# 0 0$6,0,_!1!"!"cVYY\zQ&``B%!QaUn!! . + #  4Ig g g g r#Nr9r:rrr"r#r$r%rrc Kt|dkrtd| "t| dkrtdt|dddt|dd | %t| dddt| dd t ||d d{Vd } d} | t || d d{Vd } d} |d d krNt |d|d|d|d|d|d|d|d|d } t |tddtt| |r|nd| ||||||  d{V} t|td| j tdd !d{V}tj t|jd d{VS)"NrrrrrFrrrrrr%rr&r'r(r)r*r[r.r4r ) r&r'r(r)r*r[r.r4r z"/proxy/freepik/v1/ai/image-relightrr) r9r:#transfer_light_from_reference_imagerr"r#rr$r%rz#/proxy/freepik/v1/ai/image-relight/rc|jSrrrs r!rz2MagnificImageRelightNode.execute.. rr#rr)rrrrrrrrrrrrrrrr)rr9r:rrr"r#r$r%r image_url reference_url adv_settingsrrs r!rz MagnificImageRelightNode.executes  & &! + +CDD D  &+?+P+PTU+U+UGHH H#E66%HHHH!%C3GGGG  & 'PU V V V V %o#QT U U U U4S%ANNNNNNNNNPQR   &# >>(2(2,\:*:6,\:(2!23E!F!23E!F!23E!F   L$  A& Q Q Q'$!'1vvT4A(?*C"3!1".             '  X;CVXX Y Y Y'//          }#?@XYZ@[#\#\\\\\\\]]]r#r)rrrrrrrsrr*rrrrrr r#r!rr"sh h [h T/3@^@^{@^@^"% @^  @^ $( @^ @^@^1@^t+@^ @^@^@^[@^@^@^r#rc deZdZedZedejdededede j f dZ dS) MagnificImageSkinEnhancerNodectjddddtjddtjdd d d d tjj tjd d d ddtjj tjdtjdgtjdtjdd d ddtjj gtjdtj dgddggdgtj gtj j tj j tj jgdtjtjdgd S)!NrHzMagnific Image Skin Enhancerr8z>Skin enhancement for portraits with multiple processing modes.r9zThe portrait image to enhance.rrrrzSharpening intensity level.rWrr)zSmart grain intensity level.modecreativer skin_detailPzSkin detail enhancement level.flexiblerG) enhance_skinimprove_lightingenhance_everythingtransform_to_real no_make_upz Enhancement optimization target.rzProcessing mode: creative for artistic enhancement, faithful for preserving original appearance, flexible for targeted optimization.Trcz ( $rates := {"creative": 0.29, "faithful": 0.37, "flexible": 0.45}; {"type":"usd","usd": $lookup($rates, widgets.mode)} ) rerh)rrrrsrrvrwrxr rrurzr{r|r}r~rrrs r!rz+MagnificImageSkinEnhancerNode.define_schemasy37-Xw0PQQ 9!#!1!8  !:!#!1!8 %%..z2>>..& " $1()(+,.,L131A1H !-!"!"   ..& "$3-&-&-&-O!/ !" !" @:E&%%%8t!! . + #  /AAAOP P P P r#r9rrrJrcKt|dkrtdt|dddt|ddt ||dd d{Vd }|d }|d krd }t |||}n=|dkrd}t ||||d}nd}t||||d}t|td|dt|d{V} t|td| j tddd{V} tjt| jd d{VS)NrrrrFrrrirrrJrK)r9rrrrL)r9rrrLrNrG)r9rrrGz#/proxy/freepik/v1/ai/skin-enhancer/rrrrc|jSrrrs r!rz7MagnificImageSkinEnhancerNode.execute..rr#rr)rrrrrr r r rrrrrrrrr) rr9rrrJrD selected_modeendpointrrrs r!rz%MagnificImageSkinEnhancerNode.executegs   & &! + +CDD D#E66%HHHH!%C3GGGG4S%A\ghhhhhhhhhjkl V J & &!H3'DD j ( (!H3' / DD"H3'"?3 D$  M8MMV\ ] ] ]'          '  X;CVXX Y Y Y'//          }#?@XYZ@[#\#\\\\\\\]]]r#N) rrrrrrrsr*rrrrr r#r!rHrHsQ Q [Q f4^{4^4^ 4^ $ 4^  4^4^4^[4^4^4^r#rHcLeZdZedeeejfdZdS)MagnificExtensionrcFKtttttgSr)r6rrrrH)selfs r! get_node_listzMagnificExtension.get_node_lists! . . * $ )   r#N) rrrrlisttyper ComfyNoder\r r#r!rYrYsE  T$r|*<%=   X   r#rYc"KtSr)rYr r#r!comfy_entrypointras   r#),r+typing_extensionsrcomfy_api.latestrrrcomfy_api_nodes.apis.magnificrrr r r r r rrrrrcomfy_api_nodes.utilrrrrrrrrrrr/floatr"r*r4r_r6rrrrHrYrar r#r!rgs &&&&&&6666666666                                                     %--c-#-RW----p^p^p^p^p^ p^p^p^f]^]^]^]^]^]^]^]^@R^R^R^R^R^R\R^R^R^jm^m^m^m^m^r|m^m^m^`J^J^J^J^J^BLJ^J^J^Z           1r#