Update apdl_generator/apdl_beam.py

apdl_generator/apdl_beam.py

@@ -1,32 +1,24 @@
-# apdl_generator/apdl_beam.py
-
 def generate_beam_apdl(length, width, thickness, load):
-    """
-    Generate APDL script for a beam simulation.
-
-    Parameters:
-        length (float): Length of the beam in mm.
-        width (float): Width of the beam in mm.
-        thickness (float): Thickness of the beam in mm.
-        load (float): Load applied to the beam in N.
-
-    Returns:
-        str: Path to the generated APDL file.
-    """
-    # Generate the APDL script for a beam
     apdl_script = f"""
-    ! APDL script for Beam Simulation
-    /PREP7
-    BLC4, 0, {length}, {width}, {thickness}  ! Create beam geometry
-    /SOLU
-    F, 1, FX, {load}  ! Apply load at node 1
-    SOLVE
-    FINISH
-    """
-    
-    # Save APDL script to a file
-    apdl_file_path = "/tmp/generated_beam_apdl.txt"
-    with open(apdl_file_path, "w") as f:
-        f.write(apdl_script)
-
-    return apdl_file_path
+/PREP7
+MP,EX,1,2E11
+MP,PRXY,1,0.3
+BLOCK,0,{length},0,{width},0,{thickness}
+ET,1,SOLID185
+ESIZE,5
+VMESH,ALL
+NSEL,S,LOC,X,0
+D,ALL,ALL
+NSEL,S,LOC,X,{length}
+F,ALL,FY,-{load}
+/SOLU
+ANTYPE,STATIC
+SOLVE
+/POST1
+PRNSOL,S,EQV
+PRNSOL,U,SUM
+/EXIT
+"""
+    with open("beam_simulation.inp", "w") as file:
+        file.write(apdl_script)
+    return "beam_simulation.inp"