Update app.py

app.py

@@ -1,15 +1,20 @@
 import gradio as gr
 import stone
-import tempfile, os
+import tempfile
+import os
 
-def get_colors(image):
-    # save upload to a temp file
+def get_palette(image):
+    """
+    Detects dominant skin colors in the uploaded image
+    and returns an HTML row of color swatches.
+    """
+    # Save the upload to a temp file
     with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp:
         image.save(tmp, format="PNG")
         tmp_path = tmp.name
 
     try:
-        # process and grab faces
+        # Run the stone classifier
         result = stone.process(tmp_path, image_type="auto", return_report_image=False)
     finally:
         os.remove(tmp_path)
@@ -18,23 +23,27 @@ def get_colors(image):
     if not faces:
         return "<p>No face detected.</p>"
 
-    # extract hex codes of dominant colors
+    # Extract the hex codes of the dominant colors
     colors = [c["color"] for c in faces[0]["dominant_colors"]]
-    # render as inline swatches
-    swatches = "".join(
-        f'<div style="background:{c}; width:30px; height:20px; '
+
+    # Build HTML swatches
+    swatches_html = "".join(
+        f'<div style="background:{hexcode}; width:50px; height:50px; '
         'display:inline-block; margin:2px; border-radius:4px;"></div>'
-        for c in colors
+        for hexcode in colors
     )
-    return swatches
+    return f"<div style='display:flex; flex-wrap:wrap;'>{swatches_html}</div>"
 
+# Define the Gradio interface
 with gr.Blocks() as demo:
-    gr.Markdown("# Dominant Skin‑Color Swatches")
-    img = gr.Image(type="pil", label="Upload Photo")
-    btn = gr.Button("Get Colors")
-    out = gr.HTML(label="Colors")
+    gr.Markdown("# 🎨 Skin Color Palette API")
+    gr.Markdown("Upload a clear portrait to extract a palette of your skin's dominant tones.")
+    with gr.Row():
+        inp = gr.Image(type="pil", label="Your Photo")
+        btn = gr.Button("Generate Palette")
+    out = gr.HTML(label="Color Palette")
 
-    btn.click(fn=get_colors, inputs=img, outputs=out)
+    btn.click(fn=get_palette, inputs=inp, outputs=out)
 
 if __name__ == "__main__":
     demo.launch()