koishi-plugin-hhs-huatu/lib/NaiInpaintService.d.ts

Generate images by diffusion models

interface ProcessedImageResult {
    cleanImage: Buffer;
    dimmedImage: Buffer;
    width: number;
    height: number;
}
interface InpaintPayload {
    input?: string;
    model: string;
    action: 'infill';
    parameters: any;
}
export declare class NaiInpaintService {
    /**
     * 预处理输入图像
     * 1. 移除 Alpha 通道 (强制 RGB)
     * 2. 调整尺寸为 64 的倍数 (NovelAI 限制)
     * 3. 生成一张变暗的引导图供用户涂抹
     */
    processInputImage(inputBuffer: Buffer): Promise<ProcessedImageResult>;
    /**
     * 从用户涂抹的图中提取蒙版
     * 用户会在暗图上用白色画笔涂抹
     */
    extractMaskFromUserDrawing(cleanImageBuffer: Buffer, userDrawnBuffer: Buffer): Promise<Buffer>;
    /**
     * 构造 NovelAI API Payload
     */
    constructInpaintPayload(model: string, imageBase64: string, maskBase64: string, prompt: string, negativePrompt: string, params: any): InpaintPayload;
}
export {};