myaidev-method 0.2.22 → 0.2.23

package/src/scripts/generate-visual-cli.js

@@ -12,268 +12,398 @@
  * Platform support: Claude Code, Gemini CLI, Codex CLI
  */
 
-import { generateImage, generateVideoVeo, estimateCost, validateAPIKeys } from '../lib/visual-generation-utils.js';
-import { saveImage, saveVideo, generateMarkdownReference, checkBudgetLimit, getTodaysCost, getMonthCost } from '../lib/asset-management.js';
-import { hasAnyAPIKeys, getConfigStatusMessage, getServiceDisplayInfo } from '../lib/visual-config-utils.js';
-import chalk from 'chalk';
-import ora from 'ora';
+import {
+	generateImage,
+	generateVideoVeo,
+	estimateCost,
+	validateAPIKeys,
+} from "../lib/visual-generation-utils.js";
+import {
+	saveImage,
+	saveVideo,
+	generateMarkdownReference,
+	checkBudgetLimit,
+	getTodaysCost,
+	getMonthCost,
+} from "../lib/asset-management.js";
+import {
+	hasAnyAPIKeys,
+	getConfigStatusMessage,
+	getServiceDisplayInfo,
+} from "../lib/visual-config-utils.js";
+import chalk from "chalk";
+import ora from "ora";
 
 // Parse command line arguments
 const args = process.argv.slice(2);
 
