Update app.py

app.py

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