Update app.py

app.py

@@ -1,45 +1,37 @@
 import os
 import gradio as gr
-from OCC.Core.STEPControl import STEPControl_Reader
-from OCC.Core.BRepTools import breptools_Write
-from OCC.Extend.DataExchange import write_stl_file
+import gmsh
 from ansys.mapdl.core import launch_mapdl
 
 
-def load_step_and_extract_geometry(step_file_path):
+def process_step_file(step_file_path):
     """
-    Load a STEP file and extract geometry information using pythonocc-core.
+    Load STEP file using GMSH and extract geometry.
     Parameters:
         step_file_path: Path to the uploaded STEP file.
     Returns:
         STL file path, dimensions (length, width, height).
     """
-    # 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.")
+    gmsh.initialize()
+    gmsh.open(step_file_path)
 
-    # Transfer the file into shapes
-    step_reader.TransferRoots()
-    shape = step_reader.OneShape()
+    # Extract bounding box dimensions
+    _, bbox_min, bbox_max = gmsh.model.getBoundingBox(0, 0)
+    length = bbox_max[0] - bbox_min[0]
+    width = bbox_max[1] - bbox_min[1]
+    height = bbox_max[2] - bbox_min[2]
 
     # 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()
+    gmsh.write(stl_file_path)
+    gmsh.finalize()
 
     return stl_file_path, length, width, height
 
 
 def generate_apdl_script(length, width, height):
     """
-    Generate an APDL script based on dimensions.
+    Generate APDL script based on dimensions.
     Parameters:
         length: Length of the model.
         width: Width of the model.
@@ -94,16 +86,16 @@ def run_simulation(apdl_script_path):
     return stress_plot, displacement_plot
 
 
-def process_step_file(step_file):
+def process_cad_file(cad_file):
     """
     Complete workflow for STEP file: Load -> Generate APDL -> Simulate.
     Parameters:
-        step_file: Uploaded STEP file.
+        cad_file: Uploaded STEP file.
     Returns:
         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)
+    stl_file, length, width, height = process_step_file(cad_file.name)
 
     # Generate APDL script
     apdl_script = generate_apdl_script(length, width, height)
@@ -116,7 +108,7 @@ def process_step_file(step_file):
 
 # Gradio Interface
 app = gr.Interface(
-    fn=process_step_file,
+    fn=process_cad_file,
     inputs=gr.File(label="Upload STEP File"),
     outputs=[
         gr.File(label="Download STL File"),