Update pages/01_🦷 Segment.py

pages/01_🦷 Segment.py

@@ -6,6 +6,7 @@ import numpy as np
 from sklearn import neighbors
 from scipy.spatial import distance_matrix
 from pygco import cut_from_graph
+import streamlit_ext as ste
 import open3d as o3d
 import matplotlib.pyplot as plt
 import matplotlib.colors as mcolors
@@ -586,7 +587,7 @@ def mesh_to_points_main(jaw, pcd_points, center_points, labels):
     instances_labels = upsample_label.copy()
     # '''
     # o3d.io.write_point_cloud(os.path.join(save_path, 'upsample_' + name + '.pcd'), new_pcd, write_ascii=True)
-    for i in range(0, upsample_label.shape[0]):
+    for i in stqdm(range(0, upsample_label.shape[0])):
         if jaw == 'upper':
             if (upsample_label[i] >= 1) and (upsample_label[i] <= 8):
                 upsample_label[i] = upsample_label[i] + 10
@@ -855,7 +856,7 @@ def segmentation_main(obj_path):
                 # Show the plot
                 #plotter.show()
                 ## Send to streamlit
-                with st.expander("View Segmentation Result", expanded=False):
+                with st.expander("**View Segmentation Result** - ", expanded=False):
                     stpyvista(plotter)
 
 # Configure Streamlit page
@@ -869,7 +870,7 @@ class Segment(TeethApp):
     def build_app(self):
 
         st.title("Segment Intra-oral Scans")
-        st.markdown("Select scan for segmentation")
+        st.markdown("Identify and segment teeth. Segmentation is performed using MeshSegNet, a deep learning model trained on both upper and lower jaws.")
 
         inputs = st.radio(
             "Select scan for segmentation:",
@@ -882,8 +883,11 @@ class Segment(TeethApp):
             plotter = pv.Plotter()
 
             # Add the mesh to the plotter
-            plotter.add_mesh(mesh, color='white', show_edges=True)
-            segment = st.button("Apply Segmentation")
+            plotter.add_mesh(mesh, color='white', show_edges=False)
+            segment = st.button(
+                "✔️ Submit",
+                help="Submit 3D scan for segmentation",
+            )
             with st.expander("View Scan", expanded=False):
                 stpyvista(plotter)
 
@@ -906,14 +910,18 @@ class Segment(TeethApp):
                 plotter = pv.Plotter()
 
                 # Add the mesh to the plotter
-                plotter.add_mesh(mesh, color='white', show_edges=True)
-                segment = st.button("Apply Segmentation")
+                plotter.add_mesh(mesh, color='white', show_edges=False)
+                segment = st.button(
+                "✔️ Submit",
+                help="Submit 3D scan for segmentation",
+            )
                 with st.expander("View Scan", expanded=False):
                     stpyvista(plotter)
 
                 if segment:
                     segmentation_main(obj_path)
 
+