Update app.py

app.py

@@ -140,6 +140,7 @@ def extrude_depth_3d(
     scene_lights=True,
     prepare_for_3d_printing=False,
     zip_outputs=False,
+    lift_height=0.0
 ):
     f_far_inner = -emboss
     f_far_outer = f_far_inner - f_back
@@ -177,6 +178,7 @@ def extrude_depth_3d(
     #depth_image_new[depth_image_new < 0] = 0
     depth_image = median_filter(depth_image, size=filter_size)
     depth_image = emboss*(depth_image - np.min(depth_image)) / (np.max(depth_image) - np.min(depth_image))
+    depth_image = np.where(depth_image != 0.0, depth_image + lift_height, 0.0)
     Image.fromarray((depth_image * 255).astype(np.uint8)).convert("L").save(path_out_base+".png")
     rgb_image = np.array(
         Image.open(path_rgb).convert("RGB").resize((w, h), Image.Resampling.LANCZOS)
@@ -427,7 +429,7 @@ def scale_to_width(img, length):
 
 
 # Gradio Interface function
-def process_image_and_generate_stl(image_input, depth_near, depth_far, thickness, alpha, backsheet):
+def process_image_and_generate_stl(image_input, depth_near, depth_far, thickness, alpha, backsheet, lift):
     # Depth Estimation
     raw_img = cv2.imread(image_input)
     depth = model.infer_image(raw_img) # HxW raw depth map in numpy
@@ -474,6 +476,7 @@ def process_image_and_generate_stl(image_input, depth_near, depth_far, thickness
         vertex_colors=True,
         scene_lights=True,
         prepare_for_3d_printing=True,
+        lift_height=lift
     )
 
     return stl_path # Return the path to the generated STL file
@@ -488,7 +491,8 @@ iface = gr.Interface(
         gr.Slider(minimum=0, maximum=1.0, value=1.0, label="Depth Far"),
         gr.Slider(minimum=0.1, maximum=1.0, value=0.3, label="Thickness"),
         gr.Slider(minimum=0, maximum=1.0, value=0.05, label="Alpha"),
-        gr.Slider(minimum=-0.1, maximum=0.1, value=0.01, label="BackSheet Thickness")
+        gr.Slider(minimum=0.01, maximum=0.1, value=0.01, label="BackSheet Thickness")
+        gr.Slider(minimum=0, maximum=1.0, value=0.0, label="lift"),
     ],
     outputs=gr.File(label="Download STL File"), # Use gr.File() for file downloads
     title="Image to 2.5D Relief Model Generator",