Update app.py

app.py

@@ -31,10 +31,32 @@ def generate_2d_view(file):
 
 def generate_3d_view(file):
     try:
-        model = process_cad_file(file)
-        mesh = trimesh.Trimesh(vertices=model.val().vertices(), faces=model.val().faces())
-        three_d_image = mesh.show()
-        return three_d_image
+        # Import CAD file
+        model = cq.importers.importStep(file.name)
+
+        # Extract all solids from the Compound object
+        solids = model.solids()
+        if not solids:
+            raise ValueError("No solids found in the CAD model.")
+
+        # Convert solids into trimesh meshes
+        meshes = []
+        for solid in solids:
+            vertices = []
+            faces = []
+            for face in solid.tessellate():
+                vertices.extend(face[0])  # Extract vertices
+                faces.extend([[face[1][i], face[1][i + 1], face[1][i + 2]] for i in range(0, len(face[1]), 3)])  # Faces
+            mesh = trimesh.Trimesh(vertices=vertices, faces=faces)
+            meshes.append(mesh)
+
+        # Combine all meshes into one (if multiple solids)
+        combined_mesh = trimesh.util.concatenate(meshes) if len(meshes) > 1 else meshes[0]
+
+        # Visualize the mesh
+        combined_mesh.show()
+
+        return "3D view generated successfully."
     except Exception as e:
         return f"Error generating 3D view: {str(e)}"