amz-pic-flow/poc-workflow-v6.js

/**
 * POC Workflow V6: 优化Prompt控制策略
 * 
 * 核心改进：
 * 1. 修复素材选择（使用猫咪图而非狗狗图）
 * 2. 优化Prompt - 更精准地指示AI保持主体不变
 * 3. 分类处理不同复杂度的图片
 */

require('dotenv').config();
const axios = require('axios');
const fs = require('fs');
const path = require('path');
const sharp = require('sharp');
const { S3Client, PutObjectCommand } = require('@aws-sdk/client-s3');
const imageProcessor = require('./lib/image-processor');

// ============================================================
// 配置
// ============================================================
const API_KEY = 'G9rXx3Ag2Xfa7Gs8zou6t6HqeZ';
const API_BASE = 'https://api.wuyinkeji.com/api';

// R2客户端
const r2Client = new S3Client({
  region: 'auto',
  endpoint: `https://${process.env.R2_ACCOUNT_ID}.r2.cloudflarestorage.com`,
  credentials: {
    accessKeyId: process.env.R2_ACCESS_KEY_ID,
    secretAccessKey: process.env.R2_SECRET_ACCESS_KEY,
  },
  forcePathStyle: true,
});

// ============================================================
// 工具函数
// ============================================================

async function uploadToR2(buffer, filename) {
  const fileName = `v6-${Date.now()}-${filename}`;
  
  await r2Client.send(new PutObjectCommand({
    Bucket: process.env.R2_BUCKET_NAME || 'ai-flow',
    Key: fileName,
    Body: buffer,
    ContentType: filename.endsWith('.png') ? 'image/png' : 'image/jpeg',
  }));

  return process.env.R2_PUBLIC_DOMAIN
    ? `${process.env.R2_PUBLIC_DOMAIN}/${fileName}`
    : `https://${process.env.R2_ACCOUNT_ID}.r2.cloudflarestorage.com/${process.env.R2_BUCKET_NAME}/${fileName}`;
}

async function submitImageTask(prompt, refImageUrl = null, aspectRatio = '1:1') {
  const payload = {
    key: API_KEY,
    prompt: prompt,
    aspectRatio: aspectRatio,
    imageSize: '1K'
  };
  
  if (refImageUrl) {
    payload.img_url = refImageUrl;
  }

  const response = await axios.post(`${API_BASE}/img/nanoBanana-pro`, payload, { timeout: 30000 });
  const taskId = response.data.data?.id;
  
  if (!taskId) {
    throw new Error('No task ID returned');
  }
  return taskId;
}

async function pollImageResult(taskId) {
  let attempts = 0;
  const maxAttempts = 90;
  
  while (attempts < maxAttempts) {
    try {
      const response = await axios.get(`${API_BASE}/img/drawDetail`, {
        params: { key: API_KEY, id: taskId },
        timeout: 10000
      });
      
      const data = response.data.data;
      
      if (data && data.status === 2 && data.image_url) {
        return data.image_url;
      } else if (data && data.status === 3) {
        throw new Error('Generation failed: ' + (data.fail_reason || 'Unknown'));
      }
      
      process.stdout.write('.');
      await new Promise(r => setTimeout(r, 2000));
      attempts++;
    } catch (error) {
      if (error.message.includes('Generation failed')) throw error;
      process.stdout.write('x');
      await new Promise(r => setTimeout(r, 3000));
      attempts++;
    }
  }
  throw new Error('Timeout');
}

async function downloadImage(url) {
  const response = await axios.get(url, { responseType: 'arraybuffer', timeout: 30000 });
  return Buffer.from(response.data);
}

async function generateAIImage(prompt, refImageUrl = null, aspectRatio = '1:1') {
  console.log('   提交AI任务...');
  const taskId = await submitImageTask(prompt, refImageUrl, aspectRatio);
  console.log(`   Task ID: ${taskId}`);
  
  const imageUrl = await pollImageResult(taskId);
  console.log('\n   ✓ AI生成完成');
  
  return await downloadImage(imageUrl);
}

// ============================================================
// Prompt模板 - 核心优化
// ============================================================

/**
 * 构建精准的编辑prompt
 * 核心策略：明确告诉AI这是"编辑"任务，精确描述要保留的元素
 */
