/*
 * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
 */

import {
  CannyEdgeDetectionAcceptEnum,
  imagesCannyEdgeDetection,
} from "../funcs/imagesCannyEdgeDetection.js";
import { imagesGenerateFromControlNet } from "../funcs/imagesGenerateFromControlNet.js";
import {
  GenerateFromImageAcceptEnum,
  imagesGenerateFromImage,
} from "../funcs/imagesGenerateFromImage.js";
import {
  GenerateFromPromptAcceptEnum,
  imagesGenerateFromPrompt,
} from "../funcs/imagesGenerateFromPrompt.js";
import {
  GenerateQRCodeAcceptEnum,
  imagesGenerateQRCode,
} from "../funcs/imagesGenerateQRCode.js";
import { ClientSDK, RequestOptions } from "../lib/sdks.js";
import * as models from "../models/index.js";
import { unwrapAsync } from "../types/fp.js";

export { GenerateFromPromptAcceptEnum } from "../funcs/imagesGenerateFromPrompt.js";

export { GenerateFromImageAcceptEnum } from "../funcs/imagesGenerateFromImage.js";

export { CannyEdgeDetectionAcceptEnum } from "../funcs/imagesCannyEdgeDetection.js";

export { GenerateQRCodeAcceptEnum } from "../funcs/imagesGenerateQRCode.js";

export class Images extends ClientSDK {
  /**
   * Generate a new image from a text prompt
   */
  async generateFromPrompt(
    request: models.GenerateImageFromPromptRequest,
    options?: RequestOptions & {
      acceptHeaderOverride?: GenerateFromPromptAcceptEnum;
    },
  ): Promise<models.GenerateImageFromPromptResponse> {
    return unwrapAsync(imagesGenerateFromPrompt(
      this,
      request,
      options,
    ));
  }

  /**
   * Generate a new image from an image
   */
  async generateFromImage(
    request: models.GenerateImageFromImageRequest,
    options?: RequestOptions & {
      acceptHeaderOverride?: GenerateFromImageAcceptEnum;
    },
  ): Promise<models.GenerateImageFromImageResponse> {
    return unwrapAsync(imagesGenerateFromImage(
      this,
      request,
      options,
    ));
  }

  /**
   * Pre-process the image with canny edge detection
   */
  async cannyEdgeDetection(
    request: models.CannyEdgeDetectionRequest,
    options?: RequestOptions & {
      acceptHeaderOverride?: CannyEdgeDetectionAcceptEnum;
    },
  ): Promise<models.CannyEdgeDetectionResponse> {
    return unwrapAsync(imagesCannyEdgeDetection(
      this,
      request,
      options,
    ));
  }

  /**
   * Generate a new image using ControlNet with provided image as a guidance
   */
  async generateFromControlNet(
    request: models.GenerateFromControlNetRequest,
    options?: RequestOptions,
  ): Promise<ReadableStream<Uint8Array>> {
    return unwrapAsync(imagesGenerateFromControlNet(
      this,
      request,
      options,
    ));
  }

  /**
   * Generate a QR code
   */
  async generateQRCode(
    request: models.GenerateQRCodeRequest,
    options?: RequestOptions & {
      acceptHeaderOverride?: GenerateQRCodeAcceptEnum;
    },
  ): Promise<models.GenerateQRCodeResponse> {
    return unwrapAsync(imagesGenerateQRCode(
      this,
      request,
      options,
    ));
  }
}