Upload app.py with huggingface_hub

app.py

@@ -15,6 +15,9 @@ def extract_colors(img, num_colors):
     # Remove any grayscale or alpha if present
     if data.shape[1] == 4:
         data = data[:, :3]
+    elif data.shape[1] == 1:
+        # If grayscale, replicate to RGB
+        data = np.repeat(data, 3, axis=1)
     
     # Fit KMeans
     kmeans = KMeans(n_clusters=num_colors, random_state=42)
@@ -23,16 +26,20 @@ def extract_colors(img, num_colors):
     # Get cluster centers
     colors = kmeans.cluster_centers_.round().astype(int)
     
+    # Calculate percentages
+    labels = kmeans.labels_
+    percentages = [(np.sum(labels == i) / len(labels)) * 100 for i in range(num_colors)]
+    
     # Convert to hex
     hex_colors = ['#' + ''.join(f'{c:02x}' for c in color) for color in colors]
     
     # Generate HTML for palette
     html = '<div style="text-align: center;"><h2>Color Palette</h2><div style="display: flex; flex-wrap: wrap; justify-content: center; gap: 10px;">'
-    for hex_color in hex_colors:
+    for hex_color, percent in zip(hex_colors, percentages):
         html += f'''
         <div style="display: flex; flex-direction: column; align-items: center; gap: 5px;">
             <div style="background-color: {hex_color}; width: 80px; height: 80px; border: 2px solid #333; border-radius: 10px;"></div>
-            <span style="font-size: 12px; font-weight: bold;">{hex_color}</span>
+            <span style="font-size: 12px; font-weight: bold;">{hex_color} ({percent:.1f}%)</span>
         </div>
         '''
     html += '</div></div>'