Update app.py

app.py

@@ -3,6 +3,13 @@ from datasets import load_dataset
 from pymesh import Mesh  # For 3D mesh generation
 import pyvista as pv  # For 3D mesh visualization
 
+# Error handling for pymesh
+try:
+    from pymesh import Mesh
+except ImportError:
+    print("Error: pymesh.Mesh not found. Please install pymesh using 'pip install pymesh'.")
+    st.error("**Mesh generation functionality requires pymesh library. Please install it.**")
+
 # Page Title
 st.title("Quick CAD Model Generator")
 
@@ -59,22 +66,20 @@ elif input_option == "Select Predefined Template":
             # Example using PyMesh for generating a simple cube
             if selected_template == "Cube":  # Replace with your template logic
                 if "width" in user_inputs and "height" in user_inputs and "depth" in user_inputs:
-                    width = float(user_inputs["width"])
-                    height = float(user_inputs["height"])
-                    depth = float(user_inputs["depth"])
-                    cube_vertices = [
-                        [0, 0, 0], [width, 0, 0], [width, height, 0], [0, height, 0],
-                        [0, 0, depth], [width, 0, depth], [width, height, depth], [0, height, depth]
-                    ]
-                    cube_faces = [
-                        [0, 1, 2, 3], [4, 5, 6, 7],
-                        [0, 4, 5, 1], [1, 5, 6, 2],
-                        [2, 6, 7, 3], [0, 3, 7, 4]
-                    ]
-                    cube_mesh = Mesh(vertices=cube_vertices, faces=cube_faces)
+                    try:
+                        width = float(user_inputs["width"])
+                        height = float(user_inputs["height"])
+                        depth = float(user_inputs["depth"])
+                        cube_vertices = [
+                            [0, 0, 0], [width, 0, 0], [width, height, 0], [0, height, 0],
+                            [0, 0, depth], [width, 0, depth], [width, height, depth], [0, height, depth]
+                        ]
+                        cube_faces = [
+                            [0, 1, 2, 3], [4, 5, 6, 7],
+                            [0, 4, 5, 1], [1, 5, 6, 2],
+                            [2, 6, 7, 3], [0, 3, 7, 4]
+                        ]
+                        cube_mesh = Mesh(vertices=cube_vertices, faces=cube_faces)
 
-                    # **Visualization using PyVista**
-                    py_mesh = pv.wrap(cube_mesh)  # Convert PyMesh to PyVista mesh
-                    plotter = pv.Plotter()
-                    plotter.add_mesh(py_mesh)
-                    plotter.show(jupyter_backend='static')
+                        # **Visualization using PyVista**
+                        py_mesh = pv.wrap(cube_mesh)  # Convert PyMesh to PyVista