Background Removal API

/**
 * BackgroundRemovalAPI - generated by OmniStream from Background Removal API's OpenAPI spec.
 * One typed method per endpoint. A single Omni key reaches the API.
 */
const DEFAULT_BASE = "https://grid.skinvaults.online/v1/proxy/remove";

export class BackgroundRemovalAPIError extends Error {
  status: number;
  code?: string;
  constructor(status: number, code: string | undefined, message?: string) {
    super(message || `BackgroundRemovalAPI error ${status}`);
    this.name = "BackgroundRemovalAPIError";
    this.status = status;
    this.code = code;
  }
}

export interface BackgroundRemovalAPIOptions {
  baseUrl?: string;
  fetch?: typeof fetch;
  timeoutMs?: number;
}

export class BackgroundRemovalAPI {
  /** @param token Your OmniStream key (one key for every API). */
  constructor(private token: string, private opts: BackgroundRemovalAPIOptions = {}) {
    if (!token) throw new Error("BackgroundRemovalAPI: token is required");
  }
  rateLimit: { remainingMinute?: number; remainingDay?: number } | null = null;

  private async _request(method: string, path: string, { params, body }: { params?: Record<string, unknown>; body?: unknown } = {}): Promise<any> {
    const f = this.opts.fetch ?? globalThis.fetch;
    const url = new URL((this.opts.baseUrl ?? DEFAULT_BASE).replace(/\/+$/, "") + path);
    if (params) for (const [k, v] of Object.entries(params)) if (v != null) url.searchParams.set(k, String(v));
    const headers: Record<string, string> = { "x-omni-key": this.token };
    if (body !== undefined) headers["content-type"] = "application/json";
    const res = await f(url, { method, headers, body: body !== undefined ? JSON.stringify(body) : undefined });
    this.rateLimit = {
      remainingMinute: Number(res.headers.get("x-ratelimit-remaining-minute")) || undefined,
      remainingDay: Number(res.headers.get("x-ratelimit-remaining-day")) || undefined,
    };
    const data = await res.json().catch(() => ({}));
    if (!res.ok) throw new BackgroundRemovalAPIError(res.status, data?.error?.code, data?.error?.message);
    return data.data ?? data;
  }

  /** Fetch credit balance and free API calls. */
  getaccount(): Promise<any> {
    return this._request("GET", "/account", {});
  }

  /** Submit an image to the remove.bg Improvement program
* Contribute an image that remove.bg is currently not able to remove the background from properly
* Help us make remove.bg better
* Get better results for similiar images in the future

Notes:
  * By submitting images through the API you agree to the <a target="_blank" rel="noopener" href="/ipc">Improvement Program Conditions</a>
  * File size: up to 12MB
  * up to 100 files per day. <br> Higher Rate Limits are available for Enterprise customers <a href="/support/contact?subject=Improvement+Program+Rate+Limit">upon request</a>.

Requires either an API Key to be provided in the `X-API-Key` request header or an OAuth 2.0 access token to be provided in the `Authorization` request header.

Please note that submissions are used on a best-effort basis and the extent of expected improvement varies depending on many factors, including the number of provided images, their complexity and visual similarity. Improvements usually take several weeks to become effective.
 */
  postimprove(body: unknown): Promise<any> {
    return this._request("POST", "/improve", { body });
  }

  /** Remove the background of an image */
  postremovebg(body: unknown): Promise<any> {
    return this._request("POST", "/removebg", { body });
  }
}

export default BackgroundRemovalAPI;