Update app.py

app.py

@@ -6,6 +6,7 @@ import matplotlib.pyplot as plt
 from fpdf import FPDF
 import matplotlib.patches as patches
 import tempfile
+import math
 
 # Constants for estimation
 BRICK_VOLUME_CFT = (9 / 12) * (4.5 / 12) * (3 / 12)  # Brick Volume in Cubic Feet
@@ -53,15 +54,15 @@ def extract_geometry(file_bytes):
                     continue
 
         elif entity.dxftype() == "LWPOLYLINE":
-            # Check if it's a closed polyline representing a room
-            if entity.closed and len(entity) == 4:
+            # Check if it's a closed polyline representing a wall
+            if entity.closed:
                 points = entity.get_points()
                 x_vals = [p[0] for p in points]
                 y_vals = [p[1] for p in points]
-                length = abs(max(x_vals) - min(x_vals)) / 12  # Convert to feet
-                width = abs(max(y_vals) - min(y_vals)) / 12  # Convert to feet
-                if length > 2 and width > 2:
-                    rooms.append((length, width, 10))  # Add room with length, width, and height (default 10 ft)
+                width = abs(max(x_vals) - min(x_vals)) / 12  # Convert to feet
+                height = abs(max(y_vals) - min(y_vals)) / 12  # Convert to feet
+                if width > 2 and height > 2:  # Make sure it's a room-like area
+                    rooms.append((width, height, 10))  # Add room with length, width, and height (default 10 ft)
                     room_shapes.append((min(x_vals), min(y_vals), max(x_vals), max(y_vals)))
 
         elif entity.dxftype() == "LINE":