Update app.py

app.py

@@ -131,7 +131,29 @@ def floorplan_generation(outline, num_of_rooms):
         image = np.array(image)
         mask = np.array(n_outline)
         mask = cv2.cvtColor(mask, cv2.COLOR_RGB2BGR)
-        image = clean_img(image, mask)
+        image_c = clean_img(image, mask)
+        
+        mask_arr = np.array(image_c)
+        mask_bgr = cv2.cvtColor(mask_arr, cv2.COLOR_RGB2BGR)
+        mask_gray = cv2.cvtColor(mask_bgr, cv2.COLOR_RGB2GRAY)
+        mask = cv2.threshold(mask_gray, 210, 255, cv2.THRESH_BINARY_INV)[1]
+        image_c[mask<250]=(255,255,255)
+        
+        image_copy = image_c.copy()
+        
+        gray = cv2.cvtColor(image_c, cv2.COLOR_BGR2GRAY)
+        thresh = cv2.threshold(gray, 250, 255, cv2.THRESH_BINARY_INV)[1]
+        
+        x,y,w,h = cv2.boundingRect(thresh)
+        image_f = image_copy[y:y+h, x:x+w]
+        
+        src = np.array(image_f)
+        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).convert('RGB')
         image_lst.append(image)
         
     return image_lst[0], image_lst[1], image_lst[2], image_lst[3], image_lst[4]