Update app.py

app.py

@@ -1,44 +1,28 @@
 import os
 import gradio as gr
-import gmsh
-from ansys.mapdl.core import launch_mapdl
+import FreeCAD
+import FreeCADGui
+import Part
 
-
-def process_step_file(step_file_path):
+def generate_apdl_from_freecad(freecad_file_path):
     """
-    Load STEP file using GMSH and extract geometry.
+    Generate an APDL script from a FreeCAD file.
     Parameters:
-        step_file_path: Path to the uploaded STEP file.
+        freecad_file_path: Path to the FreeCAD file (.FCStd).
     Returns:
-        STL file path, dimensions (length, width, height).
+        Path to the generated APDL script.
     """
-    gmsh.initialize()
-    gmsh.open(step_file_path)
+    # Open the FreeCAD file
+    doc = FreeCAD.openDocument(freecad_file_path)
+    obj = doc.Objects[0]  # Assuming a single object in the file
 
     # 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"
-    gmsh.write(stl_file_path)
-    gmsh.finalize()
-
-    return stl_file_path, length, width, height
+    bbox = obj.Shape.BoundBox
+    length = bbox.XLength
+    width = bbox.YLength
+    height = bbox.ZLength
 
-
-def generate_apdl_script(length, width, height):
-    """
-    Generate 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.
-    """
+    # Generate APDL script
     apdl_script = f"""
 /prep7
 block, 0, {length:.4f}, 0, {width:.4f}, 0, {height:.4f}
@@ -59,63 +43,29 @@ solve
     with open(apdl_path, "w") as f:
         f.write(apdl_script)
 
-    return apdl_path
-
+    # Close the FreeCAD document
+    FreeCAD.closeDocument(doc.Name)
 
-def run_simulation(apdl_script_path):
-    """
-    Run simulation using ANSYS MAPDL and return results.
-    Parameters:
-        apdl_script_path: Path to the generated APDL script.
-    Returns:
-        Stress plot and displacement plot paths.
-    """
-    # Launch ANSYS MAPDL
-    mapdl = launch_mapdl()
-
-    # Input the APDL script
-    mapdl.input(apdl_script_path)
-
-    # Extract results
-    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, displacement_plot
+    return apdl_path
 
 
-def process_cad_file(cad_file):
+def process_freecad_file(freecad_file):
     """
-    Complete workflow for STEP file: Load -> Generate APDL -> Simulate.
+    Process the FreeCAD file and generate an APDL script.
     Parameters:
-        cad_file: Uploaded STEP file.
+        freecad_file: Uploaded FreeCAD file (.FCStd).
     Returns:
-        STL file, APDL script, stress plot, displacement plot.
+        Path to the generated APDL script.
     """
-    # Extract geometry from the STEP file
-    stl_file, length, width, height = process_step_file(cad_file.name)
-
-    # Generate APDL script
-    apdl_script = generate_apdl_script(length, width, height)
-
-    # Run simulation
-    stress_plot, displacement_plot = run_simulation(apdl_script)
-
-    return stl_file, apdl_script, stress_plot, displacement_plot
+    apdl_script_path = generate_apdl_from_freecad(freecad_file.name)
+    return apdl_script_path
 
 
 # Gradio Interface
 app = gr.Interface(
-    fn=process_cad_file,
-    inputs=gr.File(label="Upload STEP File"),
-    outputs=[
-        gr.File(label="Download STL File"),
-        gr.File(label="Download APDL Script"),
-        gr.Image(label="Stress Plot"),
-        gr.Image(label="Displacement Plot"),
-    ],
+    fn=process_freecad_file,
+    inputs=gr.File(label="Upload FreeCAD File (.FCStd)"),
+    outputs=gr.File(label="Download Generated APDL Script"),
     live=True,
 )