function buildEditPrompt(config) {
  const {
    subjectDescription,  // 主体描述（产品+宠物）
    preserveElements,    // 必须保留的元素
    editTasks,           // 编辑任务
    layoutDescription,   // 排版描述
    styleGuide,          // 风格指南
    aspectRatio          // 比例
  } = config;

  return `
[IMAGE EDITING TASK - NOT GENERATION]

REFERENCE IMAGE ANALYSIS:
The reference image shows: ${subjectDescription}

CRITICAL PRESERVATION REQUIREMENTS (DO NOT MODIFY):
${preserveElements.map((e, i) => `${i + 1}. ${e}`).join('\n')}

EDITING TASKS (ONLY THESE CHANGES ARE ALLOWED):
${editTasks.map((t, i) => `${i + 1}. ${t}`).join('\n')}

LAYOUT SPECIFICATION:
${layoutDescription}

STYLE GUIDE:
${styleGuide}

OUTPUT: ${aspectRatio} ratio, professional Amazon listing quality

FINAL CHECK: Before output, verify that the pet and product in the result match the reference image EXACTLY in:
- Species, breed, fur color and pattern
- Product color (#B5E5E8 ice blue), shape (flower/petal cone), brand text "TOUCHDOG"
- Pose and positioning relative to original
`.trim();
}

// ============================================================
// 12张图的生成逻辑
// ============================================================

/**
 * Main_01: 场景首图+卖点
 */
async function generateMain01(materials, config, outputDir) {
  console.log('\n🎨 [Main_01] 场景首图+卖点');
  
  const wornBuffer = fs.readFileSync(materials.catWornImage);
  const wornUrl = await uploadToR2(wornBuffer, 'cat-worn.jpg');
  
  const prompt = buildEditPrompt({
    subjectDescription: `A Ragdoll cat (white/cream fur with light brown markings, blue eyes) wearing an ice-blue soft recovery cone collar (brand: TOUCHDOG). The cat is in a home environment.`,
    
    preserveElements: [
      `The cat MUST remain a Ragdoll with blue eyes, white/cream fur, brown ear markings - EXACTLY as shown`,
      `The cone collar MUST remain ice-blue (#B5E5E8), flower/petal shaped, with "TOUCHDOG" branding`,
      `The cat's pose, expression, and the way the collar sits on the cat`,
      `The product's texture, stitching pattern, and velcro closure detail`
    ],
    
    editTasks: [
      `Enhance background to warm, cozy home interior (fireplace, blanket, wooden furniture)`,
      `Add soft warm lighting with gentle shadows`,
      `Add a curved blue banner (#4A7C9B) across middle area with text: "DESIGNED FOR COMFORTABLE RECOVERY"`,
      `Add 3 white feature boxes at bottom with icons and text:
         - Egg icon + "LIGHTER THAN AN EGG"
         - Water drop icon + "WATERPROOF & EASY WIPE"  
         - Cloud icon + "BREATHABLE COTTON LINING"`,
      `Add subtle paw print watermarks in light blue on the banner area`
    ],
    
    layoutDescription: `
      - Top 60%: Cat wearing product in enhanced home scene
      - Middle: Curved banner with main tagline
      - Bottom 25%: Three feature callout boxes in a row`,
    
    styleGuide: `Professional Amazon main image style. Warm color palette. Text should be crisp and readable. The cat and product should be the clear focal point.`,
    
    aspectRatio: '1:1 square'
  });

  const result = await generateAIImage(prompt, wornUrl, '1:1');
  await imageProcessor.saveImage(result, path.join(outputDir, 'Main_01.jpg'));
  
  // 保存prompt供调试
  fs.writeFileSync(path.join(outputDir, 'Main_01_prompt.txt'), prompt);
  return true;
}

/**
 * Main_03: 功能调节展示 (参考效果最好的那张)
 */
async function generateMain03(materials, config, outputDir) {
  console.log('\n🎨 [Main_03] 功能调节展示');
  
  // 使用平铺图作为主参考
  const flatBuffer = fs.readFileSync(materials.flatImage);
  const flatUrl = await uploadToR2(flatBuffer, 'flat.png');
  
  const prompt = buildEditPrompt({
    subjectDescription: `An ice-blue soft pet recovery cone collar (TOUCHDOG brand) shown in flat lay position. The product has a flower/petal shape with 8 segments, velcro closure, and green inner rim.`,
    
    preserveElements: [
      `The product MUST remain EXACTLY as shown: ice-blue color (#B5E5E8), flower shape, TOUCHDOG branding`,
      `The velcro strap detail on the left side`,
      `The green/teal inner rim around the neck hole`,
      `The 8-petal segmented design with visible stitching`
    ],
    
    editTasks: [
      `Place the product on a soft blue background (#6B9AC4) that complements the product color`,
      `Add title text at top-left: "ADJUSTABLE STRAP FOR A SECURE FIT" in white, bold`,
      `Add 2 circular detail callout images on the right side:
         - Top circle: Close-up of the velcro strap, caption "SECURE THE ADJUSTABLE STRAP"
         - Bottom circle: Product being worn by a pet showing fit, caption "ADJUST FOR A SNUG FIT"`,
      `Add decorative white paw prints scattered in the background`,
      `Add thin white circular borders around the detail callouts`
    ],
    
    layoutDescription: `
      - Left side (60%): Main product image, slightly angled
      - Top-left corner: Title text
      - Right side (40%): Two stacked circular detail images with captions
      - Scattered paw prints as decoration`,
    
    styleGuide: `Clean infographic style. Blue color scheme matching the product. Professional Amazon A+ content quality. High contrast text for readability.`,
    
    aspectRatio: '1:1 square'
  });

  const result = await generateAIImage(prompt, flatUrl, '1:1');
  await imageProcessor.saveImage(result, path.join(outputDir, 'Main_03.jpg'));
  fs.writeFileSync(path.join(outputDir, 'Main_03_prompt.txt'), prompt);
  return true;
}

