fix image model dropdown on 3d

This commit is contained in:
Brandon Thomas committed 2025-07-16 22:47:08 -04:00
1 parent 933399e15c
commit f9c43c2f5a
2 files changed
+20 -2

No files matched your search

@@ -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]"
@@ -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;