Update app.py

app.py

@@ -780,14 +780,14 @@ def make_square(img: np.ndarray):
     
     return padded
 
-def predict_with_paper(image, paper_size, offset, finger_clearance=False):
+def predict_with_paper(image, paper_size, offset,offset_unit, finger_clearance=False):
     """Main prediction function using paper as reference"""
     
     logger.info(f"Starting prediction with image shape: {image.shape}")
     logger.info(f"Paper size: {paper_size}")
     
-    # if offset_unit == "inches":
-    #     offset *= 25.4
+    if offset_unit == "inches":
+        offset *= 25.4
     edge_radius = None
     if edge_radius is None or edge_radius == 0:
         edge_radius = 0.0001
@@ -902,7 +902,7 @@ def predict_with_paper(image, paper_size, offset, finger_clearance=False):
         f"Scale: {scaling_factor:.4f} mm/px | Paper: {paper_size}"
     )
 
-def predict_full_paper(image, paper_size, offset_value_mm, enable_finger_cut, selected_outputs):
+def predict_full_paper(image, paper_size, offset_value_mm,offset_unit, enable_finger_cut, selected_outputs):
     finger_flag = "On" if enable_finger_cut == "On" else "Off"
     
     # Always get all outputs from predict_with_paper
@@ -910,6 +910,7 @@ def predict_full_paper(image, paper_size, offset_value_mm, enable_finger_cut, se
         image,
         paper_size,
         offset=offset_value_mm,
+        offset_unit= offset_unit,
         finger_clearance=finger_flag,
     )
     
@@ -957,14 +958,20 @@ if __name__ == "__main__":
                 
                 with gr.Group():
                     gr.Markdown("### Contour Offset")
-                    offset_value_mm = gr.Slider(
-                        minimum=0,
-                        maximum=50,
-                        step=1,
-                        value=0,
-                        label="Offset (mm)",
-                        info="Expand contours outward by this amount"
-                    )
+                    with gr.Row():
+                        offset_value_mm = gr.Slider(
+                            minimum=0,
+                            maximum=50,
+                            step=1,
+                            value=0,
+                            label="Offset",
+                            info="Expand contours outward by this amount"
+                        )
+                        offset_unit = gr.Dropdown(
+                            choices=["mm", "inches"],
+                            value="mm",
+                            label="Unit"
+                        )
 
                 with gr.Group():
                     gr.Markdown("### Finger Cuts")
@@ -1015,7 +1022,8 @@ if __name__ == "__main__":
             inputs=[
                 input_image, 
                 paper_size, 
-                offset_value_mm, 
+                offset_value_mm,
+                offset_unit,
                 enable_finger_cut, 
                 output_options
             ],