/**
 * Main_04: 多场景使用 - 关键优化
 * 这是最难的图，需要在4个场景中保持同一产品和宠物
 */
async function generateMain04(materials, config, outputDir) {
  console.log('\n🎨 [Main_04] 多场景使用 (4宫格)');
  
  const wornBuffer = fs.readFileSync(materials.catWornImage);
  const wornUrl = await uploadToR2(wornBuffer, 'cat-worn-grid.jpg');
  
  const prompt = buildEditPrompt({
    subjectDescription: `A Ragdoll cat (white/cream long fur, light brown markings on ears and face, distinctive blue eyes, pink nose) wearing an ice-blue TOUCHDOG soft recovery cone collar.`,
    
    preserveElements: [
      `ALL 4 SCENES MUST SHOW THE EXACT SAME CAT: Ragdoll breed, blue eyes, white/cream fur, brown ear markings`,
      `ALL 4 SCENES MUST SHOW THE EXACT SAME PRODUCT: Ice-blue (#B5E5E8) flower-shaped soft cone, TOUCHDOG brand`,
      `The cat's fur texture and coloring must be consistent across all 4 images`,
      `The product's color, shape, and branding must be identical in all 4 scenes`
    ],
    
    editTasks: [
      `Create a 2x2 grid layout with 4 scenes, each in a rounded rectangle frame`,
      `TOP-LEFT scene: Cat standing alert, wearing the cone - Caption: "• HYGIENIC & EASY TO CLEAN"`,
      `TOP-RIGHT scene: Cat eating/drinking from a bowl while wearing cone - Caption: "• UNRESTRICTED EATING/DRINKING"`,
      `BOTTOM-LEFT scene: Cat stretching or playing while wearing cone - Caption: "• REVERSIBLE WEAR"`,
      `BOTTOM-RIGHT scene: Cat sleeping peacefully curled up with cone - Caption: "• 360° COMFORT"`,
      `Add light paw print watermarks in corners of the overall image`
    ],
    
    layoutDescription: `
      - Background: Warm beige/cream (#F5EBE0)
      - 2x2 grid of equal-sized rounded rectangles
      - Each scene has the same cat in different poses
      - Caption text below each scene in brown (#8B7355)
      - Subtle paw prints in corners`,
    
    styleGuide: `
      Lifestyle photography style. Warm, inviting colors.
      The cat should look comfortable and happy in all scenes.
      Consistent lighting across all 4 scenes.
      CRITICAL: The cat must be recognizably the SAME individual cat in all scenes.`,
    
    aspectRatio: '1:1 square'
  });

  const result = await generateAIImage(prompt, wornUrl, '1:1');
  await imageProcessor.saveImage(result, path.join(outputDir, 'Main_04.jpg'));
  fs.writeFileSync(path.join(outputDir, 'Main_04_prompt.txt'), prompt);
  return true;
}

/**
 * Main_05: 尺寸图
 */
