Update app.py

app.py

@@ -11,11 +11,18 @@ import os
 # Function for Progressive Die Design
 def generate_die(length, width, thickness):
     try:
+        output_dir = "outputs"
+        os.makedirs(output_dir, exist_ok=True)  # Ensure output directory exists
+
+        # Create die geometry
         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 = "progressive_die.step"
+
+        # Export STEP file
+        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)}"
@@ -24,25 +31,32 @@ def generate_die(length, width, thickness):
 # Function to visualize die in 3D
 def visualize_die(length, width, thickness):
     try:
+        output_dir = "outputs"
+        os.makedirs(output_dir, exist_ok=True)  # Ensure output directory exists
+
+        # Create die geometry
         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)
 
-        # Export to STL for visualization
-        cq.exporters.exportShape(die.val(), "STL", "progressive_die.stl")
+        # Export STL file for visualization
+        stl_file = os.path.join(output_dir, "progressive_die.stl")
+        cq.exporters.exportShape(die.val(), "STL", stl_file)
 
-        # Visualize with PyVista
-        mesh = pv.read("progressive_die.stl")
+        # Generate 3D visualization
+        pv.global_theme.off_screen = True  # Ensure off-screen rendering
+        mesh = pv.read(stl_file)
         plotter = pv.Plotter(off_screen=True)
         plotter.add_mesh(mesh, color="blue")
-        screenshot = "progressive_die_visualization.png"
+        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)}"
 
 
-# Stress Analysis Function
+# Function for Stress Analysis (including thermal stress and fatigue strength)
 def stress_analysis(force, die_width, die_height, material_strength, temperature_change=50, alpha=1e-5, elastic_modulus=200000, fatigue_strength=150):
     try:
         # Mechanical stress
@@ -119,7 +133,7 @@ with gr.Blocks() as app:
             length = gr.Number(label="Length (mm)", value=100)
             width = gr.Number(label="Width (mm)", value=50)
             thickness = gr.Number(label="Thickness (mm)", value=10)
-            die_output = gr.Textbox(label="Die Output File")
+            die_output = gr.Textbox(label="STEP File Location")
             visualization_output = gr.Image(label="3D Visualization")
             die_button = gr.Button("Generate Die")
             die_button.click(