Update app.py

app.py

@@ -138,7 +138,15 @@ def floorplan_generation(outline, num_of_rooms):
         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')
+
+
+        src = image_arr
+        tmp = cv2.cvtColor(src, cv2.COLOR_RGB2GRAY)
+        _,alpha = cv2.threshold(tmp,250,255,cv2.THRESH_BINARY_INV)
+        b, g, r = cv2.split(src)
+        rgba = [b,g,r, alpha]
+        dst = cv2.merge(rgba,4)
+        image = Image.fromarray(dst)
         image_lst.append(image)
         
     return image_lst[0], image_lst[1], image_lst[2], image_lst[3], image_lst[4]