From 8aaa22ab9994ec66b49f42fd9279103608daceec Mon Sep 17 00:00:00 2001
From: Oleg Proskurin <regx@usul.su>
Date: Sun, 14 Sep 2025 23:58:55 +0700
Subject: [PATCH] update: script

---
 src/generation/text-to-image.ts | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/src/generation/text-to-image.ts b/src/generation/text-to-image.ts
index 47af942..ec8464a 100644
--- a/src/generation/text-to-image.ts
+++ b/src/generation/text-to-image.ts
@@ -33,12 +33,12 @@ async function main() {
     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 = {
     responseModalities: ['IMAGE', 'TEXT'],
   };
-  const model = 'gemini-2.0-flash-preview-image-generation';
+  const model = 'gemini-2.5-flash-image-preview';
   const contents = [
     {
       role: 'user',
@@ -65,7 +65,7 @@ async function main() {
 
       content.parts?.forEach((part, index) => {
         if (part.inlineData) {
-          const fileName = `fantasy_warrior_${index}`;
+          const fileName = `fantasy_warrior_primary_${index}`;
           const inlineData = part.inlineData;
           const fileExtension = mime.getExtension(inlineData.mimeType || '');
           console.log(`Saving image: ${fileName}.${fileExtension}`);
@@ -80,13 +80,13 @@ async function main() {
   } catch (error) {
     console.error('Primary model failed:', error.message || error);
 
-    // Try fallback model
-    console.log('Trying fallback model (Nano Banana)...');
+    // Try fallback model (Imagen 4)
+    console.log('Trying fallback model (Imagen 4)...');
     try {
       const altResponse = await ai.models.generateContent({
-        model: 'gemini-2.5-flash-image-preview',
+        model: 'imagen-4.0-generate-001',
         config: {
-          responseModalities: ['IMAGE', 'TEXT']
+          responseModalities: ['IMAGE']
         },
         contents,
       });