download

app.py

@@ -133,17 +133,22 @@ def generate_colorbar_preview(color_name):
             rgb[..., 1] = high_half
             rgb[..., 2] = low_half
         else:
-            step_1 = np.clip(gradient * 3, 0, 1)
-            step_2 = np.clip((gradient - 0.333) * 3, 0, 1)
-            step_3 = np.clip((gradient - 0.666) * 3, 0, 1)
+            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 / 0.375, 0, 1)
+            step_2_cyan = np.clip((gradient - 0.375) / 0.5, 0, 1)
+            step_3_cyan = np.clip((gradient - 0.875) / 0.125, 0, 1)
+
             if color_name == "Red Hot":
-                rgb[..., 0] = step_1
-                rgb[..., 1] = step_2
-                rgb[..., 2] = step_3                
+                rgb[..., 0] = step_1_red
+                rgb[..., 1] = step_2_red
+                rgb[..., 2] = step_3_red                
             elif color_name == "Cyan Hot":
-                rgb[..., 2] = step_1
-                rgb[..., 1] = step_2
-                rgb[..., 0] = step_3
+                rgb[..., 2] = step_1_cyan
+                rgb[..., 1] = step_2_cyan
+                rgb[..., 0] = step_3_cyan
     
     elif color_name == "Green (GFP)": rgb[..., 1] = gradient
     elif color_name == "Red (RFP)": rgb[..., 0] = gradient
@@ -188,17 +193,22 @@ def apply_pseudocolor(image_np, color_name="Grayscale"):
             rgb[..., 1] = high_half
             rgb[..., 2] = low_half
         else:
-            step_1 = np.clip(norm_img * 3, 0, 1)
-            step_2 = np.clip((norm_img - 0.333) * 3, 0, 1)
-            step_3 = np.clip((norm_img - 0.666) * 3, 0, 1)
+            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 / 0.375, 0, 1)
+            step_2_cyan = np.clip((norm_img - 0.375) / 0.5, 0, 1)
+            step_3_cyan = np.clip((norm_img - 0.875) / 0.125, 0, 1)
+
             if color_name == "Red Hot":
-                rgb[..., 0] = step_1
-                rgb[..., 1] = step_2
-                rgb[..., 2] = step_3                
+                rgb[..., 0] = step_1_red
+                rgb[..., 1] = step_2_red
+                rgb[..., 2] = step_3_red                
             elif color_name == "Cyan Hot":
-                rgb[..., 2] = step_1
-                rgb[..., 1] = step_2
-                rgb[..., 0] = step_3
+                rgb[..., 2] = step_1_cyan
+                rgb[..., 1] = step_2_cyan
+                rgb[..., 0] = step_3_cyan
     
     elif color_name == "Green (GFP)": rgb[..., 1] = norm_img
     elif color_name == "Red (RFP)": rgb[..., 0] = norm_img
@@ -615,7 +625,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=CUSTOM_CSS) as demo:
                     t2i_btn = gr.Button("Generate", variant="primary")
                 with gr.Column(scale=2):
                     # Changed: Image to Gallery
-                    t2i_gallery_out = gr.Gallery(label="Generated Images", columns=3, height="auto")
+                    t2i_gallery_out = gr.Gallery(label="Generated Images", columns=3, height="auto", show_download_button=False, show_share_button=False)
                     with gr.Row(equal_height=True):
                         with gr.Column(scale=2):
                             t2i_color = gr.Dropdown(choices=COLOR_MAPS, value="Green (GFP)", label="Pseudocolor (Adjust after generation)")