Update app.py

app.py

@@ -2,6 +2,26 @@ import pyvista as pv
 import streamlit as st
 from streamlit_pyvista import st_pyvista
 import numpy as np
+import pyvista as pv
+
+# Generate the punch and die
+punch, die = create_geometry(length, width, height, inner_width, inner_height, die_thickness)
+
+# Create and save plot
+plotter = pv.Plotter(off_screen=True)
+plotter.add_mesh(punch, color="blue", opacity=0.5, label="Punch")
+plotter.add_mesh(die, color="red", opacity=0.5, label="Die")
+plotter.add_axes()
+plotter.show_bounds()
+plotter.view_xy()
+
+# Save the plot as an image
+plot_path = "visualization.png"
+plotter.screenshot(plot_path)
+
+# Display in Streamlit
+st.image(plot_path, caption="Punch and Die Visualization")
+
 
 # Material properties (dummy data)
 materials = {