Update app.py

app.py

@@ -1,5 +1,3 @@
-# app.py
-
 import gradio as gr
 import stone
 import tempfile
@@ -7,17 +5,17 @@ import os
 
 def get_primary_color(image):
     """
-    Detects the most dominant skin color in the first face detected
-    and returns a single color swatch plus its hex code.
+    Detects the single most dominant skin color in the first face detected
+    by selecting the cluster with the highest percentage.
     """
-    # Save upload to temp file
+    # Save uploaded image to a temp file
     tmp = tempfile.NamedTemporaryFile(suffix=".png", delete=False)
     image.save(tmp.name, format="PNG")
     tmp_path = tmp.name
     tmp.close()
 
     try:
-        # Run stone to get dominant colors
+        # Run stone to get dominant_colors
         result = stone.process(tmp_path, image_type="auto", return_report_image=False)
     except Exception as e:
         os.remove(tmp_path)
@@ -30,16 +28,18 @@ def get_primary_color(image):
     if not faces:
         return "<p>No face detected.</p>"
 
-    # Grab the first dominant color of the first face
-    dominant_colors = faces[0].get("dominant_colors", [])
-    if not dominant_colors:
-        return "<p>No skin color found.</p>"
+    # Get the list of (color, percent) for the first face
+    doms = faces[0].get("dominant_colors", [])
+    if not doms:
+        return "<p>No skin colors found.</p>"
 
-    hexcode = dominant_colors[0].get("color")
+    # Pick the one with max percent
+    best = max(doms, key=lambda c: c.get("percent", 0))
+    hexcode = best.get("color")
     if not hexcode:
         return "<p>No valid color found.</p>"
 
-    # Render a single swatch and show its hex code
+    # Render a single swatch + hex code
     swatch_html = (
         f"<div style='display:flex; align-items:center; gap:8px;'>"
         f"<div style='background:{hexcode}; width:50px; height:50px; "
@@ -50,8 +50,8 @@ def get_primary_color(image):
     return swatch_html
 
 with gr.Blocks() as demo:
-    gr.Markdown("## 🎯 Single Face Skin Color")
-    gr.Markdown("Upload a portrait and get the single most dominant skin-tone color.")
+    gr.Markdown("## 🎯 Primary Face Skin Color")
+    gr.Markdown("Upload a portrait and get the **most dominant** skin‑tone color.")
     with gr.Row():
         inp = gr.Image(type="pil", label="Your Photo")
         btn = gr.Button("Detect Color")