Create app.py

app.py

@@ -0,0 +1,108 @@
+import os
+import gradio as gr
+import trimesh
+from ansys.mapdl.core import launch_mapdl
+
+def generate_apdl_from_cad(cad_file_path):
+    """
+    Generate APDL script from a CAD file.
+    Parameters:
+        cad_file_path: Path to the uploaded CAD file (STEP/STL).
+    Returns:
+        Path to the generated APDL script.
+    """
+    # Load the CAD model using Trimesh
+    cad_model = trimesh.load(cad_file_path)
+    
+    # Get the bounding box dimensions
+    bbox = cad_model.bounds
+    length = bbox[1][0] - bbox[0][0]
+    width = bbox[1][1] - bbox[0][1]
+    height = bbox[1][2] - bbox[0][2]
+
+    # Generate APDL script
+    apdl_script = f"""
+/prep7
+block, 0, {length:.4f}, 0, {width:.4f}, 0, {height:.4f}
+/mesh
+esize, 10
+vmesh, all
+allsel, all
+/solu
+antype, static
+d, 1, ux, 0
+d, 1, uy, 0
+d, 1, uz, 0
+f, 2, fy, -1000
+solve
+/exit
+"""
+
+    # Save the APDL script
+    apdl_file_path = "/tmp/generated_apdl.txt"
+    with open(apdl_file_path, "w") as f:
+        f.write(apdl_script)
+
+    return apdl_file_path
+
+
+def run_simulation(cad_file, apdl_script_path):
+    """
+    Run simulation using ANSYS MAPDL and return results.
+    Parameters:
+        cad_file: CAD file path.
+        apdl_script_path: Generated APDL script path.
+    Returns:
+        Simulation results (stress and displacement plots).
+    """
+    # Launch ANSYS MAPDL
+    mapdl = launch_mapdl()
+
+    # Import the APDL script
+    mapdl.input(apdl_script_path)
+
+    # Extract results
+    stress = mapdl.post_processing.nodal_stress(0)
+    displacement = mapdl.post_processing.nodal_displacement(0)
+
+    # Save stress and displacement plots
+    stress_plot_path = "/tmp/stress_plot.png"
+    displacement_plot_path = "/tmp/displacement_plot.png"
+    mapdl.post_processing.plot_nodal_stress(0, savefig=stress_plot_path)
+    mapdl.post_processing.plot_nodal_displacement(0, savefig=displacement_plot_path)
+
+    mapdl.exit()
+    return stress_plot_path, displacement_plot_path
+
+
+def process_cad_file(cad_file):
+    """
+    Complete process: CAD to APDL to Simulation.
+    Parameters:
+        cad_file: Uploaded CAD file (STEP/STL).
+    Returns:
+        APDL script and simulation results.
+    """
+    # Generate APDL script
+    apdl_script_path = generate_apdl_from_cad(cad_file.name)
+
+    # Run simulation
+    stress_plot, displacement_plot = run_simulation(cad_file.name, apdl_script_path)
+
+    return apdl_script_path, stress_plot, displacement_plot
+
+
+# Gradio Interface
+app = gr.Interface(
+    fn=process_cad_file,
+    inputs=gr.File(label="Upload CAD File (STEP/STL)"),
+    outputs=[
+        gr.File(label="Download Generated APDL Script"),
+        gr.Image(label="Stress Plot"),
+        gr.Image(label="Displacement Plot"),
+    ],
+    live=True,
+)
+
+if __name__ == "__main__":
+    app.launch()