async function generateMain05(materials, config, outputDir) {
  console.log('\n🎨 [Main_05] 尺寸图');
  
  const flatBuffer = fs.readFileSync(materials.flatImage);
  const flatUrl = await uploadToR2(flatBuffer, 'flat-size.png');
  
  const prompt = buildEditPrompt({
    subjectDescription: `An ice-blue TOUCHDOG soft pet recovery cone collar in flat lay position, showing its flower/petal shape design.`,
    
    preserveElements: [
      `The product MUST remain EXACTLY as shown: ice-blue color, flower shape, TOUCHDOG branding`,
      `The product's proportions and shape`,
      `The velcro closure and inner rim details`
    ],
    
    editTasks: [
      `Place product on warm beige background (#F5EDE4)`,
      `Add title "PRODUCT SIZE" at top center in dark text`,
      `Add a size chart table below the product with columns: SIZE | NECK | DEPTH`,
      `Table data:
         XS: 5.6-6.8IN | 3.2IN
         S: 7.2-8.4IN | 4IN
         M: 8.8-10.4IN | 5IN
         L: 10.8-12.4IN | 6IN
         XL: 12.8-14.4IN | 7IN`,
      `Add note in coral/salmon color: "NOTE: ALWAYS MEASURE YOUR PET'S NECK BEFORE SELECTING A SIZE"`,
      `Add subtle paw print decorations in beige tones`
    ],
    
    layoutDescription: `
      - Top: Title
      - Center: Product image (main focus)
      - Bottom: Size chart table with clean design`,
    
    styleGuide: `Clean, informative infographic style. Easy to read table. Warm neutral background. Professional Amazon listing quality.`,
    
    aspectRatio: '1:1 square'
  });

  const result = await generateAIImage(prompt, flatUrl, '1:1');
  await imageProcessor.saveImage(result, path.join(outputDir, 'Main_05.jpg'));
  fs.writeFileSync(path.join(outputDir, 'Main_05_prompt.txt'), prompt);
  return true;
}

/**
 * Main_02: 白底平铺+细节放大
 */
async function generateMain02(materials, config, outputDir) {
  console.log('\n🎨 [Main_02] 白底平铺+细节放大');
  
  const flatBuffer = fs.readFileSync(materials.flatImage);
  const flatUrl = await uploadToR2(flatBuffer, 'flat-detail.png');
  
  const prompt = buildEditPrompt({
    subjectDescription: `An ice-blue TOUCHDOG soft pet recovery cone collar in flat lay position on black background. The product has a flower/petal shape with 8 segments, velcro closure on left, and green inner rim.`,
    
    preserveElements: [
      `The product MUST remain EXACTLY as shown: ice-blue color (#B5E5E8), flower shape, 8 petal segments`,
      `The TOUCHDOG branding text on the product`,
      `The velcro strap detail and green/teal inner rim`,
      `The product's proportions and stitching pattern`
    ],
    
    editTasks: [
      `Change background from black to clean white`,
      `Add a dark green (#2D4A3E) banner at top with text: "DURABLE WATERPROOF PU LAYER"`,
      `Add 2 circular magnified detail callouts at bottom:
         - Left circle: Close-up of outer material texture, label "DURABLE WATERPROOF PU LAYER"
         - Right circle: Close-up of inner lining, label "DOUBLE-LAYER COMFORT"`,
      `Add thin green borders around detail circles`,
      `Keep layout clean and professional`
    ],
    
    layoutDescription: `
      - Top: Dark green banner with title
      - Center: Main product image on white background
      - Bottom: Two circular detail magnifications with labels`,
    
    styleGuide: `Clean white background product photography. Professional Amazon main image style. High contrast, sharp details.`,
    
    aspectRatio: '1:1 square'
  });

  const result = await generateAIImage(prompt, flatUrl, '1:1');
  await imageProcessor.saveImage(result, path.join(outputDir, 'Main_02.jpg'));
  fs.writeFileSync(path.join(outputDir, 'Main_02_prompt.txt'), prompt);
  return true;
}

/**
 * Main_06: 多角度展示
 */
async function generateMain06(materials, config, outputDir) {
  console.log('\n🎨 [Main_06] 多角度展示');
  
  const wornBuffer = fs.readFileSync(materials.catWornImage);
  const wornUrl = await uploadToR2(wornBuffer, 'multi-angle.jpg');
  
  const prompt = buildEditPrompt({
    subjectDescription: `A Ragdoll cat (white/cream fur, brown ear markings, blue eyes) wearing an ice-blue TOUCHDOG soft recovery cone collar.`,
    
    preserveElements: [
      `The cat MUST remain a Ragdoll: blue eyes, white/cream fur, brown markings`,
      `The product MUST remain ice-blue (#B5E5E8), flower-shaped, TOUCHDOG branded`,
      `Both views must show the EXACT SAME cat and product`
    ],
    
    editTasks: [
      `Create a split image with curved decorative divider in the middle`,
      `LEFT SIDE: Front view of cat wearing the cone (from reference)`,
      `RIGHT SIDE: Side/profile view of the same cat wearing the same cone`,
      `Warm home interior background in both halves`,
      `NO text overlays - pure lifestyle imagery`
    ],
    
    layoutDescription: `
      - Split layout with elegant curved divider
      - Left: Front view
      - Right: Side view
      - Consistent warm lighting across both`,
    
    styleGuide: `Lifestyle photography. Warm, cozy atmosphere. The cat should look comfortable. NO text.`,
    
    aspectRatio: '1:1 square'
  });

  const result = await generateAIImage(prompt, wornUrl, '1:1');
  await imageProcessor.saveImage(result, path.join(outputDir, 'Main_06.jpg'));
  fs.writeFileSync(path.join(outputDir, 'Main_06_prompt.txt'), prompt);
  return true;
}

