37 lines
1.1 KiB
TypeScript
Executable File
37 lines
1.1 KiB
TypeScript
Executable File
import ZAI from 'z-ai-web-dev-sdk';
|
|
import fs from 'fs';
|
|
|
|
async function main(imageSource: string, prompt: string, size: '1024x1024' | '768x1344' | '864x1152' | '1344x768' | '1152x864' | '1440x720' | '720x1440', outFile: string) {
|
|
try {
|
|
const zai = await ZAI.create();
|
|
|
|
const response = await zai.images.generations.edit({
|
|
prompt,
|
|
images: [{ url: imageSource }], // Array of objects with url property
|
|
size
|
|
});
|
|
|
|
const base64 = response?.data?.[0]?.base64;
|
|
if (!base64) {
|
|
console.error('No image data returned by the API');
|
|
console.log('Full response:', JSON.stringify(response, null, 2));
|
|
return;
|
|
}
|
|
|
|
const buffer = Buffer.from(base64, 'base64');
|
|
fs.writeFileSync(outFile, buffer);
|
|
console.log(`Edited image saved to ${outFile}`);
|
|
} catch (err: any) {
|
|
console.error('Image editing failed:', err?.message || err);
|
|
}
|
|
}
|
|
|
|
// Example usage - Edit an image
|
|
// You can use either a URL or a base64 data URL for the imageSource
|
|
main(
|
|
'https://example.com/photo.jpg', // or use: 'data:image/jpeg;base64,/9j/4AAQ...'
|
|
'Transform this photo to have a sunset background with warm golden tones',
|
|
'1024x1024',
|
|
'./output.png'
|
|
);
|