Update app.py

app.py

@@ -787,7 +787,10 @@ def predict_with_paper(image, paper_size, offset,offset_unit, finger_clearance=F
     logger.info(f"Paper size: {paper_size}")
     
     if offset_unit == "inches":
-        offset *= 25.4
+        offset_mm = offset * 25.4  # Convert to mm for internal calculations
+    else:
+        offset_mm = offset
+        
     edge_radius = None
     if edge_radius is None or edge_radius == 0:
         edge_radius = 0.0001
@@ -838,13 +841,13 @@ def predict_with_paper(image, paper_size, offset,offset_unit, finger_clearance=F
     rounded_mask = objects_mask.copy()
     
     # Apply dilation for offset
-    if offset > 0:
-        offset_pixels = (float(offset) / scaling_factor) * 2 + 1 if scaling_factor else 1
+    if offset_mm > 0:
+        offset_pixels = (float(offset_mm) / scaling_factor  # Remove the *2 +1 which was causing excessive dilation
         kernel = np.ones((int(offset_pixels), int(offset_pixels)), np.uint8)
         dilated_mask = cv2.dilate(rounded_mask, kernel)
     else:
         dilated_mask = rounded_mask.copy()
-    
+
     # Save original dilated mask for output
     Image.fromarray(dilated_mask).save("./outputs/scaled_mask_original.jpg")
     dilated_mask_orig = dilated_mask.copy()