/**
 * APlus_01: 品牌横幅
 */
async function generateAPlus01(materials, config, outputDir) {
  console.log('\n🎨 [APlus_01] 品牌横幅');
  
  const wornBuffer = fs.readFileSync(materials.catWornImage);
  const wornUrl = await uploadToR2(wornBuffer, 'brand-banner.jpg');
  
  const prompt = buildEditPrompt({
    subjectDescription: `A Ragdoll cat wearing an ice-blue TOUCHDOG soft recovery cone collar in a home setting.`,
    
    preserveElements: [
      `The cat MUST remain a Ragdoll with blue eyes and white/cream fur`,
      `The product MUST remain ice-blue, flower-shaped, TOUCHDOG branded`
    ],
    
    editTasks: [
      `Create a horizontal banner layout (3:2 ratio)`,
      `LEFT 40%: Brand text area with:
         - "TOUCHDOG" in playful coral/salmon color (#E8A87C) 
         - Small paw print icons around the brand name
         - "CAT SOFT CONE COLLAR" below in gray`,
      `RIGHT 60%: The cat wearing the product in a warm home setting`,
      `Soft, warm color palette throughout`
    ],
    
    layoutDescription: `
      - Left side: Brand name and product title
      - Right side: Hero image of cat with product
      - Warm, cohesive color scheme`,
    
    styleGuide: `Amazon A+ brand story style. Warm and inviting. Professional yet friendly.`,
    
    aspectRatio: '3:2 landscape'
  });

  const result = await generateAIImage(prompt, wornUrl, '3:2');
  await imageProcessor.saveImage(result, path.join(outputDir, 'APlus_01.jpg'));
  fs.writeFileSync(path.join(outputDir, 'APlus_01_prompt.txt'), prompt);
  return true;
}

/**
 * APlus_02: 对比图
 */
async function generateAPlus02(materials, config, outputDir) {
  console.log('\n🎨 [APlus_02] 对比图');
  
  const wornBuffer = fs.readFileSync(materials.catWornImage);
  const wornUrl = await uploadToR2(wornBuffer, 'comparison.jpg');
  
  const prompt = buildEditPrompt({
    subjectDescription: `A Ragdoll cat wearing an ice-blue TOUCHDOG soft recovery cone collar.`,
    
    preserveElements: [
      `LEFT SIDE: The cat and product MUST match the reference - Ragdoll cat, ice-blue soft cone`,
      `The product color (#B5E5E8), flower shape, and TOUCHDOG branding on left side`
    ],
    
    editTasks: [
      `Create a side-by-side comparison layout`,
      `LEFT SIDE (60% width, colorful):
         - Happy Ragdoll cat wearing our ice-blue soft cone (from reference)
         - Green checkmark icon
         - "OUR" label on coral/orange banner
         - 3 benefits in white: "CLOUD-LIGHT COMFORT", "WIDER & CLEARER", "FOLDABLE & PORTABLE"`,
      `RIGHT SIDE (40% width, grayscale):
         - Sad-looking cat wearing a DIFFERENT product: hard plastic transparent cone (traditional e-collar)
         - Red X icon
         - "OTHER" label on gray banner
         - 3 drawbacks in gray: "HEAVY & BULKY", "BLOCKS VISION & MOVEMENT", "HARD TO STORE"`,
      `Warm beige background with paw print watermarks`
    ],
    
    layoutDescription: `
      - Split layout with curved divider
      - Left side larger, full color, positive mood
      - Right side smaller, desaturated, negative mood
      - Text overlays on each side`,
    
    styleGuide: `
      Clear visual contrast between "us" and "them".
      Left side warm and inviting, right side cold and uncomfortable.
      IMPORTANT: Right side must show a DIFFERENT type of cone (hard plastic), not our product.`,
    
    aspectRatio: '3:2 landscape'
  });

  const result = await generateAIImage(prompt, wornUrl, '3:2');
  await imageProcessor.saveImage(result, path.join(outputDir, 'APlus_02.jpg'));
  fs.writeFileSync(path.join(outputDir, 'APlus_02_prompt.txt'), prompt);
  return true;
}

/**
 * APlus_03: 功能细节展示
 */
