Skip to main content

Python SDK for the Reve image generation API

Project description

Reve Python SDK

A Pythonic interface to the Reve image-generation API. Generate, remix, and edit images with a handful of function calls.

Installation

From PyPI:

pip install reve

From source:

git clone https://github.com/reve-ai/reve-core.git
cd reve-core/sdk/python
pip install -e .

Quick Start

from reve.v1.image import create

img = create(prompt="A beautiful sunset over the ocean")
img.save("sunset.jpg")
print(img.credits_remaining)

Set the REVE_API_TOKEN environment variable before running, or pass api_token= directly to any function.

Authentication

The SDK reads credentials from environment variables by default:

Variable Description Default
REVE_API_TOKEN Bearer token (required)
REVE_API_HOST API base URL https://api.reve.com
REVE_PROXY_AUTHORIZATION Proxy-authorization header
export REVE_API_TOKEN="papi.your-token-here"

You can also pass them per-call:

img = create(
    prompt="A sunset",
    api_token="papi.your-token-here",
    api_url="https://custom-endpoint.example.com",
)

API Reference

All image functions live in reve.v1.image.

create(prompt, *, aspect_ratio, version, test_time_scaling, postprocessing, ...)

Generate an image from a text prompt.

from reve.v1.image import create
from reve.v1.postprocessing import upscale, remove_background

img = create(
    prompt="A red dragon flying over mountains",
    aspect_ratio="16:9",
    version="latest",
    test_time_scaling=3,
    postprocessing=[upscale(factor=2), remove_background()],
)
img.save("dragon.png")
Parameter Type Description
prompt str Text description of the image (positional).
aspect_ratio str | None One of "16:9", "3:2", "4:3", "1:1", "3:4", "2:3", "9:16", "auto". Default "auto".
version str | None Model version identifier or "latest".
test_time_scaling int | None Quality factor 1–5. Higher = better quality, more credits.
postprocessing list[dict] | None Postprocessing pipeline (see below).

remix(prompt, reference_images, *, ...)

Remix reference images into a new image guided by a prompt. Use <ref>0</ref>, <ref>1</ref>, … to refer to each reference image.

from reve.v1.image import remix

img = remix(
    prompt="The subject from <ref>0</ref> standing in a magical forest",
    reference_images=["photo.jpg"],
    aspect_ratio="1:1",
)
Parameter Type Description
prompt str Text prompt with optional <ref>N</ref> tags (positional).
reference_images Sequence[str | bytes | PIL.Image] Reference images — file paths, raw bytes, or PIL Images (positional).
aspect_ratio str | None Aspect ratio (see create).
version str | None Model version.
test_time_scaling int | None Quality factor 1–5.
postprocessing list[dict] | None Postprocessing pipeline.

edit(edit_instruction, reference_image, *, ...)

Edit an existing image with a natural-language instruction.

from reve.v1.image import edit

img = edit(
    edit_instruction="Make the sky more dramatic with storm clouds",
    reference_image="original.jpg",
)
Parameter Type Description
edit_instruction str Description of the edit (positional).
reference_image str | bytes | PIL.Image Source image (positional).
aspect_ratio str | None Aspect ratio (see create).
version str | None Model version.
test_time_scaling int | None Quality factor 1–5.
postprocessing list[dict] | None Postprocessing pipeline.

get_balance(*, ...)

Return the current credit balance.

from reve.v1.image import get_balance

balance = get_balance()
print(balance)  # {"budget_id": "abc123", "new_balance": 500}

Returns a dict with keys budget_id (str) and new_balance (number).

list_effects(source=None, *, ...)

List available effects for postprocessing.

from reve.v1.image import list_effects

effects = list_effects(source="preset")
for e in effects:
    print(e["name"], "-", e["description"])
Parameter Type Description
source str | None Filter by source: "all", "project", or "preset".

Returns a list of dicts with name, description, source, and category keys.

v2 Layout-Aware API

The reve.v2.image module targets the /v2/image endpoints. The layout-producing endpoints (create_layout, image_to_layout, render) work with a structured Layout (a list of labelled, bounded Regions); the image-producing create and edit endpoints take a text instruction and optional reference images. Image-producing calls can also echo the layout the model generated.

Functions live in reve.v2.image; the data structures live in reve.v2.types (and are also re-exported from reve.v2).

from reve.v2.image import create
from reve.v2.types import ImageInput

result = create(
    instruction="A dog on the left and a cat on the right",
    references=[ImageInput(ref="reference:@mypet")],  # optional, each an image
    aspect_ratio="16:9",
)
result.save("pets.png")
print(result.layout)  # the layout the model generated
from reve.v2.image import edit
from reve.v2.types import ImageInput

result = edit(
    instruction="Make the sky stormy",
    image="original.jpg",  # path, bytes, PIL Image, or ImageInput
    references=[ImageInput(ref="reference:@mystyle")],  # optional, each an image
)

Functions (reve.v2.image)

There are two families. Image-producing functions return a V2ImageResponse; layout-producing functions return a V2LayoutResponse and produce no image.

Function Kind Description
create(instruction, *, references?, aspect_ratio?, postprocessing?, version?) image Generate an image (text + optional references).
edit(instruction, image, *, references?, aspect_ratio?, postprocessing?, version?) image Edit an image with a text instruction.
render(layout, *, references?, postprocessing?, version?) image Render an image from a layout (layout2image).
image_to_layout(image, *, version?) layout Derive a layout from an image (image2layout).
create_layout(prompt, *, references?, aspect_ratio?, version?) layout Generate a layout from text and/or references.

