Spaces:
Sleeping
Sleeping
File size: 1,464 Bytes
f22f137 93d9fce f22f137 93d9fce f22f137 93d9fce f22f137 93d9fce |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 |
import cadquery as cq
import os
import pyvista as pv
def generate_die(length, width, thickness):
try:
output_dir = "outputs"
os.makedirs(output_dir, exist_ok=True)
plate = cq.Workplane("XY").box(length, width, thickness)
punch = cq.Workplane("XY").rect(10, 10).extrude(5).translate((length / 4, width / 4, thickness / 2))
die = plate.cut(punch)
filename = os.path.join(output_dir, "progressive_die.step")
cq.exporters.export(die, filename)
return filename
except Exception as e:
return f"Error generating die: {str(e)}"
def visualize_die(length, width, thickness):
try:
output_dir = "outputs"
os.makedirs(output_dir, exist_ok=True)
plate = cq.Workplane("XY").box(length, width, thickness)
punch = cq.Workplane("XY").rect(10, 10).extrude(5).translate((length / 4, width / 4, thickness / 2))
die = plate.cut(punch)
stl_file = os.path.join(output_dir, "progressive_die.stl")
cq.exporters.exportShape(die.val(), "STL", stl_file)
pv.global_theme.off_screen = True
mesh = pv.read(stl_file)
plotter = pv.Plotter(off_screen=True)
plotter.add_mesh(mesh, color="blue")
screenshot = os.path.join(output_dir, "progressive_die_visualization.png")
plotter.screenshot(screenshot)
return screenshot
except Exception as e:
return f"Error visualizing die: {str(e)}" |