add shaded color pickers

myapp/pall_app.py

@@ -1,3 +1,5 @@
+from contextlib import ExitStack
+
 import gradio as gr
 import numpy as np
 
@@ -11,10 +13,16 @@ with gr.Blocks() as demo:
     @gr.render(inputs=[image, n_colors])
     def render_palette(image_array: np.ndarray, n_clusers: int):
         model = extract_color_clusters(image_array, n_clusers)
-        shades = iter_color_shades(model, (0, 0.2, 0.4, 0.6))
+        cluster_shades = iter_color_shades(model, (0, 0.2, 0.4, 0.6))
+
+        with ExitStack() as stack:
+            for i, cluster in enumerate(cluster_shades):
+                if i % n_clusers == 0:
+                    stack.pop_all().close()
+                    stack.enter_context(gr.Group())
+                    stack.enter_context(gr.Row())
 
-        for cluster in model.cluster_centers_:
-            gr.ColorPicker(array_to_hex(cluster))
+                gr.ColorPicker(array_to_hex(cluster))
 
 
 if __name__ == "__main__":