adjusted chroma settings

flagged/log.csv

@@ -0,0 +1,2 @@
+Upload an Image,L% Min Skin,L% Max Skin,L% Min Tonality,L% Max Tonality,Chroma Threshold,Skin Analysis,Eye Analysis,Hair Analysis,Processed Image,Analysis Results,flag,username,timestamp
+flagged/Upload an Image/5517bdf8e63e3c9acf0f/Image from iOS.jpg,10,90,37,67,116,true,false,false,flagged/Processed Image/028ae1a27b4f747f5e2e/image.webp,"{""skin_analysis"": {""undertone"": ""Warm"", ""overtone"": ""Dark"", ""tonality"": ""Deep"", ""season"": ""Light Spring"", ""chroma"": ""Muted"", ""chroma_counts"": {""Bright"": 0, ""Muted"": 756205}, ""undertone_counts"": {""Warm"": 632189, ""Cool"": 124016, ""Neutral"": 0}, ""overtone_counts"": {""Very Light"": 60556, ""Light"": 73612, ""Medium"": 213157, ""Olive"": 173875, ""Dark"": 235005}, ""tonality_counts"": {""Light"": 249359, ""True"": 226941, ""Deep"": 279905}, ""season_counts"": {""Light Spring"": 0, ""True Spring"": 0, ""Warm Spring"": 0, ""Light Autumn"": 0, ""True Autumn"": 0, ""Warm Autumn"": 0, ""Light Summer"": 0, ""True Summer"": 0, ""Cool Summer"": 0, ""Light Winter"": 0, ""True Winter"": 0, ""Cool Winter"": 0, ""Unknown"": 0}}}",,,2024-06-19 11:53:23.491279

src/skin_analyzer.py

@@ -299,16 +299,20 @@ def categorize_chroma(lab_pixels, chroma_thresh):
     :return: Chroma category (Bright, Muted).
     """
     chroma = []
+    chroma_counts = {"Bright": 0, "Muted": 0}
     for pixel in lab_pixels:
         a = pixel[1]
         b = pixel[2]
         distance = np.sqrt(a**2 + b**2)
         if distance > chroma_thresh:
             chroma.append("Bright")
+            chroma_counts["Bright"] += 1
         else:
             chroma.append("Muted")
+            chroma_counts["Muted"] += 1
 
-    return chroma
+    predominant_chroma = max(chroma_counts, key=chroma_counts.get)
+    return chroma_counts, predominant_chroma, chroma
 
 
 def categorize_undertones(cluster_centers):
@@ -388,7 +392,9 @@ def analyze_skin_function(
     )
 
     # calculate chroma
-    chroma = categorize_chroma(lab_pixels, chroma_thresh)
+    chroma_counts, predominant_chroma, chroma = categorize_chroma(
+        lab_pixels, chroma_thresh
+    )
 
     season_counts, predominant_season, seasons = determine_season_with_tonality(
         undertones, chroma, tonalities
@@ -442,8 +448,10 @@ def analyze_skin_function(
         "tonality": predominant_tonality,
         # "ita": np.mean(ITA_values),
         "season": predominant_season,
+        "chroma": predominant_chroma,
         # "overlay_image": overlay,
         "filtered_skin_mask": filtered_skin_mask,
+        "chroma_counts": chroma_counts,
         "undertone_counts": undertone_counts,
         "overtone_counts": overtone_counts,
         "tonality_counts": tonality_counts,