fix: StrictTransfer

.gitignore

@@ -1,3 +1,4 @@
 node/
 __pycache__/
-.env/
+.env/
+result.jpg

recolorStrictTransfer.py

@@ -4,7 +4,7 @@ import cv2
 from sklearn.cluster import MiniBatchKMeans
 
 def quantize_global(image, k):
-  k_means = MiniBatchKMeans(n_clusters=k, compute_labels=False)
+  k_means = MiniBatchKMeans(n_clusters=k, compute_labels=False, n_init=1)
   k_means.fit(image.reshape(-1, 1))
   labels = k_means.predict(image.reshape(-1, 1))
   return np.uint8(k_means.cluster_centers_[labels]).reshape(image.shape)

server.py

@@ -77,14 +77,13 @@ async def color_palettes(colors: str = Form(...)):
     return {"inputColor": colors, "complementaryColors": complementaryColors, "adjacentColors": adjacentColors, "gradientColors": gradientColors}
 
 @app.post("/recolor/")
-async def recolor(file: UploadFile = File(...), colors: str = Form(...), new_colors: str = Form(...), random_colors: bool = Form(False),  model: str = Form(...), mask: Optional[UploadFile] = File(None)):
+async def recolor(file: UploadFile = File(...), colors: str = Form(...), new_colors: Optional[str] = Form(None), random_colors: bool = Form(False),  model: str = Form(...), mask: Optional[UploadFile] = File(None)):
     """
     Receive an image file and an array of strings representing colors of a selected pallete and recolor an image.
     """
     method = model
     invertColors = False
     colors = [color.strip() for color in colors.split(',')]
-    new_colors = [new_color.strip() for new_color in new_colors.split(',')]
     file_content = await file.read()
     image_bytes = io.BytesIO(file_content)
     image = Image.open(image_bytes)
@@ -92,6 +91,9 @@ async def recolor(file: UploadFile = File(...), colors: str = Form(...), new_col
         image = ImageOps.invert(image)
     image_np = np.array(image)
 
+    if new_colors is not None:
+        new_colors = [new_color.strip() for new_color in new_colors.split(',')]
+
     if method == "CCA":
         print('CCA generated')
         #Characteristic Color Analysis
@@ -125,10 +127,12 @@ async def recolor(file: UploadFile = File(...), colors: str = Form(...), new_col
         print('LinearColorTransfer generated')
         recolorLinearColorTransfer.recolor(image_np, colors)
 
-    elif method == "StrictTransfer":
-        print('StrictTransfer started')
+    elif method.startswith("StrictTransfer"):
+        print('StrictTransfer started', colors, new_colors)
         if random_colors:
+            random.shuffle(colors)            
             random.shuffle(new_colors)            
+            print('StrictTransfer random', colors, new_colors)
         recolorStrictTransfer.recolor(image_np, colors, new_colors)        
         
     elif method == "ColorMask":