Update app.py

app.py

@@ -11,14 +11,23 @@ def get_pixels(im):
     im_height=img.size[1]
     print(f'{im_width} :: {im_height}')
     pixel_box=[]
+    pixel_list={}
     for i in range(im_width):    # for every col:
         for j in range(im_height):    # For every row
-            pixel_box.append({"pixel":f'{pixels[i,j]}',"location":[i,j],"cnt":""})
+            pixel = pixels[i,j]
+            if pixel in pixel_box:
+                pixel_list[pixel].append((i,j))
+            else: 
+                #pixel_box.append({f'{pixels[i,j]}':[(i,j)]})
+                pixel_list[pixel]=[(i,j)]
+                pixel_box.append(pixel)
+            
             #pixels[i,j] = (i, j, 100) # set the colour accordingly
             #print(pixels[i,j])
     #for ea in pixel_box:
-        
-    return pixel_box[:20]
+    print(len(pixel_list))    
+    print(len(pixel_box))    
+    return pixel_list[:20]
 with gr.Blocks() as app:
     in_im=gr.Image(type='filepath')
     go_btn=gr.Button()