o ip@sddlmZddlmZddlmZmZddlmZddl Z ddl Z ddl m Z mZmZmZmZddlmZmZmZmZmZidd 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 ZddddddddddddddddZgdZddZGdddejZGdddejZGdddejZ Gdd„deZ!de!fddńZ"dS))BytesIO)override)IOComfyExtension)ImageN)IdeogramGenerateRequestIdeogramGenerateResponse ImageRequestIdeogramV3RequestIdeogramV3EditRequest) ApiEndpointbytesio_to_image_tensordownload_url_as_bytesioresize_mask_to_imagesync_opAutoAUTOz 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:1rrrqrproz10:16z16:10rtrsrnrmrurvrl)Fr512x1536576x1408576x1472576x1536640x1344640x1408640x1472640x1536704x1152704x1216704x1280704x1344704x1408704x1472736x1312768x1088768x1216768x1280768x1344800x1280832x960832x1024832x1088832x1152832x1216832x1248864x1152896x960896x1024896x1088896x1120896x1152960x832960x896960x1024960x10881024x8321024x8961024x960 1024x10241088x7681088x8321088x8961088x9601120x8961152x7041152x8321152x8641152x8961216x7041216x7681216x8321248x8321280x7041280x7681280x8001312x7361344x6401344x7041344x7681408x5761408x6401408x7041472x5761472x6401472x7041536x5121536x5761536x640csRg}|D]}t|IdH}t|dd}||q|r%tj|dd}|Std)zAHelper function to download and process multiple images from URLsNRGB)moder)dimzNo valid images were processed)rr appendtorchcat Exception) image_urls image_tensors image_url image_bytesio img_tensorstacked_tensorsr^z&IdeogramV1.execute..,No image URLs were generated in the response) rgetrr rrr r0lenrr NodeOutputr) r#rrrrrrrr,responserrrrexecute:s4    zIdeogramV1.executeN)Frlrrrr__name__ __module__ __qualname__ classmethodr$r@rrrrrs Qrc@s8eZdZeddZe       dd d Zd S) IdeogramV2cCs(tjddddtjjddddd tjjd d d d tjjdttddddtjjdtt ddddtjjdgddddddtj jddddddtj j ddtjjdgd d!d"dddtjjd#ddd$dd%tj jd&ddd'dtj j dd(g tj gtjjtjjtjjgdtjtjd&d gd)d*d+d, S)-NrFz Ideogram V2rz-Generates images using the Ideogram V2 model.rTrrrrFrrrrlzPThe 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)} ) rr)rrrrrrrrr V1_V1_RES_MAPrrrrrrrrrr r!r"rrrr$fs   OzIdeogramV2.define_schemaFrlrrrrNrrc st|d}t|d}|rdnd} d} d} |dkr|} n|dkr$|nd} t|tdddttt|| | || | |dkr<|nd|dkrC|nd|rH|nd| rM| ndd d d d IdH}|jrct |jd krgt ddd|jD}|sut dt t |IdHS)N V_2_TURBOV_2rrar'r(r)rN) rr,rrrrGrrHr color_paletter-r)endpointr/r0r1rr2cSr3rr4r6rrrr9r:z&IdeogramV2.execute..r;)rr<rOrr rrr r0r=rrr>r)r#rrrrGrrrHrrrRr,final_resolutionfinal_aspect_ratior?rrrrr@sF       zIdeogramV2.executeN) FrlrrrrNrrrrArrrrrFds irFc@s:eZdZeddZe         d d d ZdS) IdeogramV3cCs@tjddddtjjddddd tjjd d dd tjjd ddd tjjdtt ddddtjjdt ddddtjjdgddddddtj jddddddtj j ddtj jd ddd!dtj j dd"tjjd#gd$d%d&dddtjjd'd(dd tjjd)d*dd g tjgtjjtjjtjjgdtjtjd#d gd'gd+d,d-d. S)/NrVz 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)rrlzPThe aspect ratio for image generation. Ignored if resolution is not set to Auto.rrGrzaThe resolution for image generation. If not set to Auto, this overrides the aspect_ratio setting.rrrrrrrrrrrrrrendering_speed)DEFAULTTURBOQUALITYrZz;Controls the trade-off between generation speed and qualitycharacter_imagez$Image to use as character reference.character_maskz,Optional mask for character reference image.)rr a: ( $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)} ) rr)rrrrrMaskrr V3_RATIO_MAPrV3_RESOLUTIONSrrrrrrrrr r!r"rrrr$ s   OzIdeogramV3.define_schemaNrrlrrrrZc  s| dkrd} d} d} | dur| }| dur]t| | dd} d| } | jdd| jddkr6td| d tj}t |}t }|j |d d | d |} d | _|d tj}t |}t }|j |d d | d |} d| _n| durtd|durD|durD|}t||dd}d|}|jdd|jddkrtd|d tj}t |}t }|j |d d | d |}d|_|d tj}t |}t }|j |d d | d |}d |_t|| d}|dkr||_|d kr||_|dkr||_||d}| r*| |d<| r1| |d<t|tdddt||dddIdH}nr|dusN|durRtdt|| d}|dkra||_n|dkrqt|}|rq||_|dkry||_|d kr||_|dkr||_i}| r| |d<| r| |d<|rd|_t|tdddt||r|nddddIdH}|jrt|jd krtd d!d"|jD}|std#tt |IdHS)$NBALANCEDrZF)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)rrYr)rWrXcharacter_reference_imagescharacter_mask_binaryz /proxy/ideogram/ideogram-v3/editr(r)zmultipart/form-data)r/r0files content_typer1z;Ideogram V3 image editing requires both an image AND a maskrrlz$/proxy/ideogram/ideogram-v3/generate)rSr/r0rjrkr1r2cSr3rr4r6rrrr9r:z&IdeogramV3.execute..r;)!squeezecpurshapernumpyastypenpuint8r fromarrayrsaveseeknamer magic_promptrrrr rr rGr`r<rrHr0r=rr>r)r#rrWrXrGrrrrrYr]r^character_img_binaryri input_tensormask_npmask_img mask_byte_arrimg_npimg img_byte_arr img_binary mask_binary edit_requestrjr? gen_request v3_aspectrrrrr@}s                         zIdeogramV3.execute) NNrrlrrrrZNNrArrrrrV s prVc@s(eZdZedeeejfddZdS)IdeogramExtensionreturncs tttgSN)rrFrV)selfrrr get_node_list"s zIdeogramExtension.get_node_listN) rBrCrDrrtyper ComfyNoderrrrrr!srrcstSr)rrrrrcomfy_entrypoint+sr)#iortyping_extensionsrcomfy_api.latestrrPILrrorqrcomfy_api_nodes.apis.ideogramrrr r r comfy_api_nodes.utilr r rrrrOrr`rarrrrFrVrrrrrrs        !"#$%&'()*+,-./0123456789:;<=>?@ABCDESI~'