Skip to content

Image generation

IMAGE_GEN_COSTS = {'dalle_3': {'standard': {'square': 0.04, 'tall_or_wide': 0.08}, 'hd': {'square': 0.08, 'tall_or_wide': 0.12}}} module-attribute

client = OpenAI(api_key=settings.OPENAI_API_KEY) module-attribute

logger = logging.getLogger(__name__) module-attribute

ImageAspectRatio

Bases: StrEnum

SQUARE = 'square' class-attribute instance-attribute

TALL = 'tall' class-attribute instance-attribute

WIDE = 'wide' class-attribute instance-attribute

ImageGenState

Bases: AllLoaders, MyState

aspect_ratio = ImageAspectRatio.SQUARE class-attribute instance-attribute

download_name = 'generated_image.png' class-attribute instance-attribute

image_data = '' class-attribute instance-attribute

model = ImageModel.DALLE_3 class-attribute instance-attribute

prompt = '' class-attribute instance-attribute

quality = ImageQuality.STANDARD class-attribute instance-attribute

revised_prompt = '' class-attribute instance-attribute

user_credits = 0 class-attribute instance-attribute

credit_cost()

download()

generate_image() async

image_url()

improve_prompt() async

load_user_info() async

set_aspect_ratio(aspect_ratio)

set_model(model)

set_prompt(prompt)

set_quality(quality)

ImageModel

Bases: StrEnum

DALLE_3 = 'dalle_3' class-attribute instance-attribute

ImageQuality

Bases: StrEnum

HD = 'hd' class-attribute instance-attribute

STANDARD = 'standard' class-attribute instance-attribute

compress_and_convert_image(input_base64, output_format='WEBP', quality=80)

index()

render_controls()

render_image(url)

render_image_with_download(url)

render_select(label, default, options, on_change)

skeleton_image()