Update app.py

app.py

@@ -132,6 +132,13 @@ def floorplan_generation(outline, num_of_rooms):
         mask = np.array(n_outline)
         mask = cv2.cvtColor(mask, cv2.COLOR_RGB2BGR)
         image = clean_img(image, mask)
+
+        image_arr = np.array(image)
+        mask_bgr = cv2.cvtColor(image_arr, cv2.COLOR_RGB2BGR)
+        mask_gray = cv2.cvtColor(mask_bgr, cv2.COLOR_RGB2GRAY)
+        mask_white = cv2.threshold(mask_gray, 210, 255, cv2.THRESH_BINARY_INV)[1]
+        image_arr[mask_white<250]=(255,255,255)
+        image = Image.fromarray(image_arr).convert('RGB')
         image_lst.append(image)
         
     return image_lst[0], image_lst[1], image_lst[2], image_lst[3], image_lst[4]