Update app.py

app.py

@@ -1,7 +1,89 @@
 import gradio as gr
 from PIL import Image
 
+def sort_doc(in_list: list, steps_in: int, control: int, prev_list: str):
+    control_json={'control':'0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ','char':'','leng':62}
+    key_cnt=len(in_list)
+    print(key_cnt)
 
+    control_char=list(control_json['control'])
+    char_len=len(control_char)    
+    if not steps_in:
+        n_cnt=0
+        nx=key_cnt
+        while True:
+            if nx >= 1:
+                n_cnt+=1
+                nx = nx/char_len
+            else:
+                steps=n_cnt
+                break    
+    if steps_in:
+        steps=steps_in
+    if control:
+        control_len=control_json['leng']-control
+        control_char=list(control_json['control'][:control_len])
+        control_val=list(control_json['control'][control_len:])
+        val_len=len(control_val)
+        control_val_box=[]
+        for ea in control_val:
+            control_val_box.append(ea)
+        print(f'CONTROL_VAL_BOX:: {control_val_box}')
+        #prev_list=control_val_box
+    json_out={}
+    
+    big_cnt=0
+    cnt=0
+    go=True
+    step_cont_box=[]
+
+    if prev_list:
+        print("LOD")
+        last_key=list(prev_list.keys())[-1]
+        print(last_key)
+        for ea_dig in last_key:
+            ea_dig=control_json['control'].index(ea_dig)
+            ea_dig=int(ea_dig)
+            print(f'{ea_dig} :: {list(control_json["control"][ea_dig])[0]}')
+            #step_cont_box.append(int(list(control_json["control"][ea_dig])[0]))
+            step_cont_box.append(ea_dig)
+            print(step_cont_box)
+        cnt=int(step_cont_box[-1])+1
+    if not prev_list:
+        print("NOT LOD")
+        for ii in range(steps):
+            print(ii)
+            step_cont_box.append(0)
+
+    pos=len(step_cont_box)-1
+    if go:
+        for i, ea in enumerate(in_list):
+            if go:
+                if cnt > char_len-1:
+                    #print(step_cont_box)
+                    go1=True
+                    for ii,ev in enumerate(step_cont_box):
+                        if go:
+                            if ev >= char_len-1:
+                                step_cont_box[ii]=0
+                                if go1==True:
+                                    step_cont_box[ii-1]=step_cont_box[ii-1]+1
+                                    go1=False
+                    cnt=1
+                else:
+                    step_cont_box[pos]=cnt
+                    cnt+=1
+                #print(step_cont_box)
+                out_js=""
+                for iii,j in enumerate(step_cont_box):
+                    #print(j)
+                    out_js = out_js+control_char[j]
+                json_out[out_js]=in_list[i]
+                big_cnt+=1
+                if big_cnt==key_cnt:
+                    print("DONE")
+                    go=False
+    return json_out
 
 
 def get_pixels(im):
@@ -15,7 +97,9 @@ def get_pixels(im):
     for i in range(im_width):    # for every col:
         for j in range(im_height):    # For every row
             pixel = pixels[i,j]
+            r,g,b=pixels[i,j]
             if str(pixel) in pixel_box:
+                print("YES")
                 pixel_list[str(pixel)].append((i,j))
             else: 
                 #pixel_box.append({f'{pixels[i,j]}':[(i,j)]})