diff --git a/src/types/models.ts b/src/types/models.ts
index 04157ff2..f8928ca9 100644
--- a/src/types/models.ts
+++ b/src/types/models.ts
@@ -18,8 +18,15 @@ export type AspectRatio =
   | "16:9"
   | "21:9";
 
-// Resolution Options (only supported by Nano Banana Pro)
-export type Resolution = "1K" | "2K" | "4K";
+// Resolution Options (supported by Nano Banana Pro and Nano Banana 2)
+export type Resolution = "512" | "1K" | "2K" | "4K";
 
 // Image Generation Model Options
-export type ModelType = "nano-banana" | "nano-banana-pro";
+export type ModelType = "nano-banana" | "nano-banana-pro" | "nano-banana-2";
+
+// Display names for image generation models
+export const MODEL_DISPLAY_NAMES: Record<ModelType, string> = {
+  "nano-banana": "Nano Banana",
+  "nano-banana-pro": "Nano Banana Pro",
+  "nano-banana-2": "Nano Banana 2",
+};