Spaces:
Running
on
Zero
Running
on
Zero
| from __future__ import annotations | |
| from enum import Enum | |
| from typing import Optional | |
| from pydantic import BaseModel, Field, conint, confloat | |
| class RecraftColor: | |
| def __init__(self, r: int, g: int, b: int): | |
| self.color = [r, g, b] | |
| def create_api_model(self): | |
| return RecraftColorObject(rgb=self.color) | |
| class RecraftColorChain: | |
| def __init__(self): | |
| self.colors: list[RecraftColor] = [] | |
| def get_first(self): | |
| if len(self.colors) > 0: | |
| return self.colors[0] | |
| return None | |
| def add(self, color: RecraftColor): | |
| self.colors.append(color) | |
| def create_api_model(self): | |
| if not self.colors: | |
| return None | |
| colors_api = [x.create_api_model() for x in self.colors] | |
| return colors_api | |
| def clone(self): | |
| c = RecraftColorChain() | |
| for color in self.colors: | |
| c.add(color) | |
| return c | |
| def clone_and_merge(self, other: RecraftColorChain): | |
| c = self.clone() | |
| for color in other.colors: | |
| c.add(color) | |
| return c | |
| class RecraftControls: | |
| def __init__(self, colors: RecraftColorChain=None, background_color: RecraftColorChain=None, | |
| artistic_level: int=None, no_text: bool=None): | |
| self.colors = colors | |
| self.background_color = background_color | |
| self.artistic_level = artistic_level | |
| self.no_text = no_text | |
| def create_api_model(self): | |
| if self.colors is None and self.background_color is None and self.artistic_level is None and self.no_text is None: | |
| return None | |
| colors_api = None | |
| background_color_api = None | |
| if self.colors: | |
| colors_api = self.colors.create_api_model() | |
| if self.background_color: | |
| first_background = self.background_color.get_first() | |
| background_color_api = first_background.create_api_model() if first_background else None | |
| return RecraftControlsObject(colors=colors_api, background_color=background_color_api, | |
| artistic_level=self.artistic_level, no_text=self.no_text) | |
| class RecraftStyle: | |
| def __init__(self, style: str=None, substyle: str=None, style_id: str=None): | |
| self.style = style | |
| if substyle == "None": | |
| substyle = None | |
| self.substyle = substyle | |
| self.style_id = style_id | |
| class RecraftIO: | |
| STYLEV3 = "RECRAFT_V3_STYLE" | |
| COLOR = "RECRAFT_COLOR" | |
| CONTROLS = "RECRAFT_CONTROLS" | |
| class RecraftStyleV3(str, Enum): | |
| #any = 'any' NOTE: this does not work for some reason... why? | |
| realistic_image = 'realistic_image' | |
| digital_illustration = 'digital_illustration' | |
| vector_illustration = 'vector_illustration' | |
| logo_raster = 'logo_raster' | |
| def get_v3_substyles(style_v3: str, include_none=True) -> list[str]: | |
| substyles: list[str] = [] | |
| if include_none: | |
| substyles.append("None") | |
| return substyles + dict_recraft_substyles_v3.get(style_v3, []) | |
| dict_recraft_substyles_v3 = { | |
| RecraftStyleV3.realistic_image: [ | |
| "b_and_w", | |
| "enterprise", | |
| "evening_light", | |
| "faded_nostalgia", | |
| "forest_life", | |
| "hard_flash", | |
| "hdr", | |
| "motion_blur", | |
| "mystic_naturalism", | |
| "natural_light", | |
| "natural_tones", | |
| "organic_calm", | |
| "real_life_glow", | |
| "retro_realism", | |
| "retro_snapshot", | |
| "studio_portrait", | |
| "urban_drama", | |
| "village_realism", | |
| "warm_folk" | |
| ], | |
| RecraftStyleV3.digital_illustration: [ | |
| "2d_art_poster", | |
| "2d_art_poster_2", | |
| "antiquarian", | |
| "bold_fantasy", | |
| "child_book", | |
| "child_books", | |
| "cover", | |
| "crosshatch", | |
| "digital_engraving", | |
| "engraving_color", | |
| "expressionism", | |
| "freehand_details", | |
| "grain", | |
| "grain_20", | |
| "graphic_intensity", | |
| "hand_drawn", | |
| "hand_drawn_outline", | |
| "handmade_3d", | |
| "hard_comics", | |
| "infantile_sketch", | |
| "long_shadow", | |
| "modern_folk", | |
| "multicolor", | |
| "neon_calm", | |
| "noir", | |
| "nostalgic_pastel", | |
| "outline_details", | |
| "pastel_gradient", | |
| "pastel_sketch", | |
| "pixel_art", | |
| "plastic", | |
| "pop_art", | |
| "pop_renaissance", | |
| "seamless", | |
| "street_art", | |
| "tablet_sketch", | |
| "urban_glow", | |
| "urban_sketching", | |
| "vanilla_dreams", | |
| "young_adult_book", | |
| "young_adult_book_2" | |
| ], | |
| RecraftStyleV3.vector_illustration: [ | |
| "bold_stroke", | |
| "chemistry", | |
| "colored_stencil", | |
| "contour_pop_art", | |
| "cosmics", | |
| "cutout", | |
| "depressive", | |
| "editorial", | |
| "emotional_flat", | |
| "engraving", | |
| "infographical", | |
| "line_art", | |
| "line_circuit", | |
| "linocut", | |
| "marker_outline", | |
| "mosaic", | |
| "naivector", | |
| "roundish_flat", | |
| "seamless", | |
| "segmented_colors", | |
| "sharp_contrast", | |
| "thin", | |
| "vector_photo", | |
| "vivid_shapes" | |
| ], | |
| RecraftStyleV3.logo_raster: [ | |
| "emblem_graffiti", | |
| "emblem_pop_art", | |
| "emblem_punk", | |
| "emblem_stamp", | |
| "emblem_vintage" | |
| ], | |
| } | |
| class RecraftModel(str, Enum): | |
| recraftv3 = 'recraftv3' | |
| recraftv2 = 'recraftv2' | |
| class RecraftImageSize(str, Enum): | |
| res_1024x1024 = '1024x1024' | |
| res_1365x1024 = '1365x1024' | |
| res_1024x1365 = '1024x1365' | |
| res_1536x1024 = '1536x1024' | |
| res_1024x1536 = '1024x1536' | |
| res_1820x1024 = '1820x1024' | |
| res_1024x1820 = '1024x1820' | |
| res_1024x2048 = '1024x2048' | |
| res_2048x1024 = '2048x1024' | |
| res_1434x1024 = '1434x1024' | |
| res_1024x1434 = '1024x1434' | |
| res_1024x1280 = '1024x1280' | |
| res_1280x1024 = '1280x1024' | |
| res_1024x1707 = '1024x1707' | |
| res_1707x1024 = '1707x1024' | |
| class RecraftColorObject(BaseModel): | |
| rgb: list[int] = Field(..., description='An array of 3 integer values in range of 0...255 defining RGB Color Model') | |
| class RecraftControlsObject(BaseModel): | |
| colors: Optional[list[RecraftColorObject]] = Field(None, description='An array of preferable colors') | |
| background_color: Optional[RecraftColorObject] = Field(None, description='Use given color as a desired background color') | |
| no_text: Optional[bool] = Field(None, description='Do not embed text layouts') | |
| artistic_level: Optional[conint(ge=0, le=5)] = Field(None, description='Defines 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].') | |
| class RecraftImageGenerationRequest(BaseModel): | |
| prompt: str = Field(..., description='The text prompt describing the image to generate') | |
| size: Optional[RecraftImageSize] = Field(None, description='The size of the generated image (e.g., "1024x1024")') | |
| n: conint(ge=1, le=6) = Field(..., description='The number of images to generate') | |
| negative_prompt: Optional[str] = Field(None, description='A text description of undesired elements on an image') | |
| model: Optional[RecraftModel] = Field(RecraftModel.recraftv3, description='The model to use for generation (e.g., "recraftv3")') | |
| style: Optional[str] = Field(None, description='The style to apply to the generated image (e.g., "digital_illustration")') | |
| substyle: Optional[str] = Field(None, description='The substyle to apply to the generated image, depending on the style input') | |
| controls: Optional[RecraftControlsObject] = Field(None, description='A set of custom parameters to tweak generation process') | |
| style_id: Optional[str] = Field(None, description='Use a previously uploaded style as a reference; UUID') | |
| strength: Optional[confloat(ge=0.0, le=1.0)] = Field(None, description='Defines the difference with the original image, should lie in [0, 1], where 0 means almost identical, and 1 means miserable similarity') | |
| random_seed: Optional[int] = Field(None, description="Seed for video generation") | |
| # text_layout | |
| class RecraftReturnedObject(BaseModel): | |
| image_id: str = Field(..., description='Unique identifier for the generated image') | |
| url: str = Field(..., description='URL to access the generated image') | |
| class RecraftImageGenerationResponse(BaseModel): | |
| created: int = Field(..., description='Unix timestamp when the generation was created') | |
| credits: int = Field(..., description='Number of credits used for the generation') | |
| data: Optional[list[RecraftReturnedObject]] = Field(None, description='Array of generated image information') | |
| image: Optional[RecraftReturnedObject] = Field(None, description='Single generated image') | |