async function generateAPlus03(materials, config, outputDir) {
  console.log('\n🎨 [APlus_03] 功能细节展示');
  
  const wornBuffer = fs.readFileSync(materials.catWornImage);
  const wornUrl = await uploadToR2(wornBuffer, 'feature-detail.jpg');
  
  const prompt = buildEditPrompt({
    subjectDescription: `A Ragdoll cat wearing an ice-blue TOUCHDOG soft recovery cone collar.`,
    
    preserveElements: [
      `The cat MUST remain a Ragdoll with blue eyes`,
      `The product MUST remain ice-blue, flower-shaped, TOUCHDOG branded`
    ],
    
    editTasks: [
      `Create horizontal layout (3:2) with title and 3 detail panels`,
      `Title at top: "ENGINEERED FOR UNCOMPROMISED COMFORT" in dark text`,
      `3 detail images in a row below:
         - Panel 1: Close-up of inner cotton lining texture, caption "STURDY AND BREATHABLE"
         - Panel 2: Cat wearing the product looking comfortable, caption "EASY TO CLEAN, STYLISH"
         - Panel 3: Close-up of stitching detail, caption "REINFORCED STITCHING"`,
      `Warm beige background (#F5EBE0) with subtle paw prints`
    ],
    
    layoutDescription: `
      - Top: Title banner
      - Bottom: Three equal-width panels with captions`,
    
    styleGuide: `Amazon A+ feature module style. Clean, informative. Warm color palette.`,
    
    aspectRatio: '3:2 landscape'
  });

  const result = await generateAIImage(prompt, wornUrl, '3:2');
  await imageProcessor.saveImage(result, path.join(outputDir, 'APlus_03.jpg'));
  fs.writeFileSync(path.join(outputDir, 'APlus_03_prompt.txt'), prompt);
  return true;
}

/**
 * APlus_04: 多场景横版
 */
async function generateAPlus04(materials, config, outputDir) {
  console.log('\n🎨 [APlus_04] 多场景横版');
  
  const wornBuffer = fs.readFileSync(materials.catWornImage);
  const wornUrl = await uploadToR2(wornBuffer, 'scenes-horizontal.jpg');
  
  const prompt = buildEditPrompt({
    subjectDescription: `A Ragdoll cat (white/cream fur, blue eyes, brown ear markings) wearing an ice-blue TOUCHDOG soft recovery cone collar.`,
    
    preserveElements: [
      `ALL 4 SCENES MUST SHOW THE EXACT SAME CAT: Ragdoll breed, blue eyes, white/cream fur`,
      `ALL 4 SCENES MUST SHOW THE EXACT SAME PRODUCT: Ice-blue flower-shaped soft cone`,
      `Consistent cat appearance and product across all panels`
    ],
    
    editTasks: [
      `Create horizontal layout (3:2) with 4 scenes in a row`,
      `Scene 1: Cat standing - "HYGIENIC & EASY TO CLEAN"`,
      `Scene 2: Cat eating from bowl - "UNRESTRICTED EATING"`,
      `Scene 3: Cat playing/stretching - "REVERSIBLE WEAR"`,
      `Scene 4: Cat sleeping curled up - "360° COMFORT"`,
      `Each scene in a rounded rectangle frame`,
      `Warm beige background with paw print decorations`
    ],
    
    layoutDescription: `
      - 4 equal panels arranged horizontally
      - Each panel has image + caption below
      - Consistent warm lighting across all`,
    
    styleGuide: `Lifestyle photography. Same cat in all scenes. Warm, cozy feel.`,
    
    aspectRatio: '3:2 landscape'
  });

  const result = await generateAIImage(prompt, wornUrl, '3:2');
  await imageProcessor.saveImage(result, path.join(outputDir, 'APlus_04.jpg'));
  fs.writeFileSync(path.join(outputDir, 'APlus_04_prompt.txt'), prompt);
  return true;
}

/**
 * APlus_05: 多角度横版
 */
async function generateAPlus05(materials, config, outputDir) {
  console.log('\n🎨 [APlus_05] 多角度横版');
  
  const wornBuffer = fs.readFileSync(materials.catWornImage);
  const wornUrl = await uploadToR2(wornBuffer, 'angles-horizontal.jpg');
  
  const prompt = buildEditPrompt({
    subjectDescription: `A Ragdoll cat wearing an ice-blue TOUCHDOG soft recovery cone collar.`,
    
    preserveElements: [
      `Both views must show the SAME Ragdoll cat with blue eyes`,
      `Both views must show the SAME ice-blue TOUCHDOG cone`
    ],
    
    editTasks: [
      `Create horizontal split layout (3:2)`,
      `LEFT: Front view of cat wearing cone`,
      `RIGHT: Side/profile view of same cat with cone`,
      `Elegant curved divider between the two views`,
      `Warm home background in both`,
      `NO text - pure visual showcase`
    ],
    
    layoutDescription: `
      - Two equal halves with curved divider
      - Left: Front angle
      - Right: Side angle
      - Warm consistent lighting`,
    
    styleGuide: `High-end lifestyle photography. No text. Warm atmosphere.`,
    
    aspectRatio: '3:2 landscape'
  });

  const result = await generateAIImage(prompt, wornUrl, '3:2');
  await imageProcessor.saveImage(result, path.join(outputDir, 'APlus_05.jpg'));
  fs.writeFileSync(path.join(outputDir, 'APlus_05_prompt.txt'), prompt);
  return true;
}

