Update app.py

app.py

@@ -167,10 +167,14 @@ def extract_middle_slices(nifti_path, output_image_path, slice_size=180, center=
 
     # Choose colormap
     if label_components:
-        # Build a modified Pastel colormap with 0 = black
-        pastel = plt.cm.Pastel2(np.linspace(0, 1, 256))
-        pastel[0] = np.array([0, 0, 0, 1])  # Set the first color (label 0) to black (RGBA)
-        custom_cmap = mpl.colors.ListedColormap(pastel)
+        # Create 256 pastel colors
+        pastel = plt.cm.Pastel1(np.linspace(0, 1, 256))
+        np.random.seed(42)  # For reproducibility
+        shuffled_colors = pastel[1:].copy()
+        np.random.shuffle(shuffled_colors)
+        final_colors = np.vstack([np.array([0, 0, 0, 1]), shuffled_colors])
+ 
+        custom_cmap = mpl.colors.ListedColormap(final_colors)
         cmap = custom_cmap  # Colorful
         vmin = 0
         vmax = num_features