inst

app.py

@@ -121,6 +121,23 @@ COLOR_MAPS = [
     "Cyan Hot",
     "Magenta Hot"
 ]
+CYAN_HOT_POINTS = [
+    (0.00, (0,   0,   0)),     # 0% 亮度
+    (0.20, (0,   69,  143)),   # 20% 亮度
+    (0.50, (0,   183, 255)),   # 50% 亮度
+    (0.80, (89, 255, 255)),   # 80% 亮度
+    (1.00, (255, 255, 255)),   # 100% 亮度
+]
+def get_rgb_interpolation(val_norm, rgb_points):
+    x_pts = [p[0] for p in rgb_points]
+    r_pts = [p[1][0] / 255.0 for p in rgb_points] # 归一化到 0-1
+    g_pts = [p[1][1] / 255.0 for p in rgb_points]
+    b_pts = [p[1][2] / 255.0 for p in rgb_points]
+    
+    r = np.interp(val_norm, x_pts, r_pts)
+    g = np.interp(val_norm, x_pts, g_pts)
+    b = np.interp(val_norm, x_pts, b_pts)
+    return r, g, b
 
 # --- Helper Functions ---
 def sanitize_prompt_for_filename(prompt):
@@ -152,19 +169,13 @@ def generate_colorbar_preview(color_name):
             step_1_red = np.clip(gradient * 3, 0, 1)
             step_2_red = np.clip((gradient - 0.333) * 3, 0, 1)
             step_3_red = np.clip((gradient - 0.666) * 3, 0, 1)
-
-            step_1_cyan = np.clip(gradient * 3, 0, 1)
-            step_2_cyan = np.clip((gradient - 0.333) * 3, 0, 1)
-            step_3_cyan = np.clip((gradient - 0.666) * 3, 0, 1)
-
             if color_name == "Red Hot":
                 rgb[..., 0] = step_1_red
                 rgb[..., 1] = step_2_red
                 rgb[..., 2] = step_3_red                
             elif color_name == "Cyan Hot":
-                rgb[..., 2] = step_1_cyan
-                rgb[..., 1] = step_2_cyan
-                rgb[..., 0] = step_3_cyan
+                r, g, b = get_rgb_interpolation(gradient, CYAN_HOT_POINTS)
+                rgb = np.stack([r, g, b], axis=-1)
     
     elif color_name == "Green (GFP)": rgb[..., 1] = gradient
     elif color_name == "Red (RFP)": rgb[..., 0] = gradient
@@ -212,19 +223,13 @@ def apply_pseudocolor(image_np, color_name="Grayscale"):
             step_1_red = np.clip(norm_img * 3, 0, 1)
             step_2_red = np.clip((norm_img - 0.333) * 3, 0, 1)
             step_3_red = np.clip((norm_img - 0.666) * 3, 0, 1)
-
-            step_1_cyan = np.clip(norm_img * 3, 0, 1)
-            step_2_cyan = np.clip((norm_img - 0.333) * 3, 0, 1)
-            step_3_cyan = np.clip((norm_img - 0.666) * 3, 0, 1)
-
             if color_name == "Red Hot":
                 rgb[..., 0] = step_1_red
                 rgb[..., 1] = step_2_red
                 rgb[..., 2] = step_3_red                
             elif color_name == "Cyan Hot":
-                rgb[..., 2] = step_1_cyan
-                rgb[..., 1] = step_2_cyan
-                rgb[..., 0] = step_3_cyan
+                r, g, b = get_rgb_interpolation(norm_img, CYAN_HOT_POINTS)
+                rgb = np.stack([r, g, b], axis=-1)
     
     elif color_name == "Green (GFP)": rgb[..., 1] = norm_img
     elif color_name == "Red (RFP)": rgb[..., 0] = norm_img