o i|,@sddlmZddlmZmZmZmZmZddlmZddl m Z m Z m Z m Z Gddde ZGddde ZGd d d e ZGd d d e eeefZGd dde ZGddde ZGddde ZGddde ZGddde eZGddde ZGddde ZGdddeeZGddde ZGdd d eeZGd!d"d"eeZGd#d$d$e ZGd%d&d&e ZGd'd(d(e Z Gd)d*d*eeZ!Gd+d,d,eeZ"Gd-d.d.eeZ#Gd/d0d0e Z$Gd1d2d2e Z%d3S)4)Enum)OptionalListDictAnyUnion)datetime) BaseModelField RootModel StrictBytesc@"eZdZUedddZeed<dS)IdeogramColorPalette1.z Name of the preset color palette descriptionnameN__name__ __module__ __qualname__r rstr__annotations__rr;/mnt/c/Users/fbmor/ComfyUI/comfy_api_nodes/apis/ideogram.pyr rc@sDeZdZUeddddZeeed<eddddd Zee ed <dS) MemberNzHexadecimal color codez^#[0-9A-Fa-f]{6}$)rpatterncolorz#Optional weight for the color (0-1)gg?rgeleweight) rrrr rrrrr!floatrrrrr s rc@s&eZdZUedddZeeed<dS)IdeogramColorPalette2.z0Array of color definitions with optional weightsrmembersN)rrrr r$rrrrrrrr#s r#c@s*eZdZUedddZeeefed<dS)IdeogramColorPalette.zjA color palette specification that can either use a preset name or explicit color definitions with weightsrrootN) rrrr r&rrr#rrrrrr%s r%c@seZdZUedddZeeed<edddZee ee fed<edddZ eeed<ed d dZ eed <edd dZ eeed <edddddZeeed<ed ddZeed<edddZeeed<edddddZeeed<edddZeeed<dS) ImageRequestNzOptional. The aspect ratio (e.g., 'ASPECT_16_9', 'ASPECT_1_1'). Cannot be used with resolution. Defaults to 'ASPECT_1_1' if unspecified.r aspect_ratioz8Optional. Color palette object. Only for V_2, V_2_TURBO. color_palettez2Optional. MagicPrompt usage ('AUTO', 'ON', 'OFF').magic_prompt_option.z*The model used (e.g., 'V_2', 'V_2A_TURBO')modelzROptional. Description of what to exclude. Only for V_1, V_1_TURBO, V_2, V_2_TURBO.negative_promptzrr8rrrrr<gs r<c@r ) StyleCode.z^[0-9A-Fa-f]{8}$)rr&N)rrrr r&rrrrrrr?prr?c@sneZdZUdZeeed<dZeeed<dZ eeed<dZ ee ed<dZ eeed<dZ eeed<dS)Datum1Nr9r0r1r3r4r:)rrrr9rr;rr0rr1r3r5r4r:rrrrr@ts r@c@s2eZdZUdZeeed<dZeee ed<dS)IdeogramV3IdeogramResponseNr=r>) rrrr=rrrr>rr@rrrrrA}s rAc@eZdZdZdZdZdS)RenderingSpeed1TURBODEFAULTQUALITYN)rrrrDrErFrrrrrCrCc@seZdZUdZeeeefed<dZ ee ed<e ddddZ ee ed<dZeeed<eed <e dd d dZee ed <dZeeeed <dZeee ed<dS)IdeogramV3ReframeRequestNr)imager-r.rr r/rendering_speedr1rr2r3 style_codesstyle_reference_images)rrrr)rrrrrrIr r r/r5rKrCr3rLrrMrrrrrHs rHc@rB) MagicPromptAUTOONOFFN)rrrrOrPrQrrrrrNrGrNc@seZdZdZdZdZdZdS) StyleTyperOGENERAL REALISTICDESIGNN)rrrrOrSrTrUrrrrrRs rRc@seZdZUdZeeed<dZeeee fed<dZ ee ed<e ddddZ eeed <dZeeed <dZeeed <e ddd dZeeed <eed<dZeeed<dZeeed<e ddddZeeed<dZeeeed<dZeee ed<dZeeed<dS)IdeogramV3RemixRequestNr(r)rI2r-drJ image_weight magic_promptr,r.r/r0rKr1rr2r3rLrMr4)rrrr(rrrr)rrrIr r rYr5rZrNr,r/rKrCr1r3rLrrMr4rRrrrrrVs rVc@seZdZUdZeeeefed<dZ ee ed<dZ ee ed<e ddddZeeed<eed <dZeeed <e dd d dZeeed <dZeeeed<dZeee ed<dS)"IdeogramV3ReplaceBackgroundRequestNr)rIrZr-r.rJr/r0rKrr2r3rLrM)rrrr)rrrrrrIr rZrNr r/r5rKrCr3rLrrMrrrrr[s r[c@s&eZdZUedddgdZeed<dS) ColorPalette.zName of the color palettePASTELrexamplesrNrrrrrr\s r\c@seZdZdZdZdS) MagicPrompt2rPrQN)rrrrPrQrrrrr`sr`c@s eZdZdZdZdZdZdZdS) StyleType1rOrSrTrUFICTIONN)rrrrOrSrTrUrbrrrrras rac@rB)RenderingSpeedrErDrFN)rrrrErDrFrrrrrcrGrcc@s"eZdZUdZeeed<edddZee ed<edddZ ee ed<edddZ ee ed <edd dZ eeed <ed d dZe ed<eed<edddZeeed<edddZeeeed<edddZeee ed<edddZeee ed<edddZeee ed<dS)IdeogramV3EditRequestNr)zcThe image being edited (max size 10MB); only JPEG, WebP and PNG formats are supported at this time.rrIzIDetermine if MagicPrompt should be used in generating the request or not.rZzA black and white image of the same size as the image being edited (max size 10MB). Black regions in the mask should match up with the regions of the image that you would like to edit; only JPEG, WebP and PNG formats are supported at this time.maskz!The number of images to generate.r/.z.The prompt used to describe the edited result.r0rKz-Random seed. Set for reproducible generation.r3zA list of 8 character hexadecimal codes representing the style of the image. Cannot be used in conjunction with style_reference_images or style_type.rLzA set of images to use as style references (maximum total size 10MB across all style references). The images should be in JPEG, PNG or WebP format.rM%Generations with character reference are subject to the character reference pricing. A set of images to use as character references (maximum total size 10MB across all character references), currently only supports 1 character reference image. The images should be in JPEG, PNG or WebP format.character_reference_imagesOptional masks for character reference images. When provided, must match the number of character_reference_images. Each mask should be a grayscale image of the same dimensions as the corresponding character reference image. The images should be in JPEG, PNG or WebP format.character_reference_images_mask)rrrr)rr%rr rIr rZrrer/r5r0rcr3rLrr?rMrgrirrrrrdsP rdc@s\eZdZUedddgdZeeed<dZee ed<edddZ ee ed <edd dZ eeed <edd d dZ eeed<edddZeed<eed<edddgdZeeed<edddZeeed<edddZeeeed<edddZeeeed<edddZeeed<edddZeeeed <edd!dZeeeed"<dS)#IdeogramV3RequestNzAspect ratio in format WxH1x3r^r(r)z*Whether to enable magic prompt enhancementrrZz6Text prompt specifying what to avoid in the generationr,zNumber of images to generater-)rrr/.z$The text prompt for image generationr0rKzImage resolution in format WxH1280x800r1z&Seed value for reproducible generationr3z*Array of style codes in hexadecimal formatrLz,Array of reference image URLs or identifiersrMzThe type of style to applyr4rfrgrhri)rrrr r(rrrr)r\rZr`r,r/r5r0rcr1r3rLrr?rMr4rargrirrrrrjsN rjN)&enumrtypingrrrrrrpydanticr r r r rrr#r%r'r6r8r<rr?r@rArCrHrNrRrVr[r\r`rarcrdrjrrrrs:    )    *