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 module targets the /v2/image/create and /v2/image/edit endpoints. Instead of embedding image references in free text, a v2 request carries a structured Description (a layout of labelled, bounded regions) and a list of Reference images. The response can also echo the layout the model actually generated.

from reve.v2 import create, Bbox, Description, ImageInput, Reference, Region

result = create(
    instruction="A dog on the left and a cat on the right",
    description=Description(
        prompt="two pets",
        regions=[
            Region(label="dog", prompt="a happy dog", bbox=Bbox(0.0, 0.0, 0.5, 1.0)),
            Region(label="cat", prompt="a sleepy cat", bbox=Bbox(0.5, 0.0, 1.0, 1.0)),
        ],
    ),
    references=[Reference(image=ImageInput(ref="reference:@mypet"), prompt="my pet")],
    aspect_ratio="16:9",
)
result.save("pets.png")
print(result.description)  # the layout the model generated
from reve.v2 import edit, Bbox, Description, Region

result = edit(
    instruction="Make the sky stormy",
    image="original.jpg",  # path, bytes, PIL Image, or ImageInput
    new_description=Description(
        regions=[Region(label="sky", prompt="dark storm clouds", bbox=Bbox(0, 0, 1, 0.5))],
    ),
)

Input types (reve.v2)

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, preserve?, image_index?, image_region_index?.
Description regions: list[Region], prompt?.
Reference image: ImageInput, prompt?.

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.

create(instruction, *, description, references, aspect_ratio, postprocessing, version, ...)

edit(instruction, image, *, references, old_description, new_description, aspect_ratio, postprocessing, version, ...)

Both return a V2ImageResponse:

Field Type Description
image PIL.Image.Image The generated image.
image_bytes bytes Raw bytes of the generated image.
description Description | 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.

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.4.tar.gz (31.1 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.4-py3-none-any.whl (26.1 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: reve-0.1.4.tar.gz
  • Upload date:
  • Size: 31.1 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.4.tar.gz
Algorithm Hash digest
SHA256 6ee71c9aa39a080d78059bc96d3dae9b00e172f7e24ce2181718b0a8ac62f552
MD5 3571862eec3c7597be437fd958008f79
BLAKE2b-256 1c5a13950bc018fa73a5a1c2d749c183cb900cd734e1d180e73fe3081f33044a

See more details on using hashes here.

File details

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

File metadata

  • Download URL: reve-0.1.4-py3-none-any.whl
  • Upload date:
  • Size: 26.1 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.4-py3-none-any.whl
Algorithm Hash digest
SHA256 5ef6e8cd8d7814e032f21d87c8aadaa6422ce51e5408d7e19b57b77a327e3cd4
MD5 78679c48e3e05b5c5fc550c7b1fa08ff
BLAKE2b-256 f8be0d32b74a59639d1505f1872df807b2d88f94e1f5dd31eab01428a839e8ce

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