generation counts

frontend/libs/model-list/src/lib/legacy/Models.ts

@@ -134,6 +134,21 @@ const ALL_MODELS: ModelConfig[] = [
     },
     tags: [ModelTag.InstructiveEdit, ModelTag.NonMaskedInpainting],
   }),
+  cfg({
+    id: "nano_banana_pro",
+    category: "image",
+    info: {
+      name: "Nano Banana Pro",
+      tauri_id: "nano_banana_pro",
+      creator: mc.Google,
+    },
+    description: "Extremely powerful",
+    badges: [{ label: "35 sec." }],
+    capabilities: {
+      maxGenerationCount: 4,
+    },
+    tags: [ModelTag.InstructiveEdit, ModelTag.NonMaskedInpainting],
+  }),
   cfg({
     id: "flux_pro_inpaint",
     category: "image",

frontend/libs/model-list/src/lib/lists/ImageModels.ts

@@ -2,6 +2,8 @@ import { ImageModel } from "../classes/ImageModel.js";
 import { ModelCreator } from "../classes/metadata/ModelCreator.js";
 import { ModelTag } from "../classes/metadata/ModelTag.js";
 
+// TODO: Some of the model configs, such as generation counts, are authoritatively controlled in `legacy/Models.ts`
+
 export const IMAGE_MODELS: ImageModel[] = [
   new ImageModel({
     id: "grok_image",
@@ -124,7 +126,7 @@ export const IMAGE_MODELS: ImageModel[] = [
     defaultGenerationCount: 1,
     canEditImages: true,
     canUseImagePrompt: true,
-    maxImagePromptCount: 6,
+    maxImagePromptCount: 4,
     canTextToImage: true,
     tags: [ModelTag.InstructiveEdit],
     progressBarTime: 25000,
@@ -138,7 +140,7 @@ export const IMAGE_MODELS: ImageModel[] = [
     selectorName: "GPT Image 1 (GPT-4o)",
     selectorDescription: "Slow, but super smart",
     selectorBadges: ["60 sec."],
-    maxGenerationCount: 1,
+    maxGenerationCount: 4,
     defaultGenerationCount: 1,
     tags: [ModelTag.InstructiveEdit],
     canEditImages: true,