/**
 * APlus_06: 尺寸表横版
 * 优化：明确描述产品的C形开口特征
 */
async function generateAPlus06(materials, config, outputDir) {
  console.log('\n🎨 [APlus_06] 尺寸表横版');
  
  const flatBuffer = fs.readFileSync(materials.flatImage);
  const flatUrl = await uploadToR2(flatBuffer, 'size-chart-h.png');
  
  const prompt = buildEditPrompt({
    subjectDescription: `An ice-blue TOUCHDOG soft pet recovery cone collar in flat lay position. 
CRITICAL SHAPE DETAIL: The product has a C-SHAPED OPENING (not a closed circle) - there is a GAP on the left side where the velcro strap attaches. This opening allows the collar to wrap around the pet's neck.`,
    
    preserveElements: [
      `The product MUST keep its C-SHAPED OPENING - DO NOT close the gap into a full circle`,
      `The velcro strap visible on the left side of the opening`,
      `Ice-blue color (#B5E5E8), flower/petal shape with 8 segments`,
      `TOUCHDOG brand text on the product`,
      `Green/teal inner rim around the neck hole`
    ],
    
    editTasks: [
      `Change background from black to warm beige (#F5EDE4)`,
      `Add title "PRODUCT SIZE" at top center in dark text`,
      `Add dimension labels: "NECK" pointing to inner circle, "DEPTH" pointing outward`,
      `Add size chart table on the right:
         SIZE | NECK | DEPTH
         XS | 5.6-6.8IN | 3.2IN
         S | 7.2-8.4IN | 4IN
         M | 8.8-10.4IN | 5IN
         L | 10.8-12.4IN | 6IN
         XL | 12.8-14.4IN | 7IN`,
      `Add note in coral: "ALWAYS MEASURE YOUR PET'S NECK BEFORE SELECTING A SIZE"`,
      `Add subtle paw print watermarks`
    ],
    
    layoutDescription: `
      - Left 45%: Product image maintaining C-shape opening
      - Right 55%: Size chart table
      - Top: Title
      - Bottom: Note text`,
    
    styleGuide: `Clean infographic style. The product's C-shaped opening must be clearly visible - this is a key feature showing how it wraps around the neck.`,
    
    aspectRatio: '3:2 landscape'
  });

  const result = await generateAIImage(prompt, flatUrl, '3:2');
  await imageProcessor.saveImage(result, path.join(outputDir, 'APlus_06.jpg'));
  fs.writeFileSync(path.join(outputDir, 'APlus_06_prompt.txt'), prompt);
  return true;
}

// ============================================================
// 主流程
// ============================================================

