Create app.py

app.py

@@ -0,0 +1,62 @@
+from sklearn.cluster import KMeans
+from collections import Counter
+import numpy as np
+import cv2
+import os
+import gradio as gr
+
+def get_image(pil_image):
+    nimg = np.array(pil_image)
+    image = cv2.cvtColor(nimg, cv2.COLOR_RGB2BGR)
+    image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
+    return image
+
+def label_image(rimg):
+  clf = KMeans(n_clusters = 6)
+  labels = clf.fit_predict(rimg)
+  return labels ,  clf
+
+def RGB(color):
+    return "#{:02x}{:02x}{:02x}".format(int(color[0]), int(color[1]), int(color[2]))
+
+def extract_colors(pimg):
+  img = get_image(pimg)
+  reshaped_img = img.reshape(img.shape[0]*img.shape[1], img.shape[2])
+  labels, clf = label_image(reshaped_img)
+  counts = Counter(labels)
+  center_colors = clf.cluster_centers_
+  ordered_colors = [center_colors[i] for i in counts.keys()]
+  hex_colors = [RGB(ordered_colors[i]) for i in counts.keys()]
+  return hex_colors[0], hex_colors[0], hex_colors[1], hex_colors[1], hex_colors[2],hex_colors[2], hex_colors[3], hex_colors[3], hex_colors[4], hex_colors[4], hex_colors[5], hex_colors[5]
+
+demo = gr.Blocks(title="Color Extraction")
+
+with demo:
+    gr.Markdown(
+        """
+    # Color Extractor from Images
+    """ 
+    )
+    with gr.Row():
+        with gr.Column():
+            inputs = [gr.Image(type="pil", label="")]
+            examples = gr.Examples(inputs=inputs, examples=[
+                        [os.path.join(os.path.dirname(__file__), "01.jpg")],
+                        [os.path.join(os.path.dirname(__file__), "02.jpg")],
+                        [os.path.join(os.path.dirname(__file__), "03.png")],
+                    ])
+        with gr.Column():
+            with gr.Row():
+                outputs = [ gr.ColorPicker(label="color 1"), gr.Label(label="color 1"),
+                            gr.ColorPicker(label="color 2"), gr.Label(label="color 2"),
+                            gr.ColorPicker(label="color 3"), gr.Label(label="color 3"),
+                            gr.ColorPicker(label="color 4"), gr.Label(label="color 4"),
+                            gr.ColorPicker(label="color 5"), gr.Label(label="color 5"),
+                            gr.ColorPicker(label="color 6"), gr.Label(label="color 6")]    
+
+    btn = gr.Button("Extract colors",variant="primary")
+    btn.click(fn=extract_colors, inputs=inputs, outputs=outputs)
+
+demo.queue()
+demo.launch()
+