Update app.py

app.py

@@ -7,9 +7,11 @@ from ansys.mapdl.core import launch_mapdl
 from pycalculix import *
 
 # Simulation Workflow
-def simulation_workflow(use_case, simulator, **kwargs):
+def simulation_workflow(
+    use_case, simulator, thickness, length, width, hole_diameter, force, load
+):
     print(f"Running simulation for use_case={use_case}, simulator={simulator}")
-    
+
     # Generate APDL script based on user inputs
     if use_case == "plate":
         print("Generating APDL script for plate simulation...")
@@ -17,16 +19,16 @@ def simulation_workflow(use_case, simulator, **kwargs):
 /PREP7
 MP,EX,1,2E11       ! Elastic modulus
 MP,PRXY,1,0.3      ! Poisson's ratio
-BLOCK,0,{kwargs["length"]},0,{kwargs["width"]},0,{kwargs["thickness"]}  ! Main block geometry
-CYLIND,0,{kwargs.get("hole_diameter", 0) / 2},0,0,{kwargs["thickness"]} ! Circular hole geometry
+BLOCK,0,{length},0,{width},0,{thickness}  ! Main block geometry
+CYLIND,0,{hole_diameter / 2},0,0,{thickness} ! Circular hole geometry
 VSUB,ALL           ! Subtract the hole from the block
 ET,1,SOLID185      ! Define element type as SOLID185
 ESIZE,5            ! Mesh element size
 VMESH,ALL          ! Mesh the entire volume
 NSEL,S,LOC,Z,0     ! Select bottom face nodes
 D,ALL,ALL          ! Apply constraints on selected nodes
-NSEL,S,LOC,Z,{kwargs["thickness"]} ! Select top face nodes
-F,ALL,FY,-{kwargs["force"]}  ! Apply vertical force
+NSEL,S,LOC,Z,{thickness} ! Select top face nodes
+F,ALL,FY,-{force}  ! Apply vertical force
 /SOLU
 ANTYPE,STATIC      ! Static structural analysis
 SOLVE
@@ -41,14 +43,14 @@ PRNSOL,U,SUM       ! Print total deformation results
 /PREP7
 MP,EX,1,2E11       ! Elastic modulus
 MP,PRXY,1,0.3      ! Poisson's ratio
-BLOCK,0,{kwargs["length"]},0,{kwargs["width"]},0,{kwargs["thickness"]}  ! Beam geometry
+BLOCK,0,{length},0,{width},0,{thickness}  ! Beam geometry
 ET,1,SOLID185      ! Define element type as SOLID185
 ESIZE,5            ! Mesh element size
 VMESH,ALL          ! Mesh the entire volume
 NSEL,S,LOC,X,0     ! Select nodes at one end of the beam
 D,ALL,ALL          ! Apply fixed constraints on one end
-NSEL,S,LOC,X,{kwargs["length"]} ! Select nodes at the other end
-F,ALL,FY,-{kwargs["load"]}   ! Apply uniform load
+NSEL,S,LOC,X,{length} ! Select nodes at the other end
+F,ALL,FY,-{load}   ! Apply uniform load
 /SOLU
 ANTYPE,STATIC      ! Static structural analysis
 SOLVE
@@ -60,11 +62,11 @@ PRNSOL,U,SUM       ! Print total deformation results
     else:
         print("Invalid use case selected.")
         return "Invalid use case selected.", None, None
-    
+
     # Save APDL script
     with open("simulation_input.inp", "w") as file:
         file.write(apdl_script)
-    
+
     print(f"APDL script saved at: simulation_input.inp")
 
     # Run the selected simulator
@@ -73,10 +75,10 @@ PRNSOL,U,SUM       ! Print total deformation results
         model = Model("pycalculix_simulation")
         model.set_units("mm")
         part = Part("block", model)
-        part.make_box(0, kwargs["length"], 0, kwargs["width"], 0, kwargs["thickness"])
+        part.make_box(0, length, 0, width, 0, thickness)
         if use_case == "plate":
