Update app.py

app.py

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