usulpro
/
magic-building
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update: script

This commit is contained in:
Oleg Proskurin 2025-09-14 23:58:55 +07:00
parent 2fc7bd7698
commit 8aaa22ab99
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/generation/text-to-image.ts
View File

@ -33,12 +33,12 @@ async function main() {
apiKey: process.env.GEMINI_API_KEY, apiKey: process.env.GEMINI_API_KEY,
}); });
// Note: Using gemini-2.0-flash-preview-image-generation with both IMAGE and TEXT response modalities // Note: Using gemini-2.5-flash-image-preview (Nano Banana) as primary model
const config = { const config = {
responseModalities: ['IMAGE', 'TEXT'], responseModalities: ['IMAGE', 'TEXT'],
}; };
const model = 'gemini-2.0-flash-preview-image-generation'; const model = 'gemini-2.5-flash-image-preview';
const contents = [ const contents = [
{ {
role: 'user', role: 'user',
@ -65,7 +65,7 @@ async function main() {
content.parts?.forEach((part, index) => { content.parts?.forEach((part, index) => {
if (part.inlineData) { if (part.inlineData) {
const fileName = `fantasy_warrior_${index}`; const fileName = `fantasy_warrior_primary_${index}`;
const inlineData = part.inlineData; const inlineData = part.inlineData;
const fileExtension = mime.getExtension(inlineData.mimeType || ''); const fileExtension = mime.getExtension(inlineData.mimeType || '');
console.log(`Saving image: ${fileName}.${fileExtension}`); console.log(`Saving image: ${fileName}.${fileExtension}`);
@ -80,13 +80,13 @@ async function main() {
} catch (error) { } catch (error) {
console.error('Primary model failed:', error.message || error); console.error('Primary model failed:', error.message || error);
// Try fallback model // Try fallback model (Imagen 4)
console.log('Trying fallback model (Nano Banana)...'); console.log('Trying fallback model (Imagen 4)...');
try { try {
const altResponse = await ai.models.generateContent({ const altResponse = await ai.models.generateContent({
model: 'gemini-2.5-flash-image-preview', model: 'imagen-4.0-generate-001',
config: { config: {
responseModalities: ['IMAGE', 'TEXT'] responseModalities: ['IMAGE']
}, },
contents, contents,
}); });