From f9c43c2f5add00eeb68bbef5be13a21c97466dbd Mon Sep 17 00:00:00 2001
From: Brandon Thomas <echelon@gmail.com>
Date: Wed, 16 Jul 2025 22:47:08 -0400
Subject: [PATCH] fix image model dropdown on 3d

---
 .../app/src/pages/PageEnigma/PageEditor.tsx    |  4 ++--
 .../model-selector/src/lib/model-lists.tsx     | 18 ++++++++++++++++++
 2 files changed, 20 insertions(+), 2 deletions(-)

diff --git a/frontend/apps/editor3d/app/src/pages/PageEnigma/PageEditor.tsx b/frontend/apps/editor3d/app/src/pages/PageEnigma/PageEditor.tsx
index 85f8f84e44..d0a672b9be 100644
--- a/frontend/apps/editor3d/app/src/pages/PageEnigma/PageEditor.tsx
+++ b/frontend/apps/editor3d/app/src/pages/PageEnigma/PageEditor.tsx
@@ -66,7 +66,7 @@ import {
   removeImageDropListener,
 } from "@storyteller/ui-gallery-modal";
 import {
-  imageGenerationModels,
+  instructiveImageEditModels,
   ModelCategory,
   // ModelCategory,
   ModelSelector,
@@ -536,7 +536,7 @@ export const PageEditor = () => {
 
                 <div className="absolute bottom-6 left-6 z-20 flex items-center gap-2">
                   <ModelSelector
-                    items={imageGenerationModels}
+                    items={instructiveImageEditModels}
                     category={ModelCategory.Editor3D}
                     panelTitle="Select Model"
                     panelClassName="min-w-[280px]"
diff --git a/frontend/libs/components/model-selector/src/lib/model-lists.tsx b/frontend/libs/components/model-selector/src/lib/model-lists.tsx
index 3be20c66c6..cbb5ad3d35 100644
--- a/frontend/libs/components/model-selector/src/lib/model-lists.tsx
+++ b/frontend/libs/components/model-selector/src/lib/model-lists.tsx
@@ -89,7 +89,25 @@ export const allModels = {
     //  badges: [{ label: "25 sec.", icon: <FontAwesomeIcon icon={faClock} /> }],
     //},
   ],
+  // These models are used for 2D and 3D editing and scene building.
+  instructiveImageEdits: [
+    {
+      label: "GPT-4o",
+      icon: <FontAwesomeIcon icon={faImage} className="h-4 w-4" />,
+      description: "High quality model",
+      badges: [{ label: "2 min.", icon: <FontAwesomeIcon icon={faClock} /> }],
+      modelInfo: IMAGE_MODELS.gpt_image_1,
+    },
+    {
+      label: "FLUX.1 Kontext",
+      icon: <FontAwesomeIcon icon={faImage} className="h-4 w-4" />,
+      description: "Fast and high-quality model",
+      badges: [{ label: "20 sec.", icon: <FontAwesomeIcon icon={faClock} /> }],
+      modelInfo: IMAGE_MODELS.gpt_image_1, // TODO: Wrong metadata!
+    },
+  ],
 };
 
 export const videoGenerationModels: ModelList = allModels.video;
 export const imageGenerationModels: ModelList = allModels.image;
+export const instructiveImageEditModels: ModelList = allModels.instructiveImageEdits;