async function main() {
  const args = process.argv.slice(2);
  let materialDir = path.join(__dirname, '素材/素材/已有的素材');
  let skuName = 'Touchdog冰蓝色伊丽莎白圈';
  
  // 可选：只生成指定的图
  let onlyGenerate = null;  // e.g., ['Main_01', 'Main_04']
  
  for (const arg of args) {
    if (arg.startsWith('--material-dir=')) {
      materialDir = arg.split('=')[1];
    } else if (arg.startsWith('--sku-name=')) {
      skuName = arg.split('=')[1];
    } else if (arg.startsWith('--only=')) {
      onlyGenerate = arg.split('=')[1].split(',');
    }
  }
  
  const timestamp = new Date().toISOString().slice(0, 10);
  const timeStr = new Date().toTimeString().slice(0, 8).replace(/:/g, '-');
  const safeName = skuName.replace(/[^a-zA-Z0-9\u4e00-\u9fa5]/g, '_').slice(0, 20);
  const outputDir = path.join(__dirname, `output_v6_${safeName}_${timestamp}-${timeStr}`);
  
  console.log('\n' + '═'.repeat(70));
  console.log('🚀 POC Workflow V6 - 优化Prompt控制策略');
  console.log('═'.repeat(70));
  console.log('\n📋 核心策略: 精准Prompt控制，明确保留元素，限定编辑范围');
  console.log('   素材目录:', materialDir);
  console.log('   输出目录:', outputDir);
  
  if (!fs.existsSync(outputDir)) {
    fs.mkdirSync(outputDir, { recursive: true });
  }
  
  // 扫描素材
  const files = fs.readdirSync(materialDir);
  const images = files.filter(f => /\.(jpg|jpeg|png)$/i.test(f));
  
  console.log('\n📁 可用素材:');
  images.forEach(img => console.log(`   - ${img}`));
  
  // ========== 修复：正确识别素材 ==========
  // 平铺图：IMG_5683.png（黑底抠图）
  const flatImage = images.find(i => i.includes('5683')) || images.find(i => i.endsWith('.png'));
  
  // 猫咪佩戴图：IMG_6216 或 IMG_6229（不是6514的狗狗图！）
  const catWornImage = images.find(i => i.includes('6216')) || 
                       images.find(i => i.includes('6229')) ||
                       images.find(i => i.toLowerCase().includes('cat'));
  
  // 狗狗佩戴图（备用）
  const dogWornImage = images.find(i => i.includes('6514'));
  
  if (!flatImage) {
    console.error('❌ 找不到平铺图素材');
    return;
  }
  if (!catWornImage) {
    console.error('❌ 找不到猫咪佩戴图素材');
    console.log('   可用文件:', images);
    return;
  }
  
  const materials = {
    flatImage: path.join(materialDir, flatImage),
    catWornImage: path.join(materialDir, catWornImage),
    dogWornImage: dogWornImage ? path.join(materialDir, dogWornImage) : null
  };
  
  console.log('\n✅ 素材选择:');
  console.log('   平铺图:', flatImage);
  console.log('   猫咪佩戴图:', catWornImage, '← 使用这个');
  if (dogWornImage) console.log('   狗狗佩戴图:', dogWornImage, '(备用)');
  
  const config = { brandName: 'TOUCHDOG', productName: 'CAT SOFT CONE COLLAR' };
  const results = [];
  
  // ========== 生成图片 ==========
  console.log('\n' + '─'.repeat(70));
  console.log('🎨 开始生成图片');
  console.log('─'.repeat(70));
  
  const generators = {
    'Main_01': () => generateMain01(materials, config, outputDir),
    'Main_02': () => generateMain02(materials, config, outputDir),
    'Main_03': () => generateMain03(materials, config, outputDir),
    'Main_04': () => generateMain04(materials, config, outputDir),
    'Main_05': () => generateMain05(materials, config, outputDir),
    'Main_06': () => generateMain06(materials, config, outputDir),
    'APlus_01': () => generateAPlus01(materials, config, outputDir),
    'APlus_02': () => generateAPlus02(materials, config, outputDir),
    'APlus_03': () => generateAPlus03(materials, config, outputDir),
    'APlus_04': () => generateAPlus04(materials, config, outputDir),
    'APlus_05': () => generateAPlus05(materials, config, outputDir),
    'APlus_06': () => generateAPlus06(materials, config, outputDir),
  };
  
  for (const [id, generator] of Object.entries(generators)) {
    if (onlyGenerate && !onlyGenerate.includes(id)) {
      console.log(`\n⏭️  [${id}] 跳过`);
      continue;
    }
    
    try {
      results.push({ id, success: await generator() });
    } catch (e) {
      console.error(`\n❌ ${id} 失败:`, e.message);
      results.push({ id, success: false, error: e.message });
    }
  }
  
  // ========== 总结 ==========
  console.log('\n' + '═'.repeat(70));
  console.log('📊 生成完成');
  console.log('═'.repeat(70));
  
  const successCount = results.filter(r => r.success).length;
  console.log(`\n✅ 成功: ${successCount}/${results.length}`);
  
  if (successCount < results.length) {
    console.log('\n❌ 失败:');
    results.filter(r => !r.success).forEach(r => console.log(`   - ${r.id}: ${r.error}`));
  }
  
  console.log(`\n📁 输出目录: ${outputDir}`);
  console.log('   每张图都保存了对应的 _prompt.txt 供调试');
  
  // 保存所有prompt到一个文件
  const allPrompts = {};
  for (const id of Object.keys(generators)) {
    const promptFile = path.join(outputDir, `${id}_prompt.txt`);
    if (fs.existsSync(promptFile)) {
      allPrompts[id] = fs.readFileSync(promptFile, 'utf-8');
    }
  }
  fs.writeFileSync(path.join(outputDir, 'all-prompts.json'), JSON.stringify(allPrompts, null, 2));
  fs.writeFileSync(path.join(outputDir, 'results.json'), JSON.stringify(results, null, 2));
}

main().catch(console.error);