-if (args.length === 0 || args[0] === '--help' || args[0] === '-h') {
-  showHelp();
-  process.exit(0);
+if (args.length === 0 || args[0] === "--help" || args[0] === "-h") {
+	showHelp();
+	process.exit(0);
 }
 
 // Extract prompt (first non-option argument)
-const prompt = args.find(arg => !arg.startsWith('--'));
+const prompt = args.find((arg) => !arg.startsWith("--"));
 
 if (!prompt) {
-  console.error(chalk.red('❌ Error: Prompt is required'));
-  console.log('\nUsage: node generate-visual-cli.js "your prompt" --type=hero --service=gemini');
-  process.exit(1);
+	console.error(chalk.red("❌ Error: Prompt is required"));
+	console.log(
+		'\nUsage: node generate-visual-cli.js "your prompt" --type=hero --service=gemini',
+	);
+	process.exit(1);
 }
 
 // Parse options
 const options = {
-  type: getOption('type', 'hero'),
-  service: getOption('service', null),
-  size: getOption('size', '1024x1024'),
-  quality: getOption('quality', 'standard'),
-  description: getOption('description', null)
+	type: getOption("type", "hero"),
+	service: getOption("service", null),
+	size: getOption("size", "1024x1024"),
+	quality: getOption("quality", "standard"),
+	description: getOption("description", null),
 };
 
 // Main execution
 (async () => {
-  try {
-    // Check configuration
-    if (!hasAnyAPIKeys()) {
-      console.log(chalk.yellow(getConfigStatusMessage()));
-      process.exit(1);
-    }
-
-    // Validate type
-    const validTypes = ['hero', 'illustration', 'diagram', 'screenshot', 'video'];
-    if (!validTypes.includes(options.type)) {
-      console.error(chalk.red(`❌ Invalid type: ${options.type}`));
-      console.log(chalk.gray(`Valid types: ${validTypes.join(', ')}`));
-      process.exit(1);
-    }
-
-    // Display configuration
-    console.log(chalk.blue('\n🎨 Visual Content Generation\n'));
-    console.log(chalk.gray(`Type: ${options.type}`));
-    console.log(chalk.gray(`Prompt: "${prompt}"`));
-    if (options.service) {
-      console.log(chalk.gray(`Service: ${options.service}`));
-    }
-    console.log('');
-
-    // Estimate cost
-    let service = options.service;
-    if (!service) {
-      // Auto-select based on available services
-      const validation = validateAPIKeys();
-      service = validation.availableServices[0];
-      console.log(chalk.gray(`Auto-selected service: ${service}`));
-    }
-
-    const cost = estimateCost(service, { quality: options.quality, size: options.size });
-
-    // Check budget
-    const budget = await checkBudgetLimit(cost);
-    const todayCost = await getTodaysCost();
-    const monthCost = await getMonthCost();
-
-    console.log(chalk.cyan('💰 Cost Estimate:'));
-    console.log(chalk.gray(`   This generation: $${cost.toFixed(2)}`));
-    console.log(chalk.gray(`   Today: $${todayCost.toFixed(2)} / $${budget.dailyBudget.toFixed(2)} (${Math.round((todayCost/budget.dailyBudget)*100)}%)`));
-    console.log(chalk.gray(`   Month: $${monthCost.toFixed(2)} / $${budget.monthlyBudget.toFixed(2)} (${Math.round((monthCost/budget.monthlyBudget)*100)}%)`));
-    console.log('');
-
-    // Check if budget exceeded
-    if (budget.exceedsDailyBudget) {
-      console.error(chalk.red('❌ Daily budget limit exceeded!'));
-      console.log(chalk.yellow(`\nCurrent usage: $${budget.todaysCost.toFixed(2)}`));
-      console.log(chalk.yellow(`Daily limit: $${budget.dailyBudget.toFixed(2)}`));
-      console.log(chalk.gray('\nOptions:'));
-      console.log(chalk.gray('1. Increase VISUAL_DAILY_BUDGET in .env'));
-      console.log(chalk.gray('2. Wait until tomorrow (resets at midnight)'));
-      process.exit(1);
-    }
-
-    if (budget.exceedsMonthlyBudget) {
-      console.error(chalk.red('❌ Monthly budget limit exceeded!'));
-      console.log(chalk.yellow(`\nCurrent usage: $${budget.monthCost.toFixed(2)}`));
-      console.log(chalk.yellow(`Monthly limit: $${budget.monthlyBudget.toFixed(2)}`));
-      console.log(chalk.gray('\nOptions:'));
-      console.log(chalk.gray('1. Increase VISUAL_MONTHLY_BUDGET in .env'));
-      console.log(chalk.gray('2. Wait until next month'));
-      process.exit(1);
-    }
-
-    // Warnings
-    if (budget.dailyWarning) {
-      console.log(chalk.yellow(`⚠️  Warning: Approaching daily budget limit (${Math.round((budget.newDailyCost/budget.dailyBudget)*100)}%)`));
-      console.log('');
-    }
-
-    // Generate content
-    const spinner = ora(`Generating ${options.type} using ${service}...`).start();
-
-    let result;
-    let saved;
-
-    if (options.type === 'video') {
-      // Generate video
-      spinner.text = `Generating video using ${service}...`;
-      result = await generateVideoVeo(prompt, {
-        duration: 5,
-        aspectRatio: '16:9'
-      });
-
-      spinner.text = 'Saving video...';
-      saved = await saveVideo(result.buffer || Buffer.from(result.data, 'base64'), {
-        description: options.description || prompt,
-        service: result.service,
-        cost: result.cost,
-        prompt: prompt,
-        duration: result.duration || 5
-      });
-
-    } else {
-      // Generate image
-      spinner.text = `Generating ${options.type} using ${service}...`;
-      result = await generateImage(prompt, {
-        preferredService: service,
-        type: options.type,
-        quality: options.quality,
-        size: options.size
-      });
-
-      spinner.text = 'Saving image...';
-      saved = await saveImage(result.buffer, {
-        type: options.type,
-        description: options.description || prompt,
-        service: result.service,
-        cost: result.cost,
-        prompt: result.prompt
-      });
-    }
-
-    spinner.succeed(chalk.green(`${options.type === 'video' ? 'Video' : 'Image'} generated successfully!`));
-
-    // Display results
-    console.log('');
-    console.log(chalk.cyan(`📸 Details:`));
-
-    const serviceInfo = getServiceDisplayInfo(result.service);
-    if (serviceInfo) {
-      console.log(chalk.gray(`   Service: ${serviceInfo.name}`));
-    }
-
-    console.log(chalk.gray(`   Type: ${options.type.charAt(0).toUpperCase() + options.type.slice(1)}`));
-    console.log(chalk.gray(`   Cost: $${result.cost.toFixed(2)}`));
-    console.log(chalk.gray(`   File: ${saved.relativePath}`));
-    console.log(chalk.gray(`   Size: ${(saved.size / 1024).toFixed(1)} KB`));
-    if (options.type === 'video' && result.duration) {
-      console.log(chalk.gray(`   Duration: ${result.duration}s`));
-    }
-
-    // Display updated budget
-    const newTodayCost = await getTodaysCost();
-    const newMonthCost = await getMonthCost();
-
-    console.log('');
-    console.log(chalk.cyan('💰 Updated Budget:'));
-    console.log(chalk.gray(`   Today: $${newTodayCost.toFixed(2)} / $${budget.dailyBudget.toFixed(2)} (${Math.round((newTodayCost/budget.dailyBudget)*100)}%)`));
-    console.log(chalk.gray(`   Month: $${newMonthCost.toFixed(2)} / $${budget.monthlyBudget.toFixed(2)} (${Math.round((newMonthCost/budget.monthlyBudget)*100)}%)`));
-
-    // Generate markdown
-    const altText = `${options.type} ${options.type === 'video' ? 'video' : 'image'}: ${prompt}`;
-    const markdown = generateMarkdownReference(saved.relativePath, altText);
-
-    console.log('');
-    console.log(chalk.cyan('📋 Markdown Reference:'));
-    console.log(chalk.white(markdown));
-    console.log('');
-    console.log(chalk.gray('Copy and paste into your content ↑'));
-    console.log('');
-
-  } catch (error) {
-    console.error(chalk.red(`\n❌ Error: ${error.message}`));
-
-    // Provide helpful error messages
-    if (error.message.includes('rate_limit') || error.message.includes('rate limit')) {
-      console.log(chalk.yellow('\n⚠️  Rate limited by API'));
-      console.log(chalk.gray('Options:'));
-      console.log(chalk.gray('1. Wait 60 seconds and try again'));
-      console.log(chalk.gray('2. Switch to different service: --service=dalle'));
-      console.log(chalk.gray('3. Check your API quota limits'));
-    } else if (error.message.includes('API key') || error.message.includes('unauthorized')) {
-      console.log(chalk.yellow('\n⚠️  API key issue'));
-      console.log(chalk.gray('Run configuration: node src/scripts/configure-visual-apis.js'));
-    }
-
-    process.exit(1);
-  }
+	try {
+		// Check configuration
+		if (!hasAnyAPIKeys()) {
+			console.log(chalk.yellow(getConfigStatusMessage()));
+			process.exit(1);
+		}
+
+		// Validate type
+		const validTypes = [
+			"hero",
+			"illustration",
+			"diagram",
+			"screenshot",
+			"video",
+		];
+		if (!validTypes.includes(options.type)) {
+			console.error(chalk.red(`❌ Invalid type: ${options.type}`));
+			console.log(chalk.gray(`Valid types: ${validTypes.join(", ")}`));
+			process.exit(1);
+		}
+
+		// Display configuration
+		console.log(chalk.blue("\n🎨 Visual Content Generation\n"));
+		console.log(chalk.gray(`Type: ${options.type}`));
+		console.log(chalk.gray(`Prompt: "${prompt}"`));
+		if (options.service) {
+			console.log(chalk.gray(`Service: ${options.service}`));
+		}
+		console.log("");
+
+		// Estimate cost
+		let service = options.service;
+		if (!service) {
+			// Auto-select based on available services
+			const validation = validateAPIKeys();
+			service = validation.availableServices[0];
+			console.log(chalk.gray(`Auto-selected service: ${service}`));
+		}
+
+		const cost = estimateCost(service, {
+			quality: options.quality,
+			size: options.size,
+		});
+
+		// Check budget
+		const budget = await checkBudgetLimit(cost);
+		const todayCost = await getTodaysCost();
+		const monthCost = await getMonthCost();
+
+		console.log(chalk.cyan("💰 Cost Estimate:"));
+		console.log(chalk.gray(`   This generation: $${cost.toFixed(2)}`));
+		console.log(
+			chalk.gray(
+				`   Today: $${todayCost.toFixed(2)} / $${budget.dailyBudget.toFixed(2)} (${Math.round((todayCost / budget.dailyBudget) * 100)}%)`,
+			),
+		);
+		console.log(
+			chalk.gray(
+				`   Month: $${monthCost.toFixed(2)} / $${budget.monthlyBudget.toFixed(2)} (${Math.round((monthCost / budget.monthlyBudget) * 100)}%)`,
+			),
+		);
+		console.log("");
+
+		// Check if budget exceeded
+		if (budget.exceedsDailyBudget) {
+			console.error(chalk.red("❌ Daily budget limit exceeded!"));
+			console.log(
+				chalk.yellow(`\nCurrent usage: $${budget.todaysCost.toFixed(2)}`),
+			);
+			console.log(
+				chalk.yellow(`Daily limit: $${budget.dailyBudget.toFixed(2)}`),
+			);
+			console.log(chalk.gray("\nOptions:"));
+			console.log(chalk.gray("1. Increase VISUAL_DAILY_BUDGET in .env"));
+			console.log(chalk.gray("2. Wait until tomorrow (resets at midnight)"));
+			process.exit(1);
+		}
+
+		if (budget.exceedsMonthlyBudget) {
+			console.error(chalk.red("❌ Monthly budget limit exceeded!"));
+			console.log(
+				chalk.yellow(`\nCurrent usage: $${budget.monthCost.toFixed(2)}`),
+			);
+			console.log(
+				chalk.yellow(`Monthly limit: $${budget.monthlyBudget.toFixed(2)}`),
+			);
+			console.log(chalk.gray("\nOptions:"));
+			console.log(chalk.gray("1. Increase VISUAL_MONTHLY_BUDGET in .env"));
+			console.log(chalk.gray("2. Wait until next month"));
+			process.exit(1);
+		}
+
+		// Warnings
+		if (budget.dailyWarning) {
+			console.log(
+				chalk.yellow(
+					`⚠️  Warning: Approaching daily budget limit (${Math.round((budget.newDailyCost / budget.dailyBudget) * 100)}%)`,
+				),
+			);
+			console.log("");
+		}
+
+		// Generate content
+		const spinner = ora(
+			`Generating ${options.type} using ${service}...`,
+		).start();
+
+		let result;
+		let saved;
+
+		if (options.type === "video") {
+			// Generate video
+			spinner.text = `Generating video using ${service}...`;
+			result = await generateVideoVeo(prompt, {
+				duration: 5,
+				aspectRatio: "16:9",
+			});
+
+			spinner.text = "Saving video...";
+			saved = await saveVideo(
+				result.buffer || Buffer.from(result.data, "base64"),
+				{
+					description: options.description || prompt,
+					service: result.service,
+					cost: result.cost,
+					prompt: prompt,
+					duration: result.duration || 5,
+				},
+			);
+		} else {
+			// Generate image
+			spinner.text = `Generating ${options.type} using ${service}...`;
+			result = await generateImage(prompt, {
+				preferredService: service,
+				type: options.type,
+				quality: options.quality,
+				size: options.size,
+			});
+
+			spinner.text = "Saving image...";
+			saved = await saveImage(result.buffer, {
+				type: options.type,
+				description: options.description || prompt,
+				service: result.service,
+				cost: result.cost,
+				prompt: result.prompt,
+			});
+		}
+
+		spinner.succeed(
+			chalk.green(
+				`${options.type === "video" ? "Video" : "Image"} generated successfully!`,
+			),
+		);
+
+		// Display results
+		console.log("");
+		console.log(chalk.cyan(`📸 Details:`));
+
+		const serviceInfo = getServiceDisplayInfo(result.service);
+		if (serviceInfo) {
+			console.log(chalk.gray(`   Service: ${serviceInfo.name}`));
+		}
+
+		console.log(
+			chalk.gray(
+				`   Type: ${options.type.charAt(0).toUpperCase() + options.type.slice(1)}`,
+			),
+		);
+		console.log(chalk.gray(`   Cost: $${result.cost.toFixed(2)}`));
+		console.log(chalk.gray(`   File: ${saved.relativePath}`));
+		console.log(chalk.gray(`   Size: ${(saved.size / 1024).toFixed(1)} KB`));
+		if (options.type === "video" && result.duration) {
+			console.log(chalk.gray(`   Duration: ${result.duration}s`));
+		}
+
+		// Display updated budget
+		const newTodayCost = await getTodaysCost();
+		const newMonthCost = await getMonthCost();
+
+		console.log("");
+		console.log(chalk.cyan("💰 Updated Budget:"));
+		console.log(
+			chalk.gray(
+				`   Today: $${newTodayCost.toFixed(2)} / $${budget.dailyBudget.toFixed(2)} (${Math.round((newTodayCost / budget.dailyBudget) * 100)}%)`,
+			),
+		);
+		console.log(
+			chalk.gray(
+				`   Month: $${newMonthCost.toFixed(2)} / $${budget.monthlyBudget.toFixed(2)} (${Math.round((newMonthCost / budget.monthlyBudget) * 100)}%)`,
+			),
+		);
+
+		// Generate markdown
+		const altText = `${options.type} ${options.type === "video" ? "video" : "image"}: ${prompt}`;
+		const markdown = generateMarkdownReference(saved.relativePath, altText);
+
+		console.log("");
+		console.log(chalk.cyan("📋 Markdown Reference:"));
+		console.log(chalk.white(markdown));
+		console.log("");
+		console.log(chalk.gray("Copy and paste into your content ↑"));
+		console.log("");
+	} catch (error) {
+		console.error(chalk.red(`\n❌ Error: ${error.message}`));
+
+		// Provide helpful error messages
+		if (
+			error.message.includes("rate_limit") ||
+			error.message.includes("rate limit")
+		) {
+			console.log(chalk.yellow("\n⚠️  Rate limited by API"));
+			console.log(chalk.gray("Options:"));
+			console.log(chalk.gray("1. Wait 60 seconds and try again"));
+			console.log(
+				chalk.gray("2. Switch to different service: --service=dalle"),
+			);
+			console.log(chalk.gray("3. Check your API quota limits"));
+		} else if (
+			error.message.includes("API key") ||
+			error.message.includes("unauthorized")
+		) {
+			console.log(chalk.yellow("\n⚠️  API key issue"));
+			console.log(
+				chalk.gray(
+					"Run configuration: node src/scripts/configure-visual-apis.js",
+				),
+			);
+		}
+
+		process.exit(1);
+	}
 })();
 
 /**
  * Get option value from command line arguments
  */
 function getOption(name, defaultValue) {
-  const arg = args.find(a => a.startsWith(`--${name}=`));
-  if (arg) {
-    return arg.split('=')[1];
-  }
-  return defaultValue;
+	const arg = args.find((a) => a.startsWith(`--${name}=`));
+	if (arg) {
+		return arg.split("=")[1];
+	}
+	return defaultValue;
 }
 
 /**
  * Show help message
  */
 function showHelp() {
-  console.log(chalk.blue('\n🎨 Visual Content Generation CLI\n'));
-
-  console.log(chalk.white('Usage:'));
-  console.log(chalk.gray('  node generate-visual-cli.js "prompt" [options]\n'));
-
-  console.log(chalk.white('Options:'));
-  console.log(chalk.gray('  --type=<type>         Content type (default: hero)'));
-  console.log(chalk.gray('                        Values: hero, illustration, diagram, screenshot, video'));
-  console.log(chalk.gray('  --service=<service>   AI service (default: auto-select)'));
-  console.log(chalk.gray('                        Values: gemini, imagen, dalle, veo'));
-  console.log(chalk.gray('  --size=<size>         Image size (default: 1024x1024)'));
-  console.log(chalk.gray('                        Values: 1024x1024, 1792x1024, 1024x1792'));
-  console.log(chalk.gray('  --quality=<quality>   Quality level (default: standard)'));
-  console.log(chalk.gray('                        Values: standard, hd'));
-  console.log(chalk.gray('  --description=<desc>  Filename description (default: from prompt)\n'));
-
-  console.log(chalk.white('Examples:'));
-  console.log(chalk.gray('  # Generate hero image (auto-select service)'));
-  console.log(chalk.cyan('  node generate-visual-cli.js "Modern developer workspace"\n'));
-
-  console.log(chalk.gray('  # Generate diagram with Gemini'));
-  console.log(chalk.cyan('  node generate-visual-cli.js "System architecture" --type=diagram --service=gemini\n'));
-
-  console.log(chalk.gray('  # Generate HD illustration with DALL-E'));
-  console.log(chalk.cyan('  node generate-visual-cli.js "AI brain network" --type=illustration --service=dalle --quality=hd\n'));
-
-  console.log(chalk.gray('  # Generate video'));
-  console.log(chalk.cyan('  node generate-visual-cli.js "Product demo" --type=video --service=veo\n'));
-
-  console.log(chalk.white('Services:'));
-  console.log(chalk.gray('  gemini     Gemini 2.5 Flash - Fast, $0.02/image'));
-  console.log(chalk.gray('  imagen     Imagen 3 - Premium quality, $0.03/image'));
-  console.log(chalk.gray('  dalle      DALL-E 3 - Creative, $0.04-0.12/image'));
-  console.log(chalk.gray('  veo        Veo 2 - Video generation, $0.10/video\n'));
-
-  console.log(chalk.white('Prerequisites:'));
-  console.log(chalk.gray('  1. Configure API keys: node src/scripts/configure-visual-apis.js'));
-  console.log(chalk.gray('  2. Or add to .env: GOOGLE_API_KEY=xxx and/or OPENAI_API_KEY=xxx\n'));
-
-  console.log(chalk.white('Documentation:'));
-  console.log(chalk.gray('  See VISUAL_CONTENT_GENERATION_GUIDE.md for complete guide\n'));
+	console.log(chalk.blue("\n🎨 Visual Content Generation CLI\n"));
+
+	console.log(chalk.white("Usage:"));
+	console.log(chalk.gray('  node generate-visual-cli.js "prompt" [options]\n'));
+
+	console.log(chalk.white("Options:"));
+	console.log(
+		chalk.gray("  --type=<type>         Content type (default: hero)"),
+	);
+	console.log(
+		chalk.gray(
+			"                        Values: hero, illustration, diagram, screenshot, video",
+		),
+	);
+	console.log(
+		chalk.gray("  --service=<service>   AI service (default: auto-select)"),
+	);
+	console.log(
+		chalk.gray(
+			"                        Values: gemini, imagen, dalle, veo, nano-banana-pro, flux-pro, flux-dev",
+		),
+	);
+	console.log(
+		chalk.gray("  --size=<size>         Image size (default: 1024x1024)"),
+	);
+	console.log(
+		chalk.gray(
+			"                        Values: 1024x1024, 1792x1024, 1024x1792",
+		),
+	);
+	console.log(
+		chalk.gray("  --quality=<quality>   Quality level (default: standard)"),
+	);
+	console.log(chalk.gray("                        Values: standard, hd"));
+	console.log(
+		chalk.gray(
+			"  --description=<desc>  Filename description (default: from prompt)\n",
+		),
+	);
+
+	console.log(chalk.white("Examples:"));
+	console.log(chalk.gray("  # Generate hero image (auto-select service)"));
+	console.log(
+		chalk.cyan('  node generate-visual-cli.js "Modern developer workspace"\n'),
+	);
+
+	console.log(chalk.gray("  # Generate diagram with Gemini"));
+	console.log(
+		chalk.cyan(
+			'  node generate-visual-cli.js "System architecture" --type=diagram --service=gemini\n',
+		),
+	);
+
+	console.log(chalk.gray("  # Generate HD illustration with DALL-E"));
+	console.log(
+		chalk.cyan(
+			'  node generate-visual-cli.js "AI brain network" --type=illustration --service=dalle --quality=hd\n',
+		),
+	);
+
+	console.log(chalk.gray("  # Generate video"));
+	console.log(
+		chalk.cyan(
+			'  node generate-visual-cli.js "Product demo" --type=video --service=veo\n',
+		),
+	);
+
+	console.log(chalk.white("Services:"));
+	console.log(chalk.gray("  gemini     Gemini 2.5 Flash - Fast, $0.02/image"));
+	console.log(
+		chalk.gray("  imagen     Imagen 3 - Premium quality, $0.03/image"),
+	);
+	console.log(chalk.gray("  dalle      DALL-E 3 - Creative, $0.04-0.12/image"));
+	console.log(
+		chalk.gray("  veo        Veo 2 - Video generation, $0.10/video\n"),
+	);
+
+	console.log(chalk.white("Prerequisites:"));
+	console.log(
+		chalk.gray(
+			"  1. Configure API keys: node src/scripts/configure-visual-apis.js",
+		),
+	);
+	console.log(
+		chalk.gray(
+			"  2. Or add to .env: GOOGLE_API_KEY=xxx and/or OPENAI_API_KEY=xxx\n",
+		),
+	);
+
+	console.log(chalk.white("Documentation:"));
+	console.log(
+		chalk.gray("  See VISUAL_CONTENT_GENERATION_GUIDE.md for complete guide\n"),
+	);
 }