/// <reference types="node" />
import { NovelAI } from "../NovelAI";
import { ImageRequest } from "../Structures/ImageRequest";
import { SuggestedTag } from "../Structures/SuggestedTag";
export declare class ImageManager {
    private ai;
    constructor(ai: NovelAI);
    getSuggestedTags(options: {
        model: "nai-diffusion" | "safe-diffusion" | "nai-diffusuin-furry";
        prompt: string;
    }): Promise<SuggestedTag[]>;
    generate(options: Omit<ImageRequest, 'action'> & {
        parameters?: {
            n_samples: 1 | undefined;
        };
    }): Promise<Buffer>;
    generate(options: Omit<ImageRequest, 'action'> & {
        parameters?: {
            n_samples: number;
        };
    }): Promise<Buffer[]>;
    generate(options: Omit<ImageRequest, 'action'>): Promise<Buffer | Buffer[]>;
    enhance(options: Omit<ImageRequest, 'action'> & {
        parameters: {
            add_original_image?: boolean;
            controlnet_strength?: number;
            dynamic_thresholding?: boolean;
            extra_noise_seed?: number;
            image: Buffer;
            legacy?: boolean;
            noise?: number;
            sm?: boolean;
            sm_dyn?: boolean;
            strength?: number;
            n_samples: number;
        };
    }): Promise<Buffer[]>;
    enhance(options: Omit<ImageRequest, 'action'> & {
        parameters: {
            add_original_image?: boolean;
            controlnet_strength?: number;
            dynamic_thresholding?: boolean;
            extra_noise_seed?: number;
            image: Buffer;
            legacy?: boolean;
            noise?: number;
            sm?: boolean;
            sm_dyn?: boolean;
            strength?: number;
            n_samples: 1 | undefined;
        };
    }): Promise<Buffer>;
    enhance(options: Omit<ImageRequest, 'action'> & {
        parameters: {
            add_original_image?: boolean;
            controlnet_strength?: number;
            dynamic_thresholding?: boolean;
            extra_noise_seed?: number;
            image: Buffer;
            legacy?: boolean;
            noise?: number;
            sm?: boolean;
            sm_dyn?: boolean;
            strength?: number;
        };
    }): Promise<Buffer | Buffer[]>;
    generateInFill(options: Parameters<typeof this.enhance>[0] & {
        parameters: {
            n_samples?: 1;
            mask: Buffer;
        };
    }): Promise<Buffer>;
    generateInFill(options: Parameters<typeof this.enhance>[0] & {
        parameters: {
            n_samples: number;
            mask: Buffer;
        };
    }): Promise<Buffer[]>;
    generateInFill(options: Parameters<typeof this.enhance>[0] & {
        parameters: {
            mask: Buffer;
        };
    }): Promise<Buffer | Buffer[]>;
    generateVariations(options: Parameters<typeof this.enhance>[0] & {
        parameters?: {
            n_samples: 1 | undefined;
        };
    }): Promise<Buffer>;
    generateVariations(options: Parameters<typeof this.enhance>[0] & {
        parameters?: {
            n_samples: number;
        };
    }): Promise<Buffer[]>;
    generateVariations(options: Parameters<typeof this.enhance>[0]): Promise<Buffer | Buffer[]>;
    upscale(options: {
        height: number;
        image: Buffer;
        scale: number;
        width: number;
    }): Promise<Buffer>;
    getMask(options: {
        model: "hed" | "midas" | "fake_scribble" | "mlsd" | "uniformer";
        parameters: {
            image: Buffer;
        };
    }): Promise<Buffer>;
}