Update app.py

app.py

@@ -348,6 +348,9 @@ def switch_rows(v):
         return frames
 
 def draw_mask(f, n, c, v, d, evt: gr.EventData):
+    global frame_selected
+    global masks
+    
     print(v)
     points = json.loads(v)
     pts = np.array(points, np.int32)
@@ -363,6 +366,8 @@ def draw_mask(f, n, c, v, d, evt: gr.EventData):
 
     if len(d["layers"]) == 1:
         d["layers"].append(np.zeros_like(d["layers"][0]))
+    else:
+        d["background"] = masks[frame_selected]
 
     #indices = np.arange(0,256)   # List of all colors 
     #divider = np.linspace(0,255,n+1)[1] # we get a divider