+j ~\ddlmZddlmZddlmZmZddlmZddl Z ddl Z ddl m Z mZmZmZmZmZddlmZmZmZmZmZmZidd d d d d ddddddddddddddddddd d!d"d#d$d%d&d'd(d)id*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;dd?d@dAdBdCdDdEdFdGdHdIdJdKidLdMdNdOdPdQdRdSdTdUdVdWdXdYdZd[d\d]d^d_d`dadbdcdddedfdgdhdidjdkdldmidndodpdqdrdsdtdudvdwdxdydzd{d|d}d~ddddddddddddddddddddddddddddd Zdddddddddddd ZddddddddddddddddZgdZdZGddejZ GddejZ!GddejZ"GddejZ#Gd„deZ$de$fdńZ%dS))BytesIO)override)IOComfyExtension)ImageN)IdeogramGenerateRequestIdeogramGenerateResponse ImageRequestIdeogramV3RequestIdeogramV3EditRequestIdeogramV4Request) ApiEndpointbytesio_to_image_tensordownload_url_as_bytesioresize_mask_to_imagesync_opvalidate_stringAutoAUTOz 512 x 1536RESOLUTION_512_1536z 576 x 1408RESOLUTION_576_1408z 576 x 1472RESOLUTION_576_1472z 576 x 1536RESOLUTION_576_1536z 640 x 1024RESOLUTION_640_1024z 640 x 1344RESOLUTION_640_1344z 640 x 1408RESOLUTION_640_1408z 640 x 1472RESOLUTION_640_1472z 640 x 1536RESOLUTION_640_1536z 704 x 1152RESOLUTION_704_1152z 704 x 1216RESOLUTION_704_1216z 704 x 1280RESOLUTION_704_1280z 704 x 1344RESOLUTION_704_1344z 704 x 1408RESOLUTION_704_1408z 704 x 1472RESOLUTION_704_1472z 720 x 1280RESOLUTION_720_1280z 736 x 1312RESOLUTION_736_1312z 768 x 1024RESOLUTION_768_1024z 768 x 1088RESOLUTION_768_1088z 768 x 1152RESOLUTION_768_1152z 768 x 1216RESOLUTION_768_1216z 768 x 1232RESOLUTION_768_1232z 768 x 1280RESOLUTION_768_1280z 768 x 1344RESOLUTION_768_1344z 832 x 960RESOLUTION_832_960z 832 x 1024RESOLUTION_832_1024z 832 x 1088RESOLUTION_832_1088z 832 x 1152RESOLUTION_832_1152z 832 x 1216RESOLUTION_832_1216z 832 x 1248RESOLUTION_832_1248z 864 x 1152RESOLUTION_864_1152z 896 x 960RESOLUTION_896_960z 896 x 1024RESOLUTION_896_1024z 896 x 1088RESOLUTION_896_1088z 896 x 1120RESOLUTION_896_1120z 896 x 1152RESOLUTION_896_1152z 960 x 832RESOLUTION_960_832z 960 x 896RESOLUTION_960_896z 960 x 1024RESOLUTION_960_1024z 960 x 1088RESOLUTION_960_1088z 1024 x 640RESOLUTION_1024_640z 1024 x 768RESOLUTION_1024_768z 1024 x 832RESOLUTION_1024_832z 1024 x 896RESOLUTION_1024_896z 1024 x 960RESOLUTION_1024_960z 1024 x 1024RESOLUTION_1024_1024z 1088 x 768RESOLUTION_1088_768z 1088 x 832RESOLUTION_1088_832z 1088 x 896RESOLUTION_1088_896z 1088 x 960RESOLUTION_1088_960z 1120 x 896RESOLUTION_1120_896z 1152 x 704RESOLUTION_1152_704z 1152 x 768RESOLUTION_1152_768z 1152 x 832RESOLUTION_1152_832z 1152 x 864RESOLUTION_1152_864z 1152 x 896RESOLUTION_1152_896z 1216 x 704RESOLUTION_1216_704z 1216 x 768RESOLUTION_1216_768z 1216 x 832RESOLUTION_1216_832z 1232 x 768RESOLUTION_1232_768z 1248 x 832RESOLUTION_1248_832z 1280 x 704RESOLUTION_1280_704z 1280 x 720RESOLUTION_1280_720z 1280 x 768RESOLUTION_1280_768z 1280 x 800RESOLUTION_1280_800z 1312 x 736RESOLUTION_1312_736z 1344 x 640RESOLUTION_1344_640RESOLUTION_1344_704RESOLUTION_1344_768RESOLUTION_1408_576RESOLUTION_1408_640RESOLUTION_1408_704RESOLUTION_1472_576RESOLUTION_1472_640RESOLUTION_1472_704RESOLUTION_1536_512RESOLUTION_1536_576RESOLUTION_1536_640) z 1344 x 704z 1344 x 768z 1408 x 576z 1408 x 640z 1408 x 704z 1472 x 576z 1472 x 640z 1472 x 704z 1536 x 512z 1536 x 576z 1536 x 640 ASPECT_1_1 ASPECT_4_3 ASPECT_3_4 ASPECT_16_9 ASPECT_9_16 ASPECT_2_1 ASPECT_1_2 ASPECT_3_2 ASPECT_2_3 ASPECT_4_5 ASPECT_5_4) 1:14:33:416:99:162:11:23:22:34:55:41x33x11x22x19x1616x910x1616x102x33x23x44x34x55x41x1)z1:3z3:1rurtrsrrz10:16z16:10rwrvrqrprxryro)Fr512x1536576x1408576x1472576x1536640x1344640x1408640x1472640x1536704x1152704x1216704x1280704x1344704x1408704x1472736x1312768x1088768x1216768x1280768x1344800x1280832x960832x1024832x1088832x1152832x1216832x1248864x1152896x960896x1024896x1088896x1120896x1152960x832960x896960x1024960x10881024x8321024x8961024x960 1024x10241088x7681088x8321088x8961088x9601120x8961152x7041152x8321152x8641152x8961216x7041216x7681216x8321248x8321280x7041280x7681280x8001312x7361344x6401344x7041344x7681408x5761408x6401408x7041472x5761472x6401472x7041536x5121536x5761536x640cKg}|D]=}t|d{V}t|d}||>|rtj|d}nt d|S)zAHelper function to download and process multiple images from URLsNRGB)moder)dimzNo valid images were processed)rrappendtorchcat Exception) image_urls image_tensors image_url image_bytesio img_tensorstacked_tensorss >/home/wildlama/comfy/ComfyUI/comfy_api_nodes/nodes_ideogram.pydownload_and_process_imagesrsM)) 5i@@@@@@@@ ,]GGG Z((((:)Mq9998999 cHeZdZedZe d dZd S) IdeogramV1ctjddddtjdddd tjd d d tjdt tdddtjdgdddddtj ddddddtj j dtjdddddtj dddd dtj j d!gtj gtjjtjjtjjgdtjtjdd g"d#$% S)&Nrz Ideogram V1partner/image/Ideogramz-Generates images using the Ideogram V1 model.promptTPrompt for the image generation multilinedefaulttooltipturboFHWhether to use turbo mode (faster generation, potentially lower quality)rr aspect_ratioroz&The aspect ratio for image generation.optionsrroptionalmagic_prompt_optionrONOFFr5Determine if MagicPrompt should be used in generationrrrradvancedseedrrminmaxstepcontrol_after_generate display_modernegative_prompt-Description of what to exclude from the imagerrrr num_imagesrrrrrrwidgetsz ( $n := widgets.num_images; $base := (widgets.turbo = true) ? 0.0286 : 0.0858; {"type":"usd","usd": $round($base * $n, 2)} )  depends_onexpr node_id display_namecategory descriptioninputsoutputshidden is_api_node price_badge)rSchemaStringInputBooleanCombolistV1_V2_RATIO_MAPkeysInt NumberDisplaynumberrOutputHiddenauth_token_comfy_orgapi_key_comfy_org unique_id PriceBadgePriceBadgeDependsclss r define_schemazIdeogramV1.define_schemasy &-G "=      !f! " !5!5!7!788!D! )111"S!!  "+/!#!1!8!   %"K!    !#!1!8!Y5n!! . + #  /w8OPPP   IN N N N rFrorrrrcKt|d}|rdnd}t|tddtt t |||||dkr|nd|dkr|nd|r|nd d d{V} | jrt| jd krtd d| jD} | stdtj t| d{VS)N V_1_TURBOV_1/proxy/ideogram/generatePOSTpathmethodrdr)rmodelrrrrr image_requestrresponse_modeldata max_retriesr(No images were generated in the responsec*g|]}|j |jSurl.0 image_datas r z&IdeogramV1.execute..`!WWW WjnWWWr,No image URLs were generated in the response) rgetrrr rr r8lenrr NodeOutputr) r)rrrrrrrr3responsers rexecutezIdeogramV1.execute<sd'**<>> $/ %  7 G G G3(*!)1=1M1MSW@SW]@]@])<)z#J#JJJJJJJKKKrN)Frorrrr__name__ __module__ __qualname__ classmethodr*rIr<rrrrskO O [O b" &L&L&L[&L&L&LrrcNeZdZedZe d d Zd S) IdeogramV2cztjddddtjdddd tjd d d tjdt tdddtjdt tdddtjdgdddddtj ddddddtj j dtjdgd d!d"ddtjd#ddd$d%tj d&ddd'dtj j d(g tj gtjjtjjtjjgdtjtjd&d g)d*+, S)-NrPz Ideogram V2rz-Generates images using the Ideogram V2 model.rTrrrrFrrrrozPThe aspect ratio for image generation. Ignored if resolution is not set to AUTO.r resolutionrzaThe resolution for image generation. If not set to AUTO, this overrides the aspect_ratio setting.rrrrrrrrrr style_type)rGENERAL REALISTICDESIGN RENDER_3DANIMENONEz#Style type for generation (V2 only)rrrrrrrz ( $n := widgets.num_images; $base := (widgets.turbo = true) ? 0.0715 : 0.1144; {"type":"usd","usd": $round($base * $n, 2)} ) r r )rrrrrrrrr V1_V1_RES_MAPrrr rr!r"r#r$r%r&r'r(s rr*zIdeogramV2.define_schemahs]y &-G "=      !f! " !5!5!7!788!n!   !3!3!5!566"[! )111"S!!  "+/!#!1!8!   \\\"A!!  %"K!    !#!1!8!yM^!! . + #  /w8OPPP   yf f f f rFrorrrrYrrc pKt|d}t|d}|rdnd} d} d} |dkr|} n |dkr|nd} t|t ddt t t|| | || | |dkr|nd|dkr|nd|r|nd| r| nd d d{V}|jrt|jd krtdd|jD}|stdtj t|d{VS)N V_2_TURBOV_2rrdr.r/r0rY) rr3rrrrRrrSr color_paletter4r)endpointr7r8r9rr:c*g|]}|j |jSr<r=r?s rrBz&IdeogramV2.execute..rCrrD)rrErZrrr rr r8rFrrrGr)r)rrrrRrrrSrrr^r3final_resolutionfinal_aspect_ratiorHrs rrIzIdeogramV2.executes'**<>> "&&z488 $/ % !   )  1=1M1MSW   &@PPP3(*!)!3/@SW]@]@])<)z#J#JJJJJJJKKKrN) FrorrrrYrrrrJr<rrrPrPfstg g [g R" 6L6L6L[6L6L6LrrPcPeZdZedZe d d ZdS) IdeogramV3ctjddddtjdddd tjd d d tjd dd tjdtt dddtjdtdddtjdgdddddtj ddddddtj j dtj d ddd!dtj j d"tjd#gd$d%d&ddtjd'd(d tjd)d*d g tjgtjjtjjtjjgdtjtjd#d gd'g+d,-. S)/Nrdz Ideogram V3rzGenerates images using the Ideogram V3 model. Supports both regular image generation from text prompts and image editing with mask.rTrz*Prompt for the image generation or editingrimagez+Optional reference image for image editing.)rrmaskz;Optional mask for inpainting (white areas will be replaced)rrozPThe aspect ratio for image generation. Ignored if resolution is not set to Auto.rrRrzaThe resolution for image generation. If not set to Auto, this overrides the aspect_ratio setting.rrrrrrrrrrrrrrendering_speedDEFAULTTURBOQUALITYrjz;Controls the trade-off between generation speed and qualitycharacter_imagez$Image to use as character reference.character_maskz,Optional mask for character reference image.)rra: ( $n := widgets.num_images; $speed := widgets.rendering_speed; $hasChar := inputs.character_image.connected; $base := $contains($speed,"quality") ? ($hasChar ? 0.286 : 0.1287) : $contains($speed,"default") ? ($hasChar ? 0.2145 : 0.0858) : $contains($speed,"turbo") ? ($hasChar ? 0.143 : 0.0429) : 0.0858; {"type":"usd","usd": $round($base * $n, 2)} ) r r )rrrrrMaskrr V3_RATIO_MAPrV3_RESOLUTIONSrrr r!r"r#r$r%r&r'r(s rr*zIdeogramV3.define_schemasy &-p "H   I!  Y! " !2!2!4!455!n!  *"[! )111"S!!  "+/!#!1!8!    !#!1!8!%;;;%Y!! %B!  $J!QM^!! . + #  /9JL8Yctbuvvv {m m m m rNrrorrrrjc t K| dkrd} d} d} | | }| t| | d} d| z } | jdd| jddkrt d| d ztj}tj |}t}| |d | d |} d | _|d ztj}tj |}t}| |d | d |} d| _n| t d|%|"|}t||d}d|z }|jdd|jddkrt d|d ztj}tj |}t}| |d | d |}d|_|d ztj}tj |}t}| |d | d |}d |_t|| }|dkr||_|d kr||_|dkr||_||d}| r| |d<| r| |d<t'|t)ddt*||ddd{V}n||t dt-|| }|dkr||_n)|dkr#t0|}|r||_|dkr||_|d kr||_|dkr||_i}| r| |d<| r| |d<|rd|_t'|t)ddt*||r|ndddd{V}|jrt;|jd krt d d!|jD}|st d"t=jtA|d{VS)#NBALANCEDrjF)allow_gradientg?rz.Character mask and image must be the same sizePNG)formatrzmask.pngz image.pngz5Character mask requires character image to be presentz$Mask and Image must be the same size)rrhr)rfrgcharacter_reference_imagescharacter_mask_binaryz /proxy/ideogram/ideogram-v3/editr/r0zmultipart/form-data)r7r8files content_typer9z;Ideogram V3 image editing requires both an image AND a maskrroz$/proxy/ideogram/ideogram-v3/generate)r_r7r8r{r|r9r:c*g|]}|j |jSr<r=r?s rrBz&IdeogramV3.execute..rCrrD)!squeezecpurshapernumpyastypenpuint8r fromarrayrsaveseeknamer magic_promptrrrrr r rRrprErrSr8rFrrGr)r)rrfrgrRrrrrrhrmrncharacter_img_binaryrz input_tensormask_npmask_img mask_byte_arrimg_npimg img_byte_arr img_binary mask_binary edit_requestr{rH gen_request v3_aspectrs rrIzIdeogramV3.executes j ( ('O# $  &*224488::L)!5nofk!l!l!l!$~!5!'+/DQrT/JJJ#$TUUU)11337799??AACGOOPRPXYY ?733 '  mE :::""1%%%(5%-7%*"((**S088BBF/&))C"99L HH\%H 0 0 0   a #/ (3 % %  'STT T  !1 ==??..00L'eEJJJD:Dz!""~QrT!222 FGGG#((**S088BBF/&))C"99L HH\%H 0 0 0   a %J)JO||~~))++1133c9AA"(KKGw//H#IIM MM-M 6 6 6   q ! ! !'K)K 1 /L #f,,,? )qyy$( !A~~*4 '$#E$ K6J23$ G1F-.$!CFSSS7!2HH $"2YZZ Z, /K V##)3 &&&&(,,\:: 9/8K,#f,,+> (qyy#'  A~~)3 &E# K6J23$ G1F-. 0)/ &$$*PY_```7 $.ee$2H} HHM 2 2a 7 7FGG GWWx}WWW  LJKK K}#>z#J#JJJJJJJKKKr) NNrrorrrrjNNrJr<rrrdrd s}n n [n ` " !`L`L`L[`L`L`Lrrdc LeZdZedZededededefdZdS) IdeogramV4c`tjddddtjdddd tjd gd d tjdgdddtjddddddtjjgtj gtj j tj j tj j gdtjtjdgd S)Nrz Ideogram V4rzAGenerates images using the Ideogram 4.0 model from a text prompt.rTrz%Text prompt for the image generation.rrR)rz2048x2048 (1:1)z1440x2880 (1:2)z2880x1440 (2:1)z1664x2496 (2:3)z2496x1664 (3:2)z1792x2240 (4:5)z2240x1792 (5:4)z1440x2560 (9:16)z2560x1440 (16:9)z1600x2560 (5:8)z2560x1600 (8:5)z1728x2304 (3:4)z2304x1728 (4:3)z1296x3168 (9:22)z3168x1296 (22:9)z1152x2944 (9:23)z2944x1152 (23:9)z1248x3328 (3:8)z3328x1248 (8:3)z1280x3072 (5:12)z3072x1280 (12:5)r)rrrhrirjz.rCrrD) rrrr r splitr8rFrrrGr)r)rrRrhrrHrs rrIzIdeogramV4.executews1 !DDDD  CF S S S3""7AV7K7K:++C0033QU /           } HHM 2 2a 7 7FGG GWWx}WWW  LJKK K}#>z#J#JJJJJJJKKKrN)rKrLrMrNr*strintrIr<rrrr#sO O [O bLLL L  LLL[LLLrrcLeZdZedeeejfdZdS)IdeogramExtensionreturnc:KttttgSN)rrPrdr)selfs r get_node_listzIdeogramExtension.get_node_lists       rN) rKrLrMrrtyper ComfyNoderr<rrrrsE  T$r|*<%=   X   rrrc"KtSr)rr<rrcomfy_entrypointrs   r)&iortyping_extensionsrcomfy_api.latestrrPILrrrrcomfy_api_nodes.apis.ideogramrr r r r r comfy_api_nodes.utilrrrrrrrZrrprqrrrrPrdrrrr<rrrs;&&&&&&//////// PP$P$P$ P $ P $ P$P$P$P$P$P$P$P$P$P $!P"$#PP$$%P&$'P($)P*$+P,$-P.$/P0$1P2$3P4"5P6$7P8$9P:$;P<$=P>$?P@$APB"CPD$EPPPF$GPH$IPJ$KPL"MPN"OPP$QPR$SPT$UPV$WPX$YPZ$[P\$]P^&_P`$aPb$cPd$ePf$gPPPh$iPj$kPl$mPn$oPp$qPr$sPt$uPv$wPx$yPz${P|$}P~$P@$APB$CPD$EPF$GPH$IPPJ%$$$$$$$$$$_PPP f                     $GGGR*{L{L{L{L{L{L{L{L|cLcLcLcLcLcLcLcLLTLTLTLTLTLTLTLTLnnLnLnLnLnLnLnLnLb         1r