Create app.py

app.py

@@ -0,0 +1,123 @@
+import gradio as gr
+import subprocess
+import numpy as np
+import pyvista as pv
+import matplotlib.pyplot as plt
+from ansys.mapdl.core import launch_mapdl
+from pycalculix import *
+
+# Simulation Workflow
+def simulation_workflow(use_case, simulator, **kwargs):
+    # Generate APDL script based on user inputs
+    if use_case == "plate":
+        apdl_script = f"""
+/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["hole_diameter"]/2},0,0,{kwargs["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
+/SOLU
+ANTYPE,STATIC      ! Static structural analysis
+SOLVE
+/POST1
+PRNSOL,S,EQV       ! Print equivalent stress results
+PRNSOL,U,SUM       ! Print total deformation results
+/EXIT
+"""
+    elif use_case == "beam":
+        apdl_script = f"""
+/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
+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
+/SOLU
+ANTYPE,STATIC      ! Static structural analysis
+SOLVE
+/POST1
+PRNSOL,S,EQV       ! Print equivalent stress results
+PRNSOL,U,SUM       ! Print total deformation results
+/EXIT
+"""
+    else:
+        return "Invalid use case selected.", None, None
+    
+    # Save APDL script
+    with open("simulation_input.inp", "w") as file:
+        file.write(apdl_script)
+    
+    # Run the selected simulator
+    if simulator == "PyCalculix":
+        model = Model("pycalculix_simulation")
+        model.set_units("mm")
+        part = Part("block", model)
+        part.make_box(0, kwargs["length"], 0, kwargs["width"], 0, kwargs["thickness"])
+        if use_case == "plate":
+            part.cut_cylinder(0, kwargs["hole_diameter"] / 2, 0, kwargs["hole_diameter"] / 2, kwargs["thickness"])
+        model.make_step_static(kwargs.get("force", kwargs.get("load", 0)))
+        model.run()
+        stress = model.results().max_stress()
+        deformation = model.results().max_displacement()
+    elif simulator == "ANSYS":
+        mapdl = launch_mapdl()
+        mapdl.input("simulation_input.inp")
+        max_stress = mapdl.get_value("NODE", 0, "S", "EQV")
+        deformation = mapdl.get_value("NODE", 0, "U", "SUM")
+        mapdl.exit()
+        stress, deformation = max_stress, deformation
+    else:
+        return "Invalid simulator selected.", None, None
+
+    # Visualize results (both 2D and 3D)
+    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)
+    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"
+
+# Define Gradio interface
+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)", optional=True),  # Input: Hole Diameter (for plate)
+        gr.Slider(1000, 10000, step=500, label="Force (N)", optional=True),  # Input: Force (for plate)
+        gr.Slider(1000, 20000, step=1000, label="Load (N)", optional=True)  # Input: Load (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)
+    ],
+    title="Unified Simulation Tool (PyCalculix and ANSYS)",
+    live=True
+)
+
+# Launch Gradio interface
+interface.launch()