try-on.js
1 /** 2 * Yollomi virtual try-on — POST /api/ai/virtual-try-on 3 */ 4 import * as path from 'node:path'; 5 import { cli, Strategy } from '@jackwener/opencli/registry'; 6 import { CliError } from '@jackwener/opencli/errors'; 7 import { log } from '@jackwener/opencli/logger'; 8 import { YOLLOMI_DOMAIN, yollomiPost, downloadOutput, fmtBytes } from './utils.js'; 9 cli({ 10 site: 'yollomi', 11 name: 'try-on', 12 description: 'Virtual try-on — see how clothes look on a person (3 credits)', 13 domain: YOLLOMI_DOMAIN, 14 strategy: Strategy.COOKIE, 15 args: [ 16 { name: 'person', required: true, help: 'Person photo URL (upload via "opencli yollomi upload" first)' }, 17 { name: 'cloth', required: true, help: 'Clothing image URL' }, 18 { name: 'cloth-type', default: 'upper', choices: ['upper', 'lower', 'overall'], help: 'Clothing type' }, 19 { name: 'output', default: './yollomi-output', help: 'Output directory' }, 20 { name: 'no-download', type: 'boolean', default: false, help: 'Only show URL' }, 21 ], 22 columns: ['status', 'file', 'size', 'url'], 23 func: async (page, kwargs) => { 24 log.status('Processing virtual try-on...'); 25 const data = await yollomiPost(page, '/api/ai/virtual-try-on', { 26 person_image: kwargs.person, 27 cloth_image: kwargs.cloth, 28 cloth_type: kwargs['cloth-type'], 29 output_format: 'png', 30 output_quality: 100, 31 }); 32 const url = data.image || (data.images?.[0]); 33 if (!url) 34 throw new CliError('EMPTY_RESPONSE', 'No result', 'Check both images have clear subjects'); 35 if (kwargs['no-download']) 36 return [{ status: 'generated', file: '-', size: '-', url }]; 37 try { 38 const filename = `yollomi_tryon_${Date.now()}.png`; 39 const { path: fp, size } = await downloadOutput(url, kwargs.output, filename); 40 return [{ status: 'saved', file: path.relative('.', fp), size: fmtBytes(size), url }]; 41 } 42 catch { 43 return [{ status: 'download-failed', file: '-', size: '-', url }]; 44 } 45 }, 46 });