add UI for parameter settings

app.py

@@ -59,7 +59,7 @@ def check_delimiter(file_path):
     else:
         return " "
 
-def meshing(file_obj):
+def meshing(file_obj, normal_rad_coef, normal_max_nn, orient_k, bpa_min_coef, bpa_max_coef):
     try:
         if DEBUG:
             temp_path = os.path.join(file_obj.dir, file_obj.name)
@@ -81,12 +81,13 @@ def meshing(file_obj):
 
         dists = pcd.compute_nearest_neighbor_distance()
         avg_spacing = np.mean(dists)
-        normal_radius = avg_spacing*5.0 #2.0  # start with 2–3x spacing
-        pcd.estimate_normals(search_param=o3d.geometry.KDTreeSearchParamHybrid(radius=normal_radius, max_nn=100)) #30))
-        pcd.orient_normals_consistent_tangent_plane(k=100) #30)
+        normal_radius = avg_spacing*normal_rad_coef # start with 2–3x spacing
+        pcd.estimate_normals(search_param=o3d.geometry.KDTreeSearchParamHybrid(radius=normal_radius, max_nn=normal_max_nn))
+        pcd.orient_normals_consistent_tangent_plane(k=orient_k)
 
         # Mesh using Ball Pivoting
-        mesh_radii = [avg_spacing * x for x in [3, 4, 5, 6]] #[1.5, 2.5, 3.5]]
+        coeffs = np.linspace(bpa_min_coef, bpa_max_coef, int(4))
+        mesh_radii = [avg_spacing * x for x in coeffs]
         mesh = o3d.geometry.TriangleMesh.create_from_point_cloud_ball_pivoting(pcd, o3d.utility.DoubleVector(mesh_radii))
 
         # Mesh using Poisson
@@ -117,7 +118,14 @@ def meshing(file_obj):
 
 iface_mesh = gr.Interface(
     fn=meshing,
-    inputs=gr.File(file_types=[".txt", ".ply", ".pcd"], label="Upload Point Cloud"),
+    inputs=[
+        gr.File(file_types=[".txt", ".ply", ".pcd"], label="Upload Point Cloud"),
+        gr.Slider(1.0, 10.0, value=2.0, step=0.1, label="Normal Radius (Hybrid Search)"),
+        gr.Slider(1, 100, value=30, step=1, label="Max NN for Normals"),
+        gr.Slider(1, 100, value=30, step=1, label="K for Consistent Normal Orientation"),
+        gr.Slider(0.1, 10.0, value=1.5, step=0.1, label="Ball Pivoting Radius Min"),
+        gr.Slider(0.1, 10.0, value=3.5, step=0.1, label="Ball Pivoting Radius Max"),
+    ],
     outputs=[
         gr.Image(type="numpy", label="Mesh Preview"),
         gr.File(label="Download Mesh (STL)")