mlbench123 commited on
Commit
67f9099
·
verified ·
1 Parent(s): 32c6ede

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -11
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
- # Build return tuple based on selected outputs
920
- results = [dxf_path] # DXF always included
921
-
922
- if "Annotated Image" in selected_outputs:
923
- results.append(ann)
924
- if "Outlines" in selected_outputs:
925
- results.append(outlines)
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