Initial commit
This commit is contained in:
Zhongwei Li
349
templates/image-generation.ts
Normal file
349
templates/image-generation.ts
Normal file
@@ -0,0 +1,349 @@
|
||||
/**
|
||||
* OpenAI Images API - DALL-E 3 Generation Examples
|
||||
*
|
||||
* This template demonstrates:
|
||||
* - Basic image generation
|
||||
* - Quality settings (standard vs HD)
|
||||
* - Style options (vivid vs natural)
|
||||
* - Different sizes and formats
|
||||
* - Base64 output
|
||||
* - Saving images to disk
|
||||
*/
|
||||
|
||||
import OpenAI from 'openai';
|
||||
import fs from 'fs';
|
||||
|
||||
const openai = new OpenAI({
|
||||
apiKey: process.env.OPENAI_API_KEY,
|
||||
});
|
||||
|
||||
// =============================================================================
|
||||
// BASIC IMAGE GENERATION
|
||||
// =============================================================================
|
||||
|
||||
async function basicGeneration() {
|
||||
const image = await openai.images.generate({
|
||||
model: 'dall-e-3',
|
||||
prompt: 'A white siamese cat with striking blue eyes',
|
||||
size: '1024x1024',
|
||||
quality: 'standard',
|
||||
style: 'vivid',
|
||||
n: 1,
|
||||
});
|
||||
|
||||
console.log('Generated image URL:', image.data[0].url);
|
||||
console.log('Revised prompt:', image.data[0].revised_prompt);
|
||||
|
||||
return image.data[0].url;
|
||||
}
|
||||
|
||||
// =============================================================================
|
||||
// QUALITY COMPARISON
|
||||
// =============================================================================
|
||||
|
||||
async function qualityComparison() {
|
||||
const prompt = 'A futuristic city at sunset with flying cars';
|
||||
|
||||
// Standard quality (faster, cheaper)
|
||||
console.log('Generating standard quality image...');
|
||||
const standard = await openai.images.generate({
|
||||
model: 'dall-e-3',
|
||||
prompt,
|
||||
quality: 'standard',
|
||||
});
|
||||
|
||||
console.log('Standard quality URL:', standard.data[0].url);
|
||||
|
||||
// HD quality (finer details, more expensive)
|
||||
console.log('Generating HD quality image...');
|
||||
const hd = await openai.images.generate({
|
||||
model: 'dall-e-3',
|
||||
prompt,
|
||||
quality: 'hd',
|
||||
});
|
||||
|
||||
console.log('HD quality URL:', hd.data[0].url);
|
||||
|
||||
return { standard: standard.data[0].url, hd: hd.data[0].url };
|
||||
}
|
||||
|
||||
// =============================================================================
|
||||
// STYLE COMPARISON
|
||||
// =============================================================================
|
||||
|
||||
async function styleComparison() {
|
||||
const prompt = 'A mountain landscape with a lake';
|
||||
|
||||
// Vivid style (hyper-real, dramatic)
|
||||
console.log('Generating vivid style image...');
|
||||
const vivid = await openai.images.generate({
|
||||
model: 'dall-e-3',
|
||||
prompt,
|
||||
style: 'vivid',
|
||||
});
|
||||
|
||||
console.log('Vivid style URL:', vivid.data[0].url);
|
||||
|
||||
// Natural style (more realistic, less dramatic)
|
||||
console.log('Generating natural style image...');
|
||||
const natural = await openai.images.generate({
|
||||
model: 'dall-e-3',
|
||||
prompt,
|
||||
style: 'natural',
|
||||
});
|
||||
|
||||
console.log('Natural style URL:', natural.data[0].url);
|
||||
|
||||
return { vivid: vivid.data[0].url, natural: natural.data[0].url };
|
||||
}
|
||||
|
||||
// =============================================================================
|
||||
// DIFFERENT SIZES
|
||||
// =============================================================================
|
||||
|
||||
async function differentSizes() {
|
||||
const prompt = 'A minimalist logo for a tech company';
|
||||
|
||||
// Square
|
||||
const square = await openai.images.generate({
|
||||
model: 'dall-e-3',
|
||||
prompt,
|
||||
size: '1024x1024',
|
||||
});
|
||||
|
||||
console.log('Square (1024x1024):', square.data[0].url);
|
||||
|
||||
// Portrait
|
||||
const portrait = await openai.images.generate({
|
||||
model: 'dall-e-3',
|
||||
prompt,
|
||||
size: '1024x1792',
|
||||
});
|
||||
|
||||
console.log('Portrait (1024x1792):', portrait.data[0].url);
|
||||
|
||||
// Landscape
|
||||
const landscape = await openai.images.generate({
|
||||
model: 'dall-e-3',
|
||||
prompt,
|
||||
size: '1792x1024',
|
||||
});
|
||||
|
||||
console.log('Landscape (1792x1024):', landscape.data[0].url);
|
||||
|
||||
return { square, portrait, landscape };
|
||||
}
|
||||
|
||||
// =============================================================================
|
||||
// BASE64 OUTPUT
|
||||
// =============================================================================
|
||||
|
||||
async function base64Output() {
|
||||
const image = await openai.images.generate({
|
||||
model: 'dall-e-3',
|
||||
prompt: 'A cyberpunk street scene at night',
|
||||
response_format: 'b64_json',
|
||||
});
|
||||
|
||||
const base64Data = image.data[0].b64_json;
|
||||
|
||||
console.log('Base64 data length:', base64Data?.length);
|
||||
console.log('First 100 chars:', base64Data?.substring(0, 100));
|
||||
|
||||
// Convert to buffer and save
|
||||
if (base64Data) {
|
||||
const buffer = Buffer.from(base64Data, 'base64');
|
||||
fs.writeFileSync('generated-image.png', buffer);
|
||||
console.log('Image saved to: generated-image.png');
|
||||
}
|
||||
|
||||
return base64Data;
|
||||
}
|
||||
|
||||
// =============================================================================
|
||||
// DOWNLOAD AND SAVE IMAGE
|
||||
// =============================================================================
|
||||
|
||||
async function downloadAndSave(url: string, filename: string) {
|
||||
const response = await fetch(url);
|
||||
const arrayBuffer = await response.arrayBuffer();
|
||||
const buffer = Buffer.from(arrayBuffer);
|
||||
|
||||
fs.writeFileSync(filename, buffer);
|
||||
console.log(`Image saved to: ${filename}`);
|
||||
}
|
||||
|
||||
async function generateAndSave() {
|
||||
const image = await openai.images.generate({
|
||||
model: 'dall-e-3',
|
||||
prompt: 'A serene Japanese garden with cherry blossoms',
|
||||
size: '1024x1024',
|
||||
quality: 'hd',
|
||||
style: 'natural',
|
||||
});
|
||||
|
||||
const url = image.data[0].url;
|
||||
await downloadAndSave(url, 'japanese-garden.png');
|
||||
|
||||
return url;
|
||||
}
|
||||
|
||||
// =============================================================================
|
||||
// DETAILED PROMPT EXAMPLES
|
||||
// =============================================================================
|
||||
|
||||
async function detailedPrompts() {
|
||||
// Example 1: Specific art style
|
||||
const artStyle = await openai.images.generate({
|
||||
model: 'dall-e-3',
|
||||
prompt: 'An oil painting of a sunset over the ocean in the style of Claude Monet',
|
||||
style: 'natural',
|
||||
});
|
||||
|
||||
console.log('Art style result:', artStyle.data[0].url);
|
||||
|
||||
// Example 2: Detailed composition
|
||||
const detailed = await openai.images.generate({
|
||||
model: 'dall-e-3',
|
||||
prompt: 'A professional product photo of a smartwatch on a white marble surface, with soft studio lighting from the left, shallow depth of field, commercial photography style',
|
||||
quality: 'hd',
|
||||
style: 'natural',
|
||||
});
|
||||
|
||||
console.log('Detailed composition:', detailed.data[0].url);
|
||||
|
||||
// Example 3: Character design
|
||||
const character = await openai.images.generate({
|
||||
model: 'dall-e-3',
|
||||
prompt: 'A friendly robot character with round edges, bright blue and white colors, large expressive eyes, modern minimalist design, 3D render style',
|
||||
style: 'vivid',
|
||||
});
|
||||
|
||||
console.log('Character design:', character.data[0].url);
|
||||
|
||||
return { artStyle, detailed, character };
|
||||
}
|
||||
|
||||
// =============================================================================
|
||||
// ERROR HANDLING
|
||||
// =============================================================================
|
||||
|
||||
async function withErrorHandling() {
|
||||
try {
|
||||
const image = await openai.images.generate({
|
||||
model: 'dall-e-3',
|
||||
prompt: 'A beautiful landscape',
|
||||
});
|
||||
|
||||
return image.data[0].url;
|
||||
} catch (error: any) {
|
||||
if (error.status === 400) {
|
||||
console.error('Bad request - check your prompt for policy violations');
|
||||
} else if (error.status === 401) {
|
||||
console.error('Invalid API key');
|
||||
} else if (error.status === 429) {
|
||||
console.error('Rate limit exceeded - wait and retry');
|
||||
} else {
|
||||
console.error('Unexpected error:', error.message);
|
||||
}
|
||||
|
||||
throw error;
|
||||
}
|
||||
}
|
||||
|
||||
// =============================================================================
|
||||
// BATCH GENERATION (Sequential)
|
||||
// =============================================================================
|
||||
|
||||
async function batchGeneration() {
|
||||
const prompts = [
|
||||
'A red apple on a wooden table',
|
||||
'A blue butterfly on a flower',
|
||||
'A green forest path in autumn',
|
||||
];
|
||||
|
||||
const results = [];
|
||||
|
||||
for (const prompt of prompts) {
|
||||
console.log(`Generating: ${prompt}`);
|
||||
|
||||
const image = await openai.images.generate({
|
||||
model: 'dall-e-3',
|
||||
prompt,
|
||||
size: '1024x1024',
|
||||
quality: 'standard',
|
||||
});
|
||||
|
||||
results.push({
|
||||
prompt,
|
||||
url: image.data[0].url,
|
||||
revised_prompt: image.data[0].revised_prompt,
|
||||
});
|
||||
|
||||
// Wait 1 second between requests to avoid rate limits
|
||||
await new Promise(resolve => setTimeout(resolve, 1000));
|
||||
}
|
||||
|
||||
console.log(`Generated ${results.length} images`);
|
||||
return results;
|
||||
}
|
||||
|
||||
// =============================================================================
|
||||
// MAIN EXECUTION
|
||||
// =============================================================================
|
||||
|
||||
async function main() {
|
||||
console.log('=== OpenAI DALL-E 3 Image Generation Examples ===\n');
|
||||
|
||||
// Example 1: Basic generation
|
||||
console.log('1. Basic Image Generation:');
|
||||
await basicGeneration();
|
||||
console.log();
|
||||
|
||||
// Example 2: Quality comparison
|
||||
console.log('2. Quality Comparison (Standard vs HD):');
|
||||
await qualityComparison();
|
||||
console.log();
|
||||
|
||||
// Example 3: Style comparison
|
||||
console.log('3. Style Comparison (Vivid vs Natural):');
|
||||
await styleComparison();
|
||||
console.log();
|
||||
|
||||
// Example 4: Different sizes
|
||||
console.log('4. Different Sizes:');
|
||||
await differentSizes();
|
||||
console.log();
|
||||
|
||||
// Example 5: Base64 output
|
||||
console.log('5. Base64 Output and Save:');
|
||||
await base64Output();
|
||||
console.log();
|
||||
|
||||
// Example 6: Generate and save
|
||||
console.log('6. Generate and Save:');
|
||||
await generateAndSave();
|
||||
console.log();
|
||||
|
||||
// Example 7: Detailed prompts
|
||||
console.log('7. Detailed Prompt Examples:');
|
||||
await detailedPrompts();
|
||||
console.log();
|
||||
}
|
||||
|
||||
// Run if executed directly
|
||||
if (require.main === module) {
|
||||
main().catch(console.error);
|
||||
}
|
||||
|
||||
export {
|
||||
basicGeneration,
|
||||
qualityComparison,
|
||||
styleComparison,
|
||||
differentSizes,
|
||||
base64Output,
|
||||
generateAndSave,
|
||||
detailedPrompts,
|
||||
batchGeneration,
|
||||
withErrorHandling,
|
||||
};
|
||||
Reference in New Issue
Block a user