-            part.cut_cylinder(0, kwargs.get("hole_diameter", 0) / 2, 0, kwargs.get("hole_diameter", 0) / 2, kwargs["thickness"])
-        model.make_step_static(kwargs.get("force", kwargs.get("load", 0)))
+            part.cut_cylinder(0, hole_diameter / 2, 0, hole_diameter / 2, thickness)
+        model.make_step_static(force if force else load)
         model.run()
         stress = model.results().max_stress()
         deformation = model.results().max_displacement()
@@ -102,6 +104,22 @@ PRNSOL,U,SUM       ! Print total deformation results
     plt.savefig("results_2d.png")
     plt.close(fig)
 
+    # 3D Visualization
+    mesh = pv.Box(bounds=(0, length, 0, width, 0, thickness))
+    plotter = pv.Plotter(off_screen=True)
+    plotter.add_mesh(mesh, color="white", show_edges=True)
+    plotter.screenshot("results_3d.png")
+
+    return f"Stress: {stress:.2f} MPa, Deformation: {deformation:.2f} mm", "results_2d.png", "results_3d.png"
+
+    # Visualize results (both 2D and 3D)
+    print("Visualizing results...")
+    fig, ax = plt.subplots()
+    ax.bar(["Stress", "Deformation"], [stress, deformation], color=["red", "blue"])
+    ax.set_title(f"Results ({simulator})")
+    plt.savefig("results_2d.png")
+    plt.close(fig)
+
     # 3D Visualization
     mesh = pv.Box(bounds=(0, kwargs["length"], 0, kwargs["width"], 0, kwargs["thickness"]))
     plotter = pv.Plotter(off_screen=True)
@@ -111,27 +129,24 @@ PRNSOL,U,SUM       ! Print total deformation results
     return f"Stress: {stress:.2f} MPa, Deformation: {deformation:.2f} mm", "results_2d.png", "results_3d.png"
 
 # Define Gradio interface
-interface = gr.Interface(
+gr.Interface(
     fn=simulation_workflow,
     inputs=[
-        gr.Radio(["plate", "beam"], label="Select Use Case"),  # Choose plate or beam simulation
-        gr.Dropdown(["PyCalculix", "ANSYS"], label="Select Simulator"),  # Choose simulator
-        gr.Slider(10, 50, step=1, label="Thickness (mm)"),  # Input: Thickness
-        gr.Slider(100, 500, step=10, label="Length (mm)"),  # Input: Length
-        gr.Slider(50, 200, step=10, label="Width (mm)"),  # Input: Width
-        gr.Slider(5, 25, step=1, label="Hole Diameter (mm)", value=None),  # Input: Hole Diameter (for plate)
-        gr.Slider(1000, 10000, step=500, label="Force (N)", value=None),  # Input: Force (for plate)
-        gr.Slider(1000, 20000, step=1000, label="Load (N)", value=None)  # Input: Load (for beam)
+        gr.Radio(["plate", "beam"], label="Select Use Case"),
+        gr.Dropdown(["PyCalculix", "ANSYS"], label="Select Simulator"),
+        gr.Slider(10, 50, step=1, label="Thickness (mm)"),
+        gr.Slider(100, 500, step=10, label="Length (mm)"),
+        gr.Slider(50, 200, step=10, label="Width (mm)"),
+        gr.Slider(5, 25, step=1, label="Hole Diameter (mm)", value=5),  # Default for Plate
+        gr.Slider(1000, 10000, step=500, label="Force (N)", value=5000),  # Default for Plate
+        gr.Slider(1000, 20000, step=1000, label="Load (N)", value=5000)  # Default for Beam
     ],
     outputs=[
-        gr.Textbox(label="Simulation Results"),  # Output: Simulation text results
-        gr.Image(label="2D Results Visualization"),  # Output: 2D plot (stress and deformation)
-        gr.Image(label="3D Results Visualization")  # Output: 3D plot (stress distribution)
+        gr.Textbox(label="Simulation Results"),
+        gr.Image(label="2D Results Visualization"),
+        gr.Image(label="3D Results Visualization")
     ],
     title="Unified Simulation Tool (PyCalculix and ANSYS)",
     live=True
-)
+).launch()
 
-# Launch Gradio interface
-print("Launching Gradio interface...")
-interface.launch()