Simple Image Gen Example

Below a basic example to call AKI.IO Image Gen endpoints using the aki-io interface pip.

The example includes a callback to show the progress of the image generation.

For decoding the generate image data and the file format the Aki.decode_binary(…) function is used.

from aki_io import Aki

aki = Aki(
    'z_image_turbo',
    'fc3a8c50-b12b-4d6a-ba07-c9f6a6c32c37',
)

def progress_callback(progress, progress_data):
    print(f'Progress: {progress.get("progress", 0)}%')

result = aki.do_api_request(
    {
        'prompt': 'Astronaut on Mars holding a banner which states "AKI is happy to serve your model" during sunset sitting on a giant yellow rubber duck',
        'seed': -1,
        'height': 512,
        'width': 512,
    },
    progress_callback, # optional
)

if result['success']:
    # Save images to files
    images = result.get('images')
    for idx, image_data in enumerate(images):
        media_format, image_binary = Aki.decode_binary(image_data)
        output_file = f'image_{idx}.{media_format}'
        with open(output_file, 'wb') as f:
            f.write(image_binary)
        print(f'Output image saved at {output_file}')
else:
    print("API Error:", result.get('error_code'), "-", result.get('error'))

Image Gen Python Examples

Simple Image Gen Example