/ clis / yollomi / try-on.js
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  });