From 975ff2672db3b41c450ea96cddb7165cd8d09127 Mon Sep 17 00:00:00 2001
From: Will Miao <13051207myq@gmail.com>
Date: Thu, 28 Aug 2025 16:24:01 +0800
Subject: [PATCH] feat: Add new Flux model 'FLUX_1_KREA' and update Video
 Models list for enhanced model support

---
 static/js/components/shared/ModelMetadata.js | 15 ++++--
 static/js/utils/constants.js                 | 51 +-------------------
 2 files changed, 12 insertions(+), 54 deletions(-)

diff --git a/static/js/components/shared/ModelMetadata.js b/static/js/components/shared/ModelMetadata.js
index 960ffa7d..d3e3de76 100644
--- a/static/js/components/shared/ModelMetadata.js
+++ b/static/js/components/shared/ModelMetadata.js
@@ -177,13 +177,20 @@ export function setupBaseModelEditing(filePath) {
             'Stable Diffusion 2.x': [BASE_MODELS.SD_2_0, BASE_MODELS.SD_2_1],
             'Stable Diffusion 3.x': [BASE_MODELS.SD_3, BASE_MODELS.SD_3_5, BASE_MODELS.SD_3_5_MEDIUM, BASE_MODELS.SD_3_5_LARGE, BASE_MODELS.SD_3_5_LARGE_TURBO],
             'SDXL': [BASE_MODELS.SDXL, BASE_MODELS.SDXL_LIGHTNING, BASE_MODELS.SDXL_HYPER],
-            'Video Models': [BASE_MODELS.SVD, BASE_MODELS.LTXV, BASE_MODELS.WAN_VIDEO, BASE_MODELS.HUNYUAN_VIDEO],
+            'Video Models': [
+                BASE_MODELS.SVD, BASE_MODELS.LTXV, BASE_MODELS.HUNYUAN_VIDEO, BASE_MODELS.WAN_VIDEO, 
+                BASE_MODELS.WAN_VIDEO_1_3B_T2V, BASE_MODELS.WAN_VIDEO_14B_T2V,
+                BASE_MODELS.WAN_VIDEO_14B_I2V_480P, BASE_MODELS.WAN_VIDEO_14B_I2V_720P,
+                BASE_MODELS.WAN_VIDEO_2_2_TI2V_5B, BASE_MODELS.WAN_VIDEO_2_2_T2V_A14B,
+                BASE_MODELS.WAN_VIDEO_2_2_I2V_A14B
+            ],
+            'Flux Models': [BASE_MODELS.FLUX_1_D, BASE_MODELS.FLUX_1_S, BASE_MODELS.FLUX_1_KONTEXT, BASE_MODELS.FLUX_1_KREA],
             'Other Models': [
-                BASE_MODELS.FLUX_1_D, BASE_MODELS.FLUX_1_S, BASE_MODELS.FLUX_1_KONTEXT, BASE_MODELS.AURAFLOW,
+                BASE_MODELS.ILLUSTRIOUS, BASE_MODELS.PONY, BASE_MODELS.HIDREAM,
+                BASE_MODELS.QWEN, BASE_MODELS.AURAFLOW,
                 BASE_MODELS.PIXART_A, BASE_MODELS.PIXART_E, BASE_MODELS.HUNYUAN_1,
                 BASE_MODELS.LUMINA, BASE_MODELS.KOLORS, BASE_MODELS.NOOBAI,
-                BASE_MODELS.ILLUSTRIOUS, BASE_MODELS.PONY, BASE_MODELS.HIDREAM,
-                BASE_MODELS.QWEN, BASE_MODELS.UNKNOWN
+                BASE_MODELS.UNKNOWN
             ]
         };
         
diff --git a/static/js/utils/constants.js b/static/js/utils/constants.js
index 3cb3dc6e..9f440aa5 100644
--- a/static/js/utils/constants.js
+++ b/static/js/utils/constants.js
@@ -24,6 +24,7 @@ export const BASE_MODELS = {
     // Other models
     FLUX_1_D: "Flux.1 D",
     FLUX_1_S: "Flux.1 S",
+    FLUX_1_KREA: "Flux.1 Krea",
     FLUX_1_KONTEXT: "Flux.1 Kontext",
     AURAFLOW: "AuraFlow",
     PIXART_A: "PixArt a",
@@ -53,56 +54,6 @@ export const BASE_MODELS = {
     UNKNOWN: "Other"
 };
 
-// Base model display names and their corresponding class names (for styling)
-export const BASE_MODEL_CLASSES = {
-    // Stable Diffusion 1.x models
-    [BASE_MODELS.SD_1_4]: "sd-1-4",
-    [BASE_MODELS.SD_1_5]: "sd-1-5",
-    [BASE_MODELS.SD_1_5_LCM]: "sd-1-5-lcm",
-    [BASE_MODELS.SD_1_5_HYPER]: "sd-1-5-hyper",
-    
-    // Stable Diffusion 2.x models
-    [BASE_MODELS.SD_2_0]: "sd-2-0",
-    [BASE_MODELS.SD_2_1]: "sd-2-1",
-    
-    // Stable Diffusion 3.x models
-    [BASE_MODELS.SD_3]: "sd-3",
-    [BASE_MODELS.SD_3_5]: "sd-3-5",
-    [BASE_MODELS.SD_3_5_MEDIUM]: "sd-3-5-medium",
-    [BASE_MODELS.SD_3_5_LARGE]: "sd-3-5-large",
-    [BASE_MODELS.SD_3_5_LARGE_TURBO]: "sd-3-5-large-turbo",
-    
-    // SDXL models
-    [BASE_MODELS.SDXL]: "sdxl",
-    [BASE_MODELS.SDXL_LIGHTNING]: "sdxl-lightning",
-    [BASE_MODELS.SDXL_HYPER]: "sdxl-hyper",
-    
-    // Video models
-    [BASE_MODELS.SVD]: "svd",
-    [BASE_MODELS.LTXV]: "ltxv",
-    [BASE_MODELS.WAN_VIDEO]: "wan-video",
-    [BASE_MODELS.HUNYUAN_VIDEO]: "hunyuan-video",
-    
-    // Other models
-    [BASE_MODELS.FLUX_1_D]: "flux-d",
-    [BASE_MODELS.FLUX_1_S]: "flux-s",
-    [BASE_MODELS.FLUX_1_KONTEXT]: "flux-kontext",
-    [BASE_MODELS.AURAFLOW]: "auraflow",
-    [BASE_MODELS.PIXART_A]: "pixart-a",
-    [BASE_MODELS.PIXART_E]: "pixart-e",
-    [BASE_MODELS.HUNYUAN_1]: "hunyuan-1",
-    [BASE_MODELS.LUMINA]: "lumina",
-    [BASE_MODELS.KOLORS]: "kolors",
-    [BASE_MODELS.NOOBAI]: "noobai",
-    [BASE_MODELS.ILLUSTRIOUS]: "il",
-    [BASE_MODELS.PONY]: "pony",
-    [BASE_MODELS.HIDREAM]: "hidream",
-    [BASE_MODELS.QWEN]: "qwen",
-    
-    // Default
-    [BASE_MODELS.UNKNOWN]: "unknown"
-};
-
 // Path template constants for download organization
 export const DOWNLOAD_PATH_TEMPLATES = {
     FLAT: {