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_MONOLITH_LOCATION 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.

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-ShareAlike 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.0.tar.gz (15.5 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.0-py3-none-any.whl (12.3 kB view details)

Uploaded Python 3

File details

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

File metadata

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

File hashes

Hashes for reve-0.1.0.tar.gz
Algorithm Hash digest
SHA256 420d8dab1deecf341f76a485173e339e20599b1ae24571b37a8322a67e53eb41
MD5 f5d3203758a5f3ec1573b50382d080e5
BLAKE2b-256 ab02e4e0e92eaf5bd8ab1fd52930737097fa123039ec63109899a60507058a6c

See more details on using hashes here.

File details

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

File metadata

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

File hashes

Hashes for reve-0.1.0-py3-none-any.whl
Algorithm Hash digest
SHA256 236661f5bb52059dc6770c9bb0151252a77a784f145e693bec49b747c7bb7850
MD5 895f07b72978cbe05de4e950fa4644e7
BLAKE2b-256 96c006cd483e3637ac327fa7c540393d96d08b6fdac05c9d5349847fd14e73a1

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