Update app.py

app.py

@@ -211,9 +211,13 @@ def to_dxf(contours):
     doc = ezdxf.new()
     msp = doc.modelspace()
 
-    for contour in contours:
-        points = [(point[0][0], point[0][1]) for point in contour]
-        msp.add_lwpolyline(points, close=True)  # Add a polyline for each contour
+    try:
+        for contour in contours:
+            points = [(point[0][0], point[0][1]) for point in contour]
+            msp.add_lwpolyline(points, close=True)  # Add a polyline for each contour
+    except Exception as e:
+        raise gr.Error(f"Unable to generate DXF: {e}")
+
 
     doc.saveas("./outputs/out.dxf")
     return "./outputs/out.dxf"
@@ -226,11 +230,9 @@ def smooth_contours(contour):
 def scale_image(image: np.ndarray, scale_factor: float) -> np.ndarray:
     """
     Resize image by scaling both width and height by the same factor.
-
     Args:
         image: Input numpy image
         scale_factor: Factor to scale the image (e.g., 0.5 for half size, 2 for double size)
-
     Returns:
         np.ndarray: Resized image
     """
@@ -302,7 +304,7 @@ def predict(image, offset_inches):
         scaling_box_coords,
         orig_size,
         processed_size,
-        expansion_factor=1.5,
+        expansion_factor=1.2,
     )
     objects_mask = resize_img(objects_mask, (image.shape[1], image.shape[0]))