Update app.py

app.py

@@ -1,26 +1,22 @@
 import FreeCAD, Fem, FemGui
 from FreeCAD import Gui
-import os
 import sys
-sys.path.append(r"C:\Program Files\FreeCAD 1.0\bin")  # Path to FreeCAD's Python library
-sys.path.append(r"C:\Program Files\FreeCAD 1.0\lib")  # Additional FreeCAD modules
+import os
+
+# Add FreeCAD library paths
+sys.path.append(r"C:\Program Files\FreeCAD 1.0\bin")  # Update with your FreeCAD bin path
+sys.path.append(r"C:\Program Files\FreeCAD 1.0\lib")  # Update with your FreeCAD lib path
 
 try:
     import FreeCAD, Fem, FemGui
+    from FreeCAD import Gui
     print("FreeCAD modules imported successfully.")
 except ImportError as e:
     print(f"Error importing FreeCAD modules: {e}")
-
+    sys.exit(1)
 
 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
+    """Run FEA using FreeCAD."""
     doc = FreeCAD.newDocument("FEM_Analysis")
     
     # Import STEP file
@@ -32,71 +28,68 @@ def run_fea(step_file, output_dir):
     mat = doc.addObject("App::MaterialObjectPython", "Material")
     mat.Material = {
         "Name": "Steel",
-        "YoungsModulus": 210e9,  # Pa (N/m²)
+        "YoungsModulus": 210e9,  # Pa
         "PoissonRatio": 0.3,
     }
-    doc.addObject("Fem::ConstraintMaterial", "MaterialConstraint").References = [(obj, "Face1")]  # Adjust face as needed
+    doc.addObject("Fem::ConstraintMaterial", "MaterialConstraint").References = [(obj, "Face1")]
     doc.recompute()
 
-    # Create and configure the mesh
+    # Create a mesh
     mesh = doc.addObject("Fem::FemMeshShapeNetgenObject", "FEMMesh")
     mesh.Shape = obj.Shape
-    mesh.MaxSize = 5  # Set mesh size (adjust as required)
+    mesh.MaxSize = 5
     doc.recompute()
 
     # Apply boundary conditions
     fixed_constraint = doc.addObject("Fem::ConstraintFixed", "FixedConstraint")
-    fixed_constraint.References = [(obj, "Face1")]  # Adjust face as needed
+    fixed_constraint.References = [(obj, "Face1")]
     doc.recompute()
 
-    # Apply force
+    # Apply a force
     force_constraint = doc.addObject("Fem::ConstraintForce", "ForceConstraint")
-    force_constraint.References = [(obj, "Face2")]  # Adjust face as needed
+    force_constraint.References = [(obj, "Face2")]
     force_constraint.Force = 1000  # N
     doc.recompute()
 
-    # Add solver (CalculiX)
+    # Add solver
     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
+    # Solve
     solver.write_input_file()
     solver.run()
 
-    # Load and process results
+    # Save 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")
+    # Save views as images
+    Gui.ActiveDocument.ActiveView.saveImage(os.path.join(output_dir, "2D_View.png"), 1024, 768, "White")
     Gui.ActiveDocument.ActiveView.viewIsometric()
-    Gui.ActiveDocument.ActiveView.saveImage(three_d_image_path, 1024, 768, "White")
+    Gui.ActiveDocument.ActiveView.saveImage(os.path.join(output_dir, "3D_View.png"), 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}")
+    print(f"2D View saved to {output_dir}/2D_View.png")
+    print(f"3D View saved to {output_dir}/3D_View.png")
 
-# Example Usage
+# 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
+    step_file = "path_to_your_step_file.stp"  # Replace with your STEP file
+    output_dir = "path_to_output_directory"  # Replace with your 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:
     gr.Markdown("# ANSYS APDL Script for 2D and 3D Views")