From 62947529b164d5df26aaa9f5469020462bc704f5 Mon Sep 17 00:00:00 2001
From: Brandon Thomas <echelon@gmail.com>
Date: Sat, 6 Dec 2025 04:46:13 -0500
Subject: [PATCH] generation counts

---
 frontend/libs/model-list/src/lib/legacy/Models.ts | 15 +++++++++++++++
 .../libs/model-list/src/lib/lists/ImageModels.ts  |  6 ++++--
 2 files changed, 19 insertions(+), 2 deletions(-)

diff --git a/frontend/libs/model-list/src/lib/legacy/Models.ts b/frontend/libs/model-list/src/lib/legacy/Models.ts
index 47913e7e92..e1cff5d57b 100644
--- a/frontend/libs/model-list/src/lib/legacy/Models.ts
+++ b/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",
diff --git a/frontend/libs/model-list/src/lib/lists/ImageModels.ts b/frontend/libs/model-list/src/lib/lists/ImageModels.ts
index 667eda9f48..d18a976c8e 100644
--- a/frontend/libs/model-list/src/lib/lists/ImageModels.ts
+++ b/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,