Image Generation

ARK Platform Example / Image Generation

  • Copy
    
    # NOTE: This code will output a "knight_in_space.png" file with the generated image.
    
    import openai
    import base64
    from PIL import Image
    from io import BytesIO
    
    ark_base_url = "https://api.ark-labs.cloud/api/v1"
    ark_api_key = "API_KEY"
    
    client = openai.OpenAI(api_key=ark_api_key, base_url=ark_base_url)
    
    print("Generating image...")
    
    response = client.images.generate(
        model="gpt-image-1",
        prompt="A brave knight in a futuristic space suit floating near a small rocket, lost in deep space, photorealistic CGI.",
        n=1,
        size="1024x1024",
        response_format="b64_json",
        extra_body={"steps": 10}
    )
    
    image_b64 = response.data[0].b64_json
    image_bytes = base64.b64decode(image_b64)
    image = Image.open(BytesIO(image_bytes))
    image.save("knight_in_space.png")
    
    print("Image saved as knight_in_space.png")