Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -903,6 +903,7 @@ def predict_middle(image, enable_fillet, fillet_value_mm):
|
|
| 903 |
def predict_full(image, enable_fillet, fillet_value_mm, enable_finger_cut, selected_outputs):
|
| 904 |
"""
|
| 905 |
Returns DXF + conditionally selected additional outputs
|
|
|
|
| 906 |
"""
|
| 907 |
radius = fillet_value_mm if enable_fillet == "On" else 0
|
| 908 |
finger_flag = "On" if enable_finger_cut == "On" else "Off"
|
|
@@ -916,17 +917,13 @@ def predict_full(image, enable_fillet, fillet_value_mm, enable_finger_cut, selec
|
|
| 916 |
finger_clearance=finger_flag,
|
| 917 |
)
|
| 918 |
|
| 919 |
-
#
|
| 920 |
-
|
| 921 |
-
|
| 922 |
-
|
| 923 |
-
|
| 924 |
-
|
| 925 |
-
|
| 926 |
-
if "Mask" in selected_outputs:
|
| 927 |
-
results.append(mask)
|
| 928 |
-
|
| 929 |
-
return results
|
| 930 |
|
| 931 |
|
| 932 |
|
|
|
|
| 903 |
def predict_full(image, enable_fillet, fillet_value_mm, enable_finger_cut, selected_outputs):
|
| 904 |
"""
|
| 905 |
Returns DXF + conditionally selected additional outputs
|
| 906 |
+
Always returns exactly 4 values to match output components
|
| 907 |
"""
|
| 908 |
radius = fillet_value_mm if enable_fillet == "On" else 0
|
| 909 |
finger_flag = "On" if enable_finger_cut == "On" else "Off"
|
|
|
|
| 917 |
finger_clearance=finger_flag,
|
| 918 |
)
|
| 919 |
|
| 920 |
+
# Always return 4 values to match the 4 output components
|
| 921 |
+
return (
|
| 922 |
+
dxf_path, # Always return DXF
|
| 923 |
+
ann if "Annotated Image" in selected_outputs else None,
|
| 924 |
+
outlines if "Outlines" in selected_outputs else None,
|
| 925 |
+
mask if "Mask" in selected_outputs else None
|
| 926 |
+
)
|
|
|
|
|
|
|
|
|
|
|
|
|
| 927 |
|
| 928 |
|
| 929 |
|