Update app.py

app.py

@@ -1,15 +1,14 @@
 from ansys.mapdl.core import launch_mapdl
-import numpy as np
 import gradio as gr
 
-# Function to generate progressive die design using ANSYS MAPDL
-def generate_progressive_die_ANSYS(length, width, thickness, num_stations, punch_diameter, punch_depth, die_shape):
+# Function to generate die design using ANSYS MAPDL
+def generate_die_ANSYS(length, width, thickness, die_shape):
     try:
-        if length <= 0 or width <= 0 or thickness <= 0 or num_stations <= 0:
-            return "Dimensions and number of stations must be greater than zero."
+        if length <= 0 or width <= 0 or thickness <= 0:
+            return "Dimensions must be greater than zero."
 
-        # Launch ANSYS MAPDL session in non-interactive mode
-        mapdl = launch_mapdl(interactive=False)
+        # Launch ANSYS MAPDL session (without interactive argument)
+        mapdl = launch_mapdl()  # Start MAPDL session
         mapdl.clear()  # Clear any previous settings
 
         # Preprocessing
@@ -17,56 +16,40 @@ def generate_progressive_die_ANSYS(length, width, thickness, num_stations, punch
 
         # Die shape logic
         if die_shape == "Rectangle":
-            mapdl.block(0, length, 0, width, 0, thickness)  # Create a rectangular block (base die)
+            mapdl.block(0, length, 0, width, 0, thickness)  # Create a rectangular block
         elif die_shape == "Circle":
-            mapdl.cylind(0, 0, width / 2, 0, thickness)  # Create a cylindrical block (base die)
-
-        # Create punch shapes for each station
-        punch_spacing = length / (num_stations + 1)  # Space punches evenly along length
-
-        for station in range(1, num_stations + 1):
-            punch_center_x = station * punch_spacing
-            punch_center_y = width / 2  # Center the punch along the width
-
-            # Create cylindrical punches (holes in the die)
-            mapdl.cylind(punch_center_x, punch_center_y, punch_diameter / 2, 0, punch_depth)
+            mapdl.cylind(0, 0, width / 2, 0, thickness)  # Create a cylindrical block
 
         # Save the geometry as a STEP file
-        filename = f"progressive_die_{die_shape}_{length}x{width}x{thickness}_{num_stations}_stations.step"
-        mapdl.save_as(filename)
-        mapdl.exit()  # Close the session
-
-        # Return success message with station info
-        return f"Progressive Die design saved successfully as {filename}. The die has {num_stations} punch stations."
+        filename = f"generated_die_{die_shape}_{length}x{width}x{thickness}.step"
+        mapdl.save_as(filename)  # Save the design as STEP file
+        mapdl.exit()  # Exit the MAPDL session
 
+        # Return success message with the file path
+        return f"Die design saved successfully as {filename}. You can download the file."
+    
     except Exception as e:
-        return f"Error generating progressive die design: {str(e)}"
+        return f"Error generating die design: {str(e)}"
 
-# Gradio interface function
-def progressive_die_interface(length, width, thickness, num_stations, punch_diameter, punch_depth, die_shape):
-    return generate_progressive_die_ANSYS(length, width, thickness, num_stations, punch_diameter, punch_depth, die_shape)
+# Gradio interface functions for Progressive Die Design
+def progressive_die_interface(length, width, thickness, die_shape):
+    return generate_die_ANSYS(length, width, thickness, die_shape)
 
 # Create the Gradio Interface
 with gr.Blocks() as app:
-    gr.Markdown("## Progressive Die Design with ANSYS MAPDL")
-    gr.Markdown("### Enter the parameters for the die design")
-    
-    with gr.Tabs():
-        # Tab for Progressive Die Design
-        with gr.Tab("Progressive Die Design"):
-            gr.Markdown("### Enter Dimensions and Punch Parameters")
-            length = gr.Number(label="Die Length (mm)", value=500)
-            width = gr.Number(label="Die Width (mm)", value=200)
-            thickness = gr.Number(label="Die Thickness (mm)", value=50)
-            num_stations = gr.Number(label="Number of Stations", value=5)
-            punch_diameter = gr.Number(label="Punch Diameter (mm)", value=20)
-            punch_depth = gr.Number(label="Punch Depth (mm)", value=10)
-            die_shape = gr.Dropdown(label="Die Shape", choices=["Rectangle", "Circle"], value="Rectangle")
-            die_output = gr.Textbox(label="Output", interactive=False)
-            die_button = gr.Button("Generate Die")
-            die_button.click(progressive_die_interface, inputs=[length, width, thickness, num_stations, punch_diameter, punch_depth, die_shape], outputs=die_output)
+    gr.Markdown("## Progressive Die Design Tool with ANSYS")
+    gr.Markdown("Select the dimensions and shape for the progressive die below:")
+
+    # Tab for Progressive Die Design
+    with gr.Tab("Progressive Die Design"):
+        gr.Markdown("### Enter Dimensions and Shape for Progressive Die")
+        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_shape = gr.Dropdown(label="Die Shape", choices=["Rectangle", "Circle"], value="Rectangle")
+        die_output = gr.Textbox(label="Output", interactive=False)
+        die_button = gr.Button("Generate Die")
+        die_button.click(progressive_die_interface, inputs=[length, width, thickness, die_shape], outputs=die_output)
 
 # Launch the app
 app.launch()
-
-