Update app.py

app.py

@@ -1,26 +1,52 @@
 import os
 import gradio as gr
-import trimesh
+from OCC.Core.STEPControl import STEPControl_Reader
+from OCC.Core.BRepTools import breptools_Write
+from OCC.Extend.DataExchange import write_stl_file
 from ansys.mapdl.core import launch_mapdl
 
-def generate_apdl_from_cad(cad_file_path):
+
+def load_step_and_extract_geometry(step_file_path):
     """
-    Generate APDL script from a CAD file.
+    Load a STEP file and extract geometry information using pythonocc-core.
     Parameters:
-        cad_file_path: Path to the uploaded CAD file (STEP/STL).
+        step_file_path: Path to the uploaded STEP file.
     Returns:
-        Path to the generated APDL script.
+        STL file path, dimensions (length, width, height).
     """
-    # 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]
+    # Load the STEP file
+    step_reader = STEPControl_Reader()
+    status = step_reader.ReadFile(step_file_path)
+    if status != 1:
+        raise ValueError("Failed to load the STEP file.")
 
-    # Generate APDL script
+    # Transfer the file into shapes
+    step_reader.TransferRoots()
+    shape = step_reader.OneShape()
+
+    # Save the model as an STL file
+    stl_file_path = "/tmp/model.stl"
+    write_stl_file(shape, stl_file_path, linear_deflection=0.01)
+
+    # Calculate dimensions (bounding box)
+    bbox = shape.BoundingBox()
+    length = bbox.XMax() - bbox.XMin()
+    width = bbox.YMax() - bbox.YMin()
+    height = bbox.ZMax() - bbox.ZMin()
+
+    return stl_file_path, length, width, height
+
+
+def generate_apdl_script(length, width, height):
+    """
+    Generate an APDL script based on dimensions.
+    Parameters:
+        length: Length of the model.
+        width: Width of the model.
+        height: Height of the model.
+    Returns:
+        Path to the APDL script.
+    """
     apdl_script = f"""
 /prep7
 block, 0, {length:.4f}, 0, {width:.4f}, 0, {height:.4f}
@@ -37,67 +63,64 @@ 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:
+    apdl_path = "/tmp/generated_apdl.txt"
+    with open(apdl_path, "w") as f:
         f.write(apdl_script)
 
-    return apdl_file_path
+    return apdl_path
 
 
-def run_simulation(cad_file, apdl_script_path):
+def run_simulation(apdl_script_path):
     """
     Run simulation using ANSYS MAPDL and return results.
     Parameters:
-        cad_file: CAD file path.
-        apdl_script_path: Generated APDL script path.
+        apdl_script_path: Path to the generated APDL script.
     Returns:
-        Simulation results (stress and displacement plots).
+        Stress plot and displacement plot paths.
     """
     # Launch ANSYS MAPDL
     mapdl = launch_mapdl()
 
-    # Import the APDL script
+    # Input 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)
+    stress_plot = "/tmp/stress_plot.png"
+    displacement_plot = "/tmp/displacement_plot.png"
+    mapdl.post_processing.plot_nodal_stress(0, savefig=stress_plot)
+    mapdl.post_processing.plot_nodal_displacement(0, savefig=displacement_plot)
 
     mapdl.exit()
-    return stress_plot_path, displacement_plot_path
+    return stress_plot, displacement_plot
 
 
-def process_cad_file(cad_file):
+def process_step_file(step_file):
     """
-    Complete process: CAD to APDL to Simulation.
+    Complete workflow for STEP file: Load -> Generate APDL -> Simulate.
     Parameters:
-        cad_file: Uploaded CAD file (STEP/STL).
+        step_file: Uploaded STEP file.
     Returns:
-        APDL script and simulation results.
+        STL file, APDL script, stress plot, displacement plot.
     """
+    # Extract geometry from the STEP file
+    stl_file, length, width, height = load_step_and_extract_geometry(step_file.name)
+
     # Generate APDL script
-    apdl_script_path = generate_apdl_from_cad(cad_file.name)
+    apdl_script = generate_apdl_script(length, width, height)
 
     # Run simulation
-    stress_plot, displacement_plot = run_simulation(cad_file.name, apdl_script_path)
+    stress_plot, displacement_plot = run_simulation(apdl_script)
 
-    return apdl_script_path, stress_plot, displacement_plot
+    return stl_file, apdl_script, stress_plot, displacement_plot
 
 
 # Gradio Interface
 app = gr.Interface(
-    fn=process_cad_file,
-    inputs=gr.File(label="Upload CAD File (STEP/STL)"),
+    fn=process_step_file,
+    inputs=gr.File(label="Upload STEP File"),
     outputs=[
-        gr.File(label="Download Generated APDL Script"),
+        gr.File(label="Download STL File"),
+        gr.File(label="Download APDL Script"),
         gr.Image(label="Stress Plot"),
         gr.Image(label="Displacement Plot"),
     ],