diff --git a/frontend/apps/artcraft/app/src/pages/PageDraw/PageDraw.tsx b/frontend/apps/artcraft/app/src/pages/PageDraw/PageDraw.tsx index 617c6a6bf0..ea66b0e8b9 100644 --- a/frontend/apps/artcraft/app/src/pages/PageDraw/PageDraw.tsx +++ b/frontend/apps/artcraft/app/src/pages/PageDraw/PageDraw.tsx @@ -575,7 +575,7 @@ const PageDraw = () => { page={PAGE_ID} panelTitle="Select Model" panelClassName="min-w-[300px]" - buttonClassName="bg-transparent p-0 text-lg hover:bg-transparent text-white/80 hover:text-white" + buttonClassName="bg-transparent p-0 text-lg hover:bg-transparent text-base-fg opacity-80 hover:opacity-100" showIconsInList triggerLabel="Model" /> diff --git a/frontend/libs/components/model-selector/src/lib/model-selectors-for-pages.tsx b/frontend/libs/components/model-selector/src/lib/model-selectors-for-pages.tsx index 7a0913db0b..c188c948be 100644 --- a/frontend/libs/components/model-selector/src/lib/model-selectors-for-pages.tsx +++ b/frontend/libs/components/model-selector/src/lib/model-selectors-for-pages.tsx @@ -18,7 +18,7 @@ const buildItems = ( models: Model[], fallbackIcon: any ) => - models.map((model : Model) => ({ + models.map((model: Model) => ({ label: model.selectorName, icon: withIcon(getCreatorIcon(model.creator), fallbackIcon), description: model.selectorDescription, @@ -33,14 +33,14 @@ const buildItems = ( /** * IN-PROGRESS MIGRATION (messy for now) * We're gradually going to phase out ModelList, ModelConfig, etc. - * We won't index by name, but rather id, or simply will always have full + * We won't index by name, but rather id, or simply will always have full * access to the object directly. - */ + */ -export const TEXT_TO_IMAGE_PAGE_MODEL_LIST : ModelList = +export const TEXT_TO_IMAGE_PAGE_MODEL_LIST: ModelList = buildItems( - (function() : Model[] { - const set : Set = new Set(); + (function (): Model[] { + const set: Set = new Set(); IMAGE_MODELS .filter((model) => model.canTextToImage) .forEach((m) => set.add(m)); @@ -51,10 +51,24 @@ export const TEXT_TO_IMAGE_PAGE_MODEL_LIST : ModelList = ); -export const CANVAS_2D_PAGE_MODEL_LIST : ModelList = +export const CANVAS_2D_PAGE_MODEL_LIST: ModelList = buildItems( - (function() : Model[] { - const set : Set = new Set(); + (function (): Model[] { + const set: Set = new Set(); + IMAGE_MODELS + .filter((m) => m.canEditImages || m.tags?.includes(ModelTag.InstructiveEdit)) + .forEach((m) => set.add(m)); + const list = Array.from(set); + list.sort((a, b) => a.selectorName?.localeCompare(b.selectorName)); + return list; + })(), + + ); + +export const STAGE_3D_PAGE_MODEL_LIST: ModelList = + buildItems( + (function (): Model[] { + const set: Set = new Set(); IMAGE_MODELS .filter((m) => m.tags?.includes(ModelTag.InstructiveEdit)) .forEach((m) => set.add(m)); @@ -65,29 +79,15 @@ export const CANVAS_2D_PAGE_MODEL_LIST : ModelList = ); -export const STAGE_3D_PAGE_MODEL_LIST : ModelList = - buildItems( - (function() : Model[] { - const set : Set = new Set(); - IMAGE_MODELS - .filter((m) => m.tags?.includes(ModelTag.InstructiveEdit)) - .forEach((m) => set.add(m)); - const list = Array.from(set); - list.sort((a, b) => a.selectorName?.localeCompare(b.selectorName)); - return list; - })(), - - ); - -export const IMAGE_EDITOR_PAGE_MODEL_LIST : ModelList = +export const IMAGE_EDITOR_PAGE_MODEL_LIST: ModelList = buildItems( //[ // ALL_MODELS_BY_ID.get("flux_pro_inpaint")!, // ALL_MODELS_BY_ID.get("flux_dev_juggernaut_inpaint")!, // ALL_MODELS_BY_ID.get("flux_pro_kontext_max")!, //], - (function() : Model[] { - const set : Set = new Set(); + (function (): Model[] { + const set: Set = new Set(); //set.add(IMAGE_MODELS_BY_ID.get("gpt_image_1")!); // Place gpt_image_1 first. IMAGE_MODELS .filter((m) => m.canEditImages) @@ -99,14 +99,14 @@ export const IMAGE_EDITOR_PAGE_MODEL_LIST : ModelList = ); -export const IMAGE_TO_VIDEO_PAGE_MODEL_LIST : ModelList = +export const IMAGE_TO_VIDEO_PAGE_MODEL_LIST: ModelList = buildItems( - (function() : Model[] { - const set : Set = new Set(); - VIDEO_MODELS.forEach((m) => set.add(m)); + (function (): Model[] { + const set: Set = new Set(); + VIDEO_MODELS.forEach((m) => set.add(m)); const list = Array.from(set); list.sort((a, b) => a.selectorName?.localeCompare(b.selectorName)); return list; })(), - ); \ No newline at end of file + );