Update app.py

app.py

@@ -24,6 +24,7 @@ from moviepy.editor import *
 edge = []
 gradient = None
 params = { "fnum":0, "l":16 }
+dcolor = []
 frame_selected = 0
 frames = []
 depths = []
@@ -370,6 +371,7 @@ def show_json(txt):
 
 
 def select_frame(v, evt: gr.SelectData):
+    global dcolor
     global frame_selected
     global masks
     global edge
@@ -379,7 +381,7 @@ def select_frame(v, evt: gr.SelectData):
         frame_selected = evt.index
         edge = []
         
-    return masks[frame_selected], frame_selected
+    return masks[frame_selected], frame_selected, dcolor[frame_selected]
 
 def switch_rows(v):
     global frames
@@ -392,6 +394,8 @@ def switch_rows(v):
         return frames
 
 def optimize(v, d):
+    global dcolor
+    global frame_selected
     global frames
     global depths
     
@@ -411,17 +415,17 @@ def optimize(v, d):
             frame = cv2.imread(frames[k]).astype(np.uint8)
             frame = cv2.convertScaleAbs(palette[frame]).astype(np.uint8) # Converting image back to uint
             cv2.imwrite(frames[k], frame)
-            dcolor = bincount(frame)
-            print(dcolor)
+            dcolor[k] = bincount(frame)
+            print(dcolor[k])
             depth = cv2.imread(depths[k]).astype(np.uint8)
-            mask = cv2.inRange(frame, np.array([dcolor[0]-8, dcolor[1]-8, dcolor[2]-8]), np.array([dcolor[0]+8, dcolor[1]+8, dcolor[2]+8]))
+            mask = cv2.inRange(frame, np.array([dcolor[k][0]-8, dcolor[k][1]-8, dcolor[k][2]-8]), np.array([dcolor[k][0]+8, dcolor[k][1]+8, dcolor[k][2]+8]))
             depth[mask>0] = (255,255,255)
             cv2.imwrite(depths[k], depth)
             
     if d == False:
-      return frames, dcolor
+      return frames, dcolor[frame_selected]
     else:
-      return depths, dcolor
+      return depths, dcolor[frame_selected]
 
 def bincount(a):
     a2D = a.reshape(-1,a.shape[-1])
@@ -931,7 +935,7 @@ with gr.Blocks(css=css, js=js) as demo:
       <a id='move' href='#'>move</a>    <a id='rotate' href='#'>rotate</a>    <a id='scale' href='#'>scale</a>
                 </pre>""")
                 selected = gr.Number(elem_id="fnum", value=0, minimum=0, maximum=256, interactive=False)
-                output_frame.select(fn=select_frame, inputs=[output_mask], outputs=[output_mask, selected])
+                output_frame.select(fn=select_frame, inputs=[output_mask], outputs=[output_mask, selected, bgcolor])
                 example_coords = """[
                   {"lat": 50.07379596793083, "lng": 14.437146122950555, "heading": 152.70303, "pitch": 2.607833999999997}, 
                   {"lat": 50.073799567020004, "lng": 14.437146774240507, "heading": 151.12973, "pitch": 2.8672300000000064},