Update app.py

app.py

@@ -138,9 +138,16 @@ 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_arr_copy = image_arr.copy()
+
+        gray = cv2.cvtColor(image_arr, cv2.COLOR_BGR2GRAY)
+        thresh = cv2.threshold(gray, 250, 255, cv2.THRESH_BINARY_INV)[1]
+        
+        x,y,w,h = cv2.boundingRect(thresh)
+        image_final = image_arr_copy[y:y+h, x:x+w]
 
 
-        src = image_arr
+        src = 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)