kie-ai-cli 1.0.0

package/dist/image-models.js

@@ -0,0 +1,336 @@
+import chalk from 'chalk';
+import select from '@inquirer/select';
+import { CONFIG_KEY_IMAGE_MODEL } from './models.js';
+export const DEFAULT_IMAGE_MODEL = 'gpt-image-2-text-to-image';
+export const IMAGE_MODELS = [
+    {
+        id: 'gpt-image/1.5-text-to-image',
+        label: 'GPT Image 1.5 文生图',
+        kind: 'text-to-image',
+        description: '官方推荐，支持 1:1 / 2:3 / 3:2 与 medium/high 质量',
+        supportsQuality: true,
+        aspectRatios: ['1:1', '2:3', '3:2'],
+    },
+    {
+        id: 'gpt-image/1.5-image-to-image',
+        label: 'GPT Image 1.5 图生图',
+        kind: 'image-to-image',
+        description: '基于 input_urls 编辑图片（需先上传获得 URL）',
+        supportsQuality: true,
+        aspectRatios: ['1:1', '2:3', '3:2'],
+    },
+    {
+        id: 'gpt-image-2-text-to-image',
+        label: 'GPT Image 2 文生图',
+        kind: 'text-to-image',
+        description: 'aspect_ratio 使用 auto',
+        supportsQuality: false,
+        aspectRatios: ['auto'],
+    },
+    {
+        id: 'gpt-image-2-image-to-image',
+        label: 'GPT Image 2 图生图',
+        kind: 'image-to-image',
+        description: '基于 input_urls 编辑图片，aspect_ratio 使用 auto',
+        supportsQuality: false,
+        aspectRatios: ['auto'],
+    },
+    {
+        id: 'bytedance/seedream',
+        label: 'ByteDance Seedream',
+        kind: 'text-to-image',
+        description: '字节跳动 Seedream 极速文生图，支持 square_hd / landscape_hd / portrait_hd',
+        supportsQuality: false,
+        aspectRatios: ['auto', '1:1', '16:9', '9:16', '3:2', '2:3', '4:3', '3:4'],
+    },
+    {
+        id: 'bytedance/seedream-v4-text-to-image',
+        label: 'ByteDance Seedream v4',
+        kind: 'text-to-image',
+        description: '字节跳动 Seedream v4 最新文生图，支持 1K 像素高精度排版，支持 seed 参数',
+        supportsQuality: false,
+        aspectRatios: ['auto', '1:1', '16:9', '9:16', '3:2', '2:3', '4:3', '3:4'],
+    },
+    {
+        id: 'bytedance/seedream-v4-edit',
+        label: 'ByteDance Seedream v4 编辑',
+        kind: 'image-to-image',
+        description: '字节跳动 Seedream v4 最新图生图/编辑，支持 1K 像素高精度，支持参考图与 seed',
+        supportsQuality: false,
+        aspectRatios: ['auto', '1:1', '16:9', '9:16', '3:2', '2:3', '4:3', '3:4'],
+    },
+    {
+        id: 'seedream/4.5-text-to-image',
+        label: 'Seedream 4.5 文生图',
+        kind: 'text-to-image',
+        description: 'Seedream 4.5 最新版本文生图，支持 aspect_ratio 比例与 quality 质量参数',
+        supportsQuality: true,
+        aspectRatios: ['1:1', '16:9', '9:16', '3:2', '2:3', '4:3', '3:4'],
+    },
+    {
+        id: 'seedream/4.5-edit',
+        label: 'Seedream 4.5 编辑',
+        kind: 'image-to-image',
+        description: 'Seedream 4.5 最新版本图生图/编辑，支持 aspect_ratio 比例与 quality 质量参数',
+        supportsQuality: true,
+        aspectRatios: ['1:1', '16:9', '9:16', '3:2', '2:3', '4:3', '3:4'],
+    },
+    {
+        id: 'seedream/5-lite-text-to-image',
+        label: 'Seedream 5 Lite 文生图',
+        kind: 'text-to-image',
+        description: 'Seedream 5 Lite 最新极速轻量级文生图，支持 aspect_ratio 比例与 quality 质量参数',
+        supportsQuality: true,
+        aspectRatios: ['1:1', '16:9', '9:16', '3:2', '2:3', '4:3', '3:4'],
+    },
+];
+export function findImageModel(modelId) {
+    return IMAGE_MODELS.find((m) => m.id === modelId);
+}
+export function getImageModel(config) {
+    return config.get(CONFIG_KEY_IMAGE_MODEL) || DEFAULT_IMAGE_MODEL;
+}
+export function setImageModel(config, modelId) {
+    config.set(CONFIG_KEY_IMAGE_MODEL, modelId.trim());
+}
+export function resolveImageModel(config, cliModel) {
+    return cliModel?.trim() || getImageModel(config);
+}
+export function formatImageModelLabel(modelId) {
+    const known = findImageModel(modelId);
+    if (!known)
+        return modelId;
+    const kind = known.kind === 'text-to-image' ? '文生图' : '图生图';
+    return `${known.label} (${modelId}) [${kind}]`;
+}
+/** 按模型构造 createTask 的 input 对象 */
+export function buildImageTaskInput(modelId, options) {
+    const known = findImageModel(modelId);
+    const prompt = options.prompt.trim();
+    if (!prompt) {
+        throw new Error('提示词不能为空');
+    }
+    if (modelId === 'seedream/4.5-edit') {
+        const urls = options.imageUrls?.filter(Boolean) ?? [];
+        if (urls.length === 0) {
+            throw new Error(`模型 ${modelId} 需要参考图，请使用 --image-url <url>（支持本地图片自动上传）`);
+        }
+        const aspectRatio = normalizeAspectRatio(options.aspectRatio, ['1:1', '16:9', '9:16', '3:2', '2:3', '4:3', '3:4'], '1:1');
+        return {
+            prompt,
+            image_urls: urls,
+            aspect_ratio: aspectRatio,
+            quality: options.quality ?? 'basic',
+            nsfw_checker: true,
+        };
+    }
+    if (modelId === 'seedream/5-lite-text-to-image' || modelId === 'seedream/4.5-text-to-image') {
+        const aspectRatio = normalizeAspectRatio(options.aspectRatio, ['1:1', '16:9', '9:16', '3:2', '2:3', '4:3', '3:4'], '1:1');
+        return {
+            prompt,
+            aspect_ratio: aspectRatio,
+            quality: options.quality ?? 'basic',
+            nsfw_checker: false,
+        };
+    }
+    if (modelId === 'bytedance/seedream-v4-edit') {
+        const urls = options.imageUrls?.filter(Boolean) ?? [];
+        if (urls.length === 0) {
+            throw new Error(`模型 ${modelId} 需要参考图，请使用 --image-url <url>（支持本地图片自动上传）`);
+        }
+        let image_size = 'square_hd';
+        const ratio = options.aspectRatio?.trim();
+        if (ratio) {
+            if (ratio === '1:1') {
+                image_size = 'square_hd';
+            }
+            else if (ratio === '16:9' || ratio === '3:2' || ratio === '4:3' || ratio === 'landscape') {
+                image_size = 'landscape_hd';
+            }
+            else if (ratio === '9:16' || ratio === '2:3' || ratio === '3:4' || ratio === 'portrait') {
+                image_size = 'portrait_hd';
+            }
+            else {
+                image_size = ratio;
+            }
+        }
+        const input = {
+            prompt,
+            image_urls: urls,
+            image_size,
+            image_resolution: '1K',
+            max_images: 1,
+            nsfw_checker: false,
+        };
+        if (options.seed !== undefined && !Number.isNaN(options.seed)) {
+            input.seed = options.seed;
+        }
+        return input;
+    }
+    if (modelId === 'bytedance/seedream-v4-text-to-image') {
+        let image_size = 'square_hd';
+        const ratio = options.aspectRatio?.trim();
+        if (ratio) {
+            if (ratio === '1:1') {
+                image_size = 'square_hd';
+            }
+            else if (ratio === '16:9' || ratio === '3:2' || ratio === '4:3' || ratio === 'landscape') {
+                image_size = 'landscape_hd';
+            }
+            else if (ratio === '9:16' || ratio === '2:3' || ratio === '3:4' || ratio === 'portrait') {
+                image_size = 'portrait_hd';
+            }
+            else {
+                image_size = ratio;
+            }
+        }
+        const input = {
+            prompt,
+            image_size,
+            image_resolution: '1K',
+            max_images: 1,
+            nsfw_checker: false,
+        };
+        if (options.seed !== undefined && !Number.isNaN(options.seed)) {
+            input.seed = options.seed;
+        }
+        return input;
+    }
+    if (modelId === 'bytedance/seedream') {
+        let image_size = 'square_hd';
+        const ratio = options.aspectRatio?.trim();
+        if (ratio) {
+            if (ratio === '1:1') {
+                image_size = 'square_hd';
+            }
+            else if (ratio === '16:9' || ratio === '3:2' || ratio === '4:3' || ratio === 'landscape') {
+                image_size = 'landscape_hd';
+            }
+            else if (ratio === '9:16' || ratio === '2:3' || ratio === '3:4' || ratio === 'portrait') {
+                image_size = 'portrait_hd';
+            }
+            else {
+                image_size = ratio;
+            }
+        }
+        return {
+            prompt,
+            image_size,
+            guidance_scale: 2.5,
+        };
+    }
+    if (known?.id === 'gpt-image-2-image-to-image') {
+        const urls = options.imageUrls?.filter(Boolean) ?? [];
+        if (urls.length === 0) {
+            throw new Error(`模型 ${modelId} 需要参考图，请使用 --image-url <url>（可多次指定）`);
+        }
+        return {
+            prompt,
+            input_urls: urls,
+            aspect_ratio: options.aspectRatio ?? 'auto',
+        };
+    }
+    if (known?.kind === 'image-to-image') {
+        const urls = options.imageUrls?.filter(Boolean) ?? [];
+        if (urls.length === 0) {
+            throw new Error(`模型 ${modelId} 需要参考图，请使用 --image-url <url>（可多次指定）`);
+        }
+        const aspectRatio = normalizeAspectRatio(options.aspectRatio, known.aspectRatios, '3:2');
+        const input = {
+            input_urls: urls,
+            prompt,
+            aspect_ratio: aspectRatio,
+        };
+        if (known.supportsQuality) {
+            input.quality = options.quality ?? 'medium';
+        }
+        return input;
+    }
+    if (known?.id === 'gpt-image-2-text-to-image') {
+        return {
+            prompt,
+            aspect_ratio: options.aspectRatio ?? 'auto',
+        };
+    }
+    if (known) {
+        const aspectRatio = normalizeAspectRatio(options.aspectRatio, known.aspectRatios, '1:1');
+        const input = {
+            prompt,
+            aspect_ratio: aspectRatio,
+        };
+        if (known.supportsQuality) {
+            input.quality = options.quality ?? 'medium';
+        }
+        return input;
+    }
+    // 自定义模型：尽量通用
+    const input = { prompt };
+    if (options.aspectRatio)
+        input.aspect_ratio = options.aspectRatio;
+    if (options.quality)
+        input.quality = options.quality;
+    if (options.imageUrls?.length)
+        input.input_urls = options.imageUrls;
+    return input;
+}
+function normalizeAspectRatio(value, allowed, fallback) {
+    if (!value)
+        return fallback;
+    if (allowed.includes(value))
+        return value;
+    if (allowed.includes('auto'))
+        return 'auto';
+    throw new Error(`宽高比 ${value} 不支持，可选: ${allowed.join(', ')}`);
+}
+export function printImageModels(currentModelId) {
+    console.log(chalk.yellow('\n可用图片模型 (Image，用于 kie image):\n'));
+    for (const m of IMAGE_MODELS) {
+        const current = m.id === currentModelId;
+        const marker = current ? chalk.green('● ') : chalk.gray('  ');
+        const kind = m.kind === 'text-to-image' ? chalk.cyan('文生图') : chalk.magenta('图生图');
+        console.log(`${marker}${chalk.bold(m.id.padEnd(32))} ${m.label.padEnd(20)} ${kind}  ${chalk.dim(m.description)}`);
+    }
+    if (!findImageModel(currentModelId)) {
+        console.log(chalk.gray(`\n当前使用自定义模型: ${chalk.cyan(currentModelId)}`));
+    }
+    console.log(chalk.gray('\n切换: kie image-models --set  |  kie config --set-image-model <id>\n'));
+}
+export function printImageModelPickerMenu(currentModelId) {
+    console.log(chalk.yellow('\n选择图片生成模型 — 输入序号或模型 ID，0 取消:\n'));
+    IMAGE_MODELS.forEach((m, i) => {
+        const current = m.id === currentModelId ? chalk.green(' ← 当前') : '';
+        const kind = m.kind === 'text-to-image' ? '文生图' : '图生图';
+        console.log(`  ${chalk.cyan(String(i + 1).padStart(2))}. [${kind}] ${chalk.bold(m.id)} — ${m.label}${current}`);
+    });
+    console.log(chalk.gray('\n  0. 取消\n'));
+}
+export function resolveImageModelPick(input) {
+    const trimmed = input.trim();
+    if (!trimmed || trimmed === '0')
+        return null;
+    const n = parseInt(trimmed, 10);
+    if (!Number.isNaN(n) && n >= 1 && n <= IMAGE_MODELS.length) {
+        return IMAGE_MODELS[n - 1].id;
+    }
+    return trimmed;
+}
+export async function promptSelectImageModel(currentModelId) {
+    const choices = IMAGE_MODELS.map((m) => ({
+        name: `${m.label} (${m.id}) — ${m.kind === 'text-to-image' ? '文生图' : '图生图'}`,
+        value: m.id,
+        description: m.description,
+    }));
+    if (!findImageModel(currentModelId)) {
+        choices.unshift({
+            name: `当前自定义: ${currentModelId}`,
+            value: currentModelId,
+            description: '当前正在使用',
+        });
+    }
+    return select({
+        message: '选择图片模型 (↑↓ 移动，Enter 确认)',
+        choices,
+        default: currentModelId,
+        pageSize: Math.min(choices.length, 10),
+    });
+}