aspect_ratio is one of 4:1, 3:1, 21:9, 2:1, 17:9, 16:9, 3:2, 4:3, 1:1, 3:4, 2:3, 9:16, 1:2, 1:3, 1:4, or auto (default).

Input types (reve.v2.types)

Type Fields
ImageInput data (path/bytes/PIL, base-64 in JSON) or ref (id:<uuid> / reference:@<name>).
Bbox x0, y0, x1, y1 — normalized to [0, 1], top-left origin.
Region label, prompt, bbox, image_index?, image_region_index?, parent?, region_type?.
Layout regions: list[Region], prompt?, normalized_edit_instruction?.
Reference image: ImageInput?, prompt?, layout? — an image and/or a layout.

region_type is a level-of-detail / special-handling hint, one of: coarse_detail (a high-level object such as a person or car), medium_detail (a medium-level object such as an arm or belt, whose parent is a coarse_detail), fine_detail (a fine detail such as a ring or buckle, whose parent is a medium_detail), text (embedded text), hand, or face.

The ref form of ImageInput points at an image that already exists in the project your API key belongs to:

  • id:<uuid> — the ID of an image or generation in the project (for example, one created in the Reve app). A generation ID resolves to that generation's output image.
  • reference:@<name> — the name of a reference entity defined in the project in the Reve app.

Response types

create, edit, and render return a V2ImageResponse:

Field Type Description
image PIL.Image.Image | None The generated image.
image_bytes bytes Raw bytes of the generated image.
layout Layout | None The layout the model generated.
request_id str | None Unique request identifier.
credits_used int | None Credits consumed by this request.
credits_remaining int | None Credits remaining in the budget.
version str | None Model version used.
content_violation bool Whether a content violation was flagged.

image_to_layout and create_layout return a V2LayoutResponse with the same fields minus image and image_bytes.

Postprocessing

Build postprocessing pipelines with helpers from reve.v1.postprocessing:

from reve.v1.postprocessing import upscale, remove_background, fit_image, effect
Helper Description
upscale(factor=2) Upscale the image by the given factor.
remove_background() Remove the background (produces transparent PNG).
fit_image(max_width=None, max_height=None, max_dim=None) Constrain dimensions (pixels, 1–1024).
effect(name, parameters=None) Apply a named effect. Use list_effects() for available names.

Pass them as a list to the postprocessing parameter:

img = create(
    prompt="A cat astronaut",
    postprocessing=[upscale(factor=2), remove_background()],
)

Response Object

create(), remix(), and edit() return an ImageResponse (a Pydantic BaseModel):

Field Type Description
image PIL.Image.Image The generated image.
request_id str | None Unique request identifier.
credits_used int | None Credits consumed by this request.
credits_remaining int | None Credits remaining in the budget.
version str | None Model version used.
content_violation bool Whether a content violation was flagged.
img = create(prompt="A sunset")
img.image              # PIL.Image.Image
img.request_id         # "req_abc123"
img.credits_used       # 10
img.credits_remaining  # 490
img.version            # "v1.2"
img.save("out.jpg")    # delegates to PIL.Image.save()

Error Handling

All exceptions inherit from ReveAPIError:

ReveAPIError                  # Base — any API error
├── ReveAuthenticationError   # HTTP 401 — bad or missing token
├── ReveBudgetExhaustedError  # HTTP 402 — out of credits
├── ReveRateLimitError        # HTTP 429 — rate limited (has .retry_after)
├── ReveValidationError       # HTTP 400 — invalid parameters
└── ReveContentViolationError # Content policy violation
from reve.exceptions import ReveAPIError, ReveRateLimitError
from reve.v1.image import create

try:
    img = create(prompt="A sunset")
except ReveRateLimitError as exc:
    print(f"Rate limited — retry after {exc.retry_after}s")
except ReveAPIError as exc:
    print(f"API error (status {exc.status_code}): {exc.message}")

Examples

Working example scripts are in the examples/ directory:

Development

Install development dependencies:

pip install -e ".[dev]"

Run the test suite:

pytest

License

This SDK is released under the Creative Commons Attribution 4.0 International License.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

reve-0.1.6.tar.gz (36.0 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

reve-0.1.6-py3-none-any.whl (29.4 kB view details)

Uploaded Python 3

File details

Details for the file reve-0.1.6.tar.gz.

File metadata

  • Download URL: reve-0.1.6.tar.gz
  • Upload date:
  • Size: 36.0 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.11.14

File hashes

Hashes for reve-0.1.6.tar.gz
Algorithm Hash digest
SHA256 cf6aecec89fce0675a7d6e22b09978c7ff56fdefd1daa57426d59e553e5035ab
MD5 0079e701017b562fc6e3df9140bb373a
BLAKE2b-256 243a726497c42b7598811831044a706ba5d92f2d2faf5a1bad982c1e93b9e00e

See more details on using hashes here.

File details

Details for the file reve-0.1.6-py3-none-any.whl.

File metadata

  • Download URL: reve-0.1.6-py3-none-any.whl
  • Upload date:
  • Size: 29.4 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.11.14

File hashes

Hashes for reve-0.1.6-py3-none-any.whl
Algorithm Hash digest
SHA256 cb9dd67bb7cf0a8e9ff31dc549fef19806149b3dc592f722c28b2c8177bbfd8e
MD5 2cc0ab3c0159d0e7d22e605b5762713c
BLAKE2b-256 4419e8147718c78a3dfe24d97bddea4e384a9f2dc167a59685c978bbfdd98665

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page