Changed the color indexing to 1-based

colorbynumber/main.py

@@ -107,9 +107,10 @@ class ColorByNumber:
             image[start_row:end_row, end_col] = border_color # Right Border
 
             # Draw text label below the square
-            text_size, _ = cv.getTextSize(str(i), font, font_size, 1)
+            text = str(i + 1)
+            text_size, _ = cv.getTextSize(text, font, font_size, 1)
             text_row = (end_row + text_size[1]) + 5
             text_col = start_col + (square_size // 2) - (text_size[0] // 2)
-            cv.putText(image, str(i), (text_col, text_row), font, font_size, (0, 0, 0), 1)
+            cv.putText(image, text, (text_col, text_row), font, font_size, (0, 0, 0), 1)
 
         return image

colorbynumber/simplify_image.py

@@ -26,6 +26,9 @@ def _choose_closest_colors(image, color_list):
     indices_color_choices = norm_diff.argmin(axis = -1)
     simplified_image = color_list[indices_color_choices.flatten(), :].reshape(image.shape)
 
+    # Adding 1 to indices_color_choices as so the first color is labeled as 1 and not 0.
+    indices_color_choices = indices_color_choices + 1
+
     return simplified_image, indices_color_choices
 
 def _denoise_image(image, h):