Update app.py

app.py

@@ -1,113 +1,91 @@
-import gradio as gr
-import subprocess
+import FreeCAD, Fem, FemGui
+from FreeCAD import Gui
 import os
 
-# APDL Script for ANSYS
-apdl_script = """/CLEAR            ! Clear any existing database
-/PREP7            ! Enter preprocessor
-/CWD, '{cwd}'     ! Set the working directory
-! Define material properties
-MP, EX, 1, 210E3  ! Young's Modulus (e.g., Steel)
-MP, PRXY, 1, 0.3  ! Poisson's Ratio
-! Define geometry (e.g., a rectangular plate)
-RECTNG, 0, 100, 0, 50  ! Rectangle from (0,0) to (100,50)
-! Define element type and mesh
-ET, 1, PLANE183   ! 2D structural element
-ESIZE, 5          ! Set element size
-AMESH, ALL        ! Mesh the area
-! Apply boundary conditions
-D, NODE(0, 0), ALL, 0     ! Fix the bottom-left corner
-D, NODE(100, 0), UY, 0    ! Restrict vertical movement at bottom-right
-! Apply loads
-F, NODE(50, 50), FX, 1000 ! Apply a horizontal load at the top-middle node
-! Solve
-FINISH           ! Exit preprocessor
-/SOLU            ! Enter solution processor
-ANTYPE, 0        ! Static analysis
-SOLVE            ! Solve the system
-FINISH           ! Exit solution processor
-/POST1           ! Enter post-processor
-/SHOW, JPEG          ! Set output format to JPEG
-/DEVICE, HARD, JPEG  ! Configure device for image output
-PLNSOL, U, SUM       ! Plot nodal displacement (2D view)
-FILE, '2D_View', 'JPEG'  ! Save 2D view as an image
-/SHOW, CLOSE         ! Close the graphical window
-/VIEW, 1, 1, 1, 1    ! Set 3D view (isometric)
-/SHOW, JPEG          ! Set output format to JPEG
-/DEVICE, HARD, JPEG  ! Configure device for image output
-PLNSOL, S, EQV       ! Plot equivalent stress (3D view)
-FILE, '3D_View', 'JPEG'  ! Save 3D view as an image
-/SHOW, CLOSE         ! Close the graphical window
-FINISH
-"""
-
-def upload_step_file(file):
-    """Handle STEP file upload."""
-    return f"STEP file '{file.name}' uploaded successfully."
-
-def modify_apdl_script(modified_script):
-    """Modify and save the APDL script."""
-    global apdl_script
-    apdl_script = modified_script
-    return "APDL script updated successfully."
-
-def execute_script():
-    """Execute the APDL script using ANSYS."""
-    global apdl_script
-
-    # Set the working directory
-    cwd = os.getcwd()
-    apdl_script_with_cwd = apdl_script.format(cwd=cwd)
-
-    # Save the APDL script to a file
-    script_file = "script.dat"
-    with open(script_file, "w") as f:
-        f.write(apdl_script_with_cwd)
-
-    # Define ANSYS execution command
-    ansys_command = [
-        "C:\\Program Files\\ANSYS Inc\\v211\\ansys\\bin\\winx64\\ansys.exe",  # Replace with your actual path
-        "-b",
-        "-i", script_file,
-        "-o", "output.log"
-    ]
-
-    try:
-        # Run the ANSYS command
-        subprocess.run(ansys_command, check=True)
-
-        # Check for image outputs
-        two_d_image = os.path.join(cwd, "2D_View.jpg")
-        three_d_image = os.path.join(cwd, "3D_View.jpg")
-
-        if os.path.exists(two_d_image) and os.path.exists(three_d_image):
-            return "APDL script executed successfully. 2D and 3D views generated."
-        else:
-            with open("output.log", "r") as log_file:
-                log_contents = log_file.read()
-            return f"APDL script executed, but images not found. Check log:\n{log_contents}"
-
-    except subprocess.CalledProcessError as e:
-        return f"Error executing APDL script: {str(e)}. Check output.log for details."
-    except FileNotFoundError:
-        return "ANSYS executable not found. Verify the path in ansys_command."
-
-def get_generated_images():
-    """Return the generated 2D and 3D view images if they exist."""
-    two_d_image = os.path.join(os.getcwd(), "2D_View.jpg")
-    three_d_image = os.path.join(os.getcwd(), "3D_View.jpg")
-    if os.path.exists(two_d_image) and os.path.exists(three_d_image):
-        return two_d_image, three_d_image
-    else:
-        return None, None
-
-def read_output_log():
-    """Read and return the contents of output.log."""
-    if os.path.exists("output.log"):
-        with open("output.log", "r") as f:
-            return f.read()
-    else:
-        return "No output log found."
+def run_fea(step_file, output_dir):
+    """
+    Automates FEA analysis in FreeCAD using the FEM Workbench.
+
+    Args:
+        step_file (str): Path to the STEP file.
+        output_dir (str): Directory to save output files (images and analysis data).
+    """
+    # Create a new FreeCAD document
+    doc = FreeCAD.newDocument("FEM_Analysis")
+    
+    # Import STEP file
+    obj = doc.addObject("Part::Feature", "ImportedPart")
+    obj.Shape = FreeCAD.importShape(step_file)
+    doc.recompute()
+
+    # Assign material properties
+    mat = doc.addObject("App::MaterialObjectPython", "Material")
+    mat.Material = {
+        "Name": "Steel",
+        "YoungsModulus": 210e9,  # Pa (N/m²)
+        "PoissonRatio": 0.3,
+    }
+    doc.addObject("Fem::ConstraintMaterial", "MaterialConstraint").References = [(obj, "Face1")]  # Adjust face as needed
+    doc.recompute()
+
+    # Create and configure the mesh
+    mesh = doc.addObject("Fem::FemMeshShapeNetgenObject", "FEMMesh")
+    mesh.Shape = obj.Shape
+    mesh.MaxSize = 5  # Set mesh size (adjust as required)
+    doc.recompute()
+
+    # Apply boundary conditions
+    fixed_constraint = doc.addObject("Fem::ConstraintFixed", "FixedConstraint")
+    fixed_constraint.References = [(obj, "Face1")]  # Adjust face as needed
+    doc.recompute()
+
+    # Apply force
+    force_constraint = doc.addObject("Fem::ConstraintForce", "ForceConstraint")
+    force_constraint.References = [(obj, "Face2")]  # Adjust face as needed
+    force_constraint.Force = 1000  # N
+    doc.recompute()
+
+    # Add solver (CalculiX)
+    solver = doc.addObject("Fem::SolverCalculixCcxTools", "Solver")
+    solver.InputFile = os.path.join(output_dir, "input.inp")
+    solver.WorkingDir = output_dir
+    doc.recompute()
+
+    # Write input file and run the solver
+    solver.write_input_file()
+    solver.run()
+
+    # Load and process results
+    results = solver.load_results()
+    results_obj = doc.addObject("Fem::FemResultObject", "Results")
+    results_obj.Mesh = results.Mesh
+    results_obj.NodeCount = len(results.Mesh.Nodes)
+    doc.recompute()
+
+    # Save 2D and 3D views as images
+    two_d_image_path = os.path.join(output_dir, "2D_View.png")
+    three_d_image_path = os.path.join(output_dir, "3D_View.png")
+
+    Gui.ActiveDocument.ActiveView.saveImage(two_d_image_path, 1024, 768, "White")
+    Gui.ActiveDocument.ActiveView.viewIsometric()
+    Gui.ActiveDocument.ActiveView.saveImage(three_d_image_path, 1024, 768, "White")
+
+    # Save the FreeCAD project file
+    doc.saveAs(os.path.join(output_dir, "FEM_Analysis.FCStd"))
+
+    print(f"Analysis completed successfully.")
+    print(f"2D View saved to: {two_d_image_path}")
+    print(f"3D View saved to: {three_d_image_path}")
+
+# Example Usage
+if __name__ == "__main__":
+    step_file = "path_to_your_step_file.stp"  # Replace with your STEP file path
+    output_dir = "path_to_output_directory"  # Replace with your desired output directory
+
+    if not os.path.exists(output_dir):
+        os.makedirs(output_dir)
+
+    run_fea(step_file, output_dir)
 
 # Gradio interface
 with gr.Blocks() as app: