import { OpenAICtodService } from './index.js'

type ApiResponse = {
    created: string
    data: {
        b64_json: string
    }[]
}

type Config = {
    /**
     * @zh 模型，支援 dall-e-2 和 dall-e-3
     * @en Model, support dall-e-2 and dall-e-3
     */
    model: 'dall-e-2' | 'dall-e-3'
    /**
     * @zh 解析度，例如 1024x1024
     * @en Resolution, for example 1024x1024
     */
    size: `${number}x${number}`
}

export class OpenAIImagesGeneration {
    private openai: OpenAICtodService
    private config: Config = {
        model: 'dall-e-2',
        size: '1024x1024'
    }

    constructor(openai: OpenAICtodService) {
        this.openai = openai
    }

    /**
     * @zh 改變對話的一些設定
     * @en Change some settings of the conversation
     */

    setConfig(options: Partial<Config>) {
        Object.assign(this.config, options)
    }

    /**
     * @zh 產生圖片
     * @en Generate image
     */

    async create(prompt: string) {
        const result = await this.openai._axios.post<ApiResponse>(`${this.openai._baseUrl}/v1/images/generations`, {
            prompt,
            n: 1,
            size: this.config.size,
            model: this.config.model,
            response_format: 'b64_json'
        }, {
            timeout: 1000 * 60 * 5,
            headers: {
                'Content-Type': 'application/json',
                'Authorization': `Bearer ${this.openai._apiKey}`
            }
        })
        return result.data
    }
}