Update app.py

app.py

@@ -1,13 +1,12 @@
 import pyvista as pv
 import streamlit as st
-from streamlit_pyvista import st_pyvista
-import numpy as np
-import pyvista as pv
-import streamlit as st
 
-# Define parameters for the punch and die
-length, width, height = 50, 30, 10
-inner_width, inner_height, die_thickness = 40, 20, 15
+# Material properties (dummy data)
+materials = {
+    "Steel": {"Density": 7850, "Yield Strength": 250},
+    "Aluminum": {"Density": 2700, "Yield Strength": 50},
+    "Brass": {"Density": 8500, "Yield Strength": 200},
+}
 
 # Function to create punch and die geometry
 def create_geometry(length, width, height, inner_width, inner_height, thickness):
@@ -17,45 +16,6 @@ def create_geometry(length, width, height, inner_width, inner_height, thickness)
                          -thickness, 0))
     return punch, die
 
-# Generate punch and die models
-punch, die = create_geometry(length, width, height, inner_width, inner_height, die_thickness)
-
-
-# Save the visualization as an image
-image_path = "visualization.png"
-plotter.screenshot(image_path)
-
-# Display the image in Streamlit
-st.title("Punch and Die Visualization")
-st.image(image_path, caption="3D Visualization of Punch and Die", use_column_width=True)
-
-
-# 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 = {
-    "Steel": {"Density": 7850, "Yield Strength": 250},
-    "Aluminum": {"Density": 2700, "Yield Strength": 50},
-    "Brass": {"Density": 8500, "Yield Strength": 200},
-}
-
 # Streamlit App
 st.title("Interactive Punch and Die Design Tool")
 st.sidebar.header("Input Parameters")
@@ -83,28 +43,24 @@ st.sidebar.subheader("Material Properties")
 st.sidebar.write(f"Density: {materials[material]['Density']} kg/m³")
 st.sidebar.write(f"Yield Strength: {materials[material]['Yield Strength']} MPa")
 
-# Function to create punch and die geometry
-def create_geometry(length, width, height, inner_width, inner_height, thickness):
-    punch = pv.Box(bounds=(0, length, 0, width, 0, height))
-    die = pv.Box(bounds=(-inner_width / 2, inner_width / 2,
-                         -inner_height / 2, inner_height / 2,
-                         -thickness, 0))
-    return punch, die
-
-# Generate models
+# Generate Punch and Die Models
 punch, die = create_geometry(length, width, height, inner_width, inner_height, die_thickness)
 
-# Display 3D Visualization
+# 3D Visualization
 st.subheader("3D Visualization of Punch and Die")
-plotter = pv.Plotter()
+plotter = pv.Plotter(off_screen=True)  # Use off_screen for static visualization
 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()
 
-# Show the visualization in Streamlit
-st_pyvista(plotter)
+# Save the visualization as an image
+image_path = "visualization.png"
+plotter.screenshot(image_path)
+
+# Display the image in Streamlit
+st.image(image_path, caption="3D Visualization of Punch and Die", use_column_width=True)
 
 # Generate APDL Script
 st.subheader("Generated APDL Script")
@@ -126,11 +82,11 @@ SOLVE
 /POST1
 """
 
+# Display and Download APDL Script
 st.code(apdl_script)
-
-# Option to download APDL Script
 st.download_button("Download APDL Script", apdl_script, "generated_apdl_script.txt")
 
-# Placeholder for future validation results
+# Placeholder for Validation Results
 st.subheader("Validation Feedback")
 st.write("Validation results and stress maps will be shown here.")
+