Update app.py

app.py

@@ -180,10 +180,10 @@ def remove_bg(fl, count):
         fr_fg = np.bitwise_or(fr_diff, fr_mask).astype(np.uint8)
         cv2.imwrite(fl_, fr_mask)
 
-        #m = cv2.inRange(fr, np.array([240,240,240]), np.array([255,255,255]))
-        #fr[m>0] = (239,239,239)
-        #m = cv2.inRange(fr, np.array([0,0,0]), np.array([15,15,15]))
-        #fr[m>0] = (16,16,16)
+        m = cv2.inRange(fr, np.array([240,240,240]), np.array([255,255,255]))
+        fr[m>0] = (239,239,239)
+        m = cv2.inRange(fr, np.array([0,0,0]), np.array([15,15,15]))
+        fr[m>0] = (16,16,16)
         fr[fr_shadow>0] = (fr[fr_shadow>0] / 17).astype(np.uint8)
         fr[fr_fg==0] = (255,255,255)
         
@@ -243,8 +243,11 @@ def loadf(f, r_bg):
 
             if r_bg == True:
                 fl = remove_bg(fl, i)
+                if i % 2: # odd: is photo without the flash
+                    fnew.append(fl)
+            else:
+                fnew.append(fl)
                 
-            fnew.append(fl)
         return fnew, fnew
     else:
         return f, f