o i3% @sNddlmZddlmZddlmZmZGdddZGdddZGdd d Z Gd d d Z Gd d d Z Gddde eZ d.d/ddZe jgde jgde jgde jgdiZGddde eZgdZgdZGdd d eZGd!d"d"eZGd#d$d$eZGd%d&d&eZGd'd(d(eZGd)d*d*eZGd+d,d,eZd-S)0) annotations)Enum) BaseModelFieldc@seZdZd ddZddZd S) RecraftColorrintgbcCs|||g|_dSN)color)selfrr r r:/mnt/c/Users/fbmor/ComfyUI/comfy_api_nodes/apis/recraft.py__init__ zRecraftColor.__init__cCs t|jdS)N)rgb)RecraftColorObjectr r rrrcreate_api_model s zRecraftColor.create_api_modelN)rrr rr r__name__ __module__ __qualname__rrrrrrrs  rc@s@eZdZddZddZdddZd d Zd d ZdddZdS)RecraftColorChaincCs g|_dSr colorsrrrrrs zRecraftColorChain.__init__cCst|jdkr |jdSdS)Nr)lenrrrrr get_firsts zRecraftColorChain.get_firstr rcCs|j|dSr )rappend)r r rrraddrzRecraftColorChain.addcCs|jsdSdd|jD}|S)NcSsg|]}|qSr)r).0xrrr sz6RecraftColorChain.create_api_model..r)r colors_apirrrrsz"RecraftColorChain.create_api_modelcCs t}|jD]}||q|Sr )rrr )r cr rrrclone"s  zRecraftColorChain.cloneothercCs"|}|jD]}||q|Sr )r&rr )r r'r%r rrrclone_and_merge(s  z!RecraftColorChain.clone_and_mergeN)r r)r'r) rrrrrr rr&r(rrrrrs rc@s$eZdZ  d dd d Zd d ZdS)RecraftControlsNrrbackground_colorartistic_levelrno_textboolcCs||_||_||_||_dSr rr*r+r,)r rr*r+r,rrrr0s zRecraftControls.__init__cCsx|jdur|jdur|jdur|jdurdSd}d}|jr"|j}|jr2|j}|r0|nd}t|||j|jdS)Nr.)rr*r+r,rrRecraftControlsObject)r r$background_color_apifirst_backgroundrrrr7s(  z RecraftControls.create_api_model)NNNN)rrr*rr+rr,r-rrrrrr)/s  r)c@seZdZdd ddZdS) RecraftStyleNstylestrsubstylestyle_idcCs"||_|dkr d}||_||_dSNNone)r3r5r6)r r3r5r6rrrrGs  zRecraftStyle.__init__)NNN)r3r4r5r4r6r4)rrrrrrrrr2Fsr2c@seZdZdZdZdZdS) RecraftIORECRAFT_V3_STYLE RECRAFT_COLORRECRAFT_CONTROLSN)rrrSTYLEV3COLORCONTROLSrrrrr9Osr9c@seZdZdZdZdZdZdS)RecraftStyleV3realistic_imagedigital_illustrationvector_illustration logo_rasterN)rrrrArBrCrDrrrrr@Us r@Tstyle_v3r4return list[str]cCs"g}|r |d|t|gSr7)rdict_recraft_substyles_v3get)rE include_none substylesrrrget_v3_substyles]s rL)b_and_w enterprise evening_lightfaded_nostalgia forest_life hard_flashhdr motion_blurmystic_naturalism natural_light natural_tones organic_calmreal_life_glow retro_realismretro_snapshotstudio_portrait urban_dramavillage_realism warm_folk)) 2d_art_poster2d_art_poster_2 antiquarian bold_fantasy child_book child_bookscover crosshatchdigital_engravingengraving_color expressionismfreehand_detailsgraingrain_20graphic_intensity hand_drawnhand_drawn_outline handmade_3d hard_comicsinfantile_sketch long_shadow modern_folk multicolor neon_calmnoirnostalgic_pasteloutline_detailspastel_gradient pastel_sketch pixel_artplasticpop_artpop_renaissanceseamless street_art tablet_sketch urban_glowurban_sketchingvanilla_dreamsyoung_adult_bookyoung_adult_book_2) bold_stroke chemistrycolored_stencilcontour_pop_artcosmicscutout depressive editorialemotional_flat engraving infographicalline_art line_circuitlinocutmarker_outlinemosaic naivector roundish_flatrsegmented_colorssharp_contrastthin vector_photo vivid_shapes)emblem_graffitiemblem_pop_art emblem_punk emblem_stampemblem_vintagec@sHeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdS)RecraftImageSize 1024x1024 1365x1024 1024x1365 1536x1024 1024x1536 1820x1024 1024x1820 1024x2048 2048x1024 1434x1024 1024x1434 1024x1280 1280x1024 1024x1707 1707x1024N)rrr res_1024x1024 res_1365x1024 res_1024x1365 res_1536x1024 res_1024x1536 res_1820x1024 res_1024x1820 res_1024x2048 res_2048x1024 res_1434x1024 res_1024x1434 res_1024x1280 res_1280x1024 res_1024x1707 res_1707x1024rrrrrs r)r1536x768768x15361280x832832x12801216x896896x12161152x896896x1152832x13441280x896896x12801344x768768x1344) 2048x2048 3072x1536 1536x3072 2560x1664 1664x2560 2432x1792 1792x2432 2304x1792 1792x2304 1664x2688rr 2560x1792 1792x2560c@"eZdZUedddZded<dS)r.zIAn array of 3 integer values in range of 0...255 defining RGB Color Model descriptionz list[int]rN)rrrrr__annotations__rrrrr rc@s^eZdZUedddZded<edddZded<edd dZd ed <edd dZd ed<dS)r/NzAn array of preferable colorsrzlist[RecraftColorObject] | Nonerz-Use given color as a desired background colorzRecraftColorObject | Noner*zDo not embed text layoutsz bool | Noner,zDefines artistic tone of your image. At a simple level, the person looks straight at the camera in a static and clean style. Dynamic and eccentric levels introduce movement and creativity. The value should be in range [0..5]. int | Noner+) rrrrrrr*r,r+rrrrr/ r/c@seZdZUedddZded<edddZded <edd dZd ed <edd dZded<edZ ded<edddZ ded<edddZ ded<edddZ ded<edddZ ded<edddZded<edddZded<dS)RecraftImageGenerationRequest.z0The text prompt describing the image to generaterr4promptNz3The size of the generated image (e.g., "1024x1024")z str | Nonesizez The number of images to generaternz4A text description of undesired elements on an imagenegative_promptmodelzHThe style to apply to the generated image (e.g., "digital_illustration")r3zJThe substyle to apply to the generated image, depending on the style inputr5z6A set of custom parameters to tweak generation processzRecraftControlsObject | Nonecontrolsz4Use a previously uploaded style as a reference; UUIDr6zDefines the difference with the original image, should lie in [0, 1], where 0 means almost identical, and 1 means miserable similarityz float | NonestrengthzSeed for video generationr random_seed)rrrrrrrrrrr3r5rr6rrrrrrr s rc@s6eZdZUedddZded<edddZded<dS) RecraftReturnedObject.z)Unique identifier for the generated imagerr4image_idz!URL to access the generated imageurlN)rrrrrrrrrrrrs rc@s^eZdZUedddZded<edddZded<edd dZd ed <edd dZd ed<dS)RecraftImageGenerationResponse.z.Unix timestamp when the generation was createdrrcreatedz)Number of credits used for the generationcreditsNz$Array of generated image informationz"list[RecraftReturnedObject] | NonedatazSingle generated imagezRecraftReturnedObject | Noneimage) rrrrrrrrrrrrrrrrc@r)RecraftCreateStyleRequest.zCrealistic_image, digital_illustration, vector_illustration, or iconrr4r3N)rrrrr3rrrrrr#rrc@r)RecraftCreateStyleResponse.zUUID of the created stylerr4idN)rrrrrrrrrrr'rrN)T)rEr4rFrG) __future__renumrpydanticrrrrr)r2r9r4r@rLrArBrCrDrHrRECRAFT_V4_SIZESRECRAFT_V4_PRO_SIZESrr/rrrrrrrrrs2      + e