From cebf5f3c229ddd9d2f93f625845384fcd77bfd94 Mon Sep 17 00:00:00 2001
From: Brandon Thomas <echelon@gmail.com>
Date: Fri, 15 Aug 2025 01:49:50 -0400
Subject: [PATCH] update model dropdowns

---
 frontend/libs/model-list/src/lib/ModelMapping.ts |  3 +++
 frontend/libs/model-list/src/lib/Models.ts       | 14 ++++++++++++--
 2 files changed, 15 insertions(+), 2 deletions(-)

diff --git a/frontend/libs/model-list/src/lib/ModelMapping.ts b/frontend/libs/model-list/src/lib/ModelMapping.ts
index b9caa43bbb..fae9c5403c 100644
--- a/frontend/libs/model-list/src/lib/ModelMapping.ts
+++ b/frontend/libs/model-list/src/lib/ModelMapping.ts
@@ -16,6 +16,7 @@ export const MODEL_TYPE_TO_CREATOR: Record<string, ModelCreator> = {
   veo_2: ModelCreator.Google,
   recraft_3: ModelCreator.Recraft,
   hunyuan_3d: ModelCreator.Tencent,
+  midjourney: ModelCreator.Midjourney,
 };
 
 // Get creator icon for a model type
@@ -94,6 +95,7 @@ export const getModelDisplayName = (modelType: string): string => {
     hunyuan_3d_2p1: "Hunyuan 3D 2.1",
     hunyuan_3d: "Hunyuan 3D",
     flux_pro_kontext_max: "Flux Pro Kontext Max",
+    midjourney: "Midjourney",
     // TODO: add more models here - BFlat
   };
 
@@ -105,6 +107,7 @@ export const getProviderDisplayName = (provider: string): string => {
   const displayNames: Record<string, string> = {
     artcraft: "ArtCraft",
     fal: "FAL",
+    midjourney: "Midjourney",
     sora: "Sora",
   };
 
diff --git a/frontend/libs/model-list/src/lib/Models.ts b/frontend/libs/model-list/src/lib/Models.ts
index d2eed07bb0..eace454a88 100644
--- a/frontend/libs/model-list/src/lib/Models.ts
+++ b/frontend/libs/model-list/src/lib/Models.ts
@@ -57,7 +57,6 @@ export const ALL_MODELS: ModelConfig[] = [
     description: "High quality model",
     badges: [{ label: "15 sec." }],
     capabilities: { maxGenerationCount: 4 },
-    tags: ["instructiveEdit"],
   }),
   cfg({
     id: "flux_pro_1_1",
@@ -70,7 +69,6 @@ export const ALL_MODELS: ModelConfig[] = [
     description: "High quality model",
     badges: [{ label: "15 sec." }],
     capabilities: { maxGenerationCount: 4 },
-    tags: ["instructiveEdit"],
   }),
   cfg({
     id: "flux_1_dev",
@@ -122,6 +120,18 @@ export const ALL_MODELS: ModelConfig[] = [
     capabilities: { maxGenerationCount: 4 },
     tags: ["instructiveEdit"],
   }),
+  cfg({
+    id: "midjourney",
+    category: "image",
+    info: {
+      name: "Midjourney",
+      tauri_id: "midjourney",
+      creator: mc.Midjourney,
+    },
+    description: "Incredible style and quality",
+    badges: [{ label: "15 sec." }],
+    capabilities: { maxGenerationCount: 4 },
+  }),
 
   //////////////////////////////
   // Video models