Update app.py

app.py

@@ -156,10 +156,18 @@ def remove_bg(fl):
     bg_diff = (bg-bg_gray).astype(np.int16)
     frame_c = (frame.astype(np.int16)-bg_diff).astype(np.uint8)
 
-    
+
+    blur_s = []
+    edges = cv2.Laplacian(cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY), cv2.CV_64F)
+    for i in range(0, frame.shape[0]):
+        blur_s.append([])
+        for j in range(0, frame.shape[1]):
+            d = edges[i-2:i+2,j-2:j+2].var().astype(np.uint8)
+            blur_s[i].append(d)
+            
     #remove regions of low saturation and lightness (get scene without shadow)
     m = cv2.inRange(cv2.cvtColor(frame_c, cv2.COLOR_RGB2HSV), np.array([0,0,0]), np.array([180,32,64]))
-    frame[m>0] = (127,127,127)
+    frame[m>0 and blur_s<255] = (127,127,127)
     frame = cv2.medianBlur(frame, b)