Update app.py

app.py

@@ -5,6 +5,7 @@ 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
@@ -25,41 +26,47 @@ FINISH           ! Exit preprocessor
 ANTYPE, 0        ! Static analysis
 SOLVE            ! Solve the system
 FINISH           ! Exit solution processor
-! Post-processing for 2D view
 /POST1           ! Enter post-processor
-PLNSOL, U, SUM   ! Plot nodal displacement (2D view)
-/IMAGE, SAVE, '2D_View', JPG  ! Save 2D view as an image
-! Post-processing for 3D view
-/VIEW, 1, 1, 1, 1   ! Set 3D view (isometric)
-/DV3D               ! Enable dynamic 3D view
-PLNSOL, S, EQV      ! Plot equivalent stress (3D view)
-/IMAGE, SAVE, '3D_View', JPG  ! Save 3D view as an image
-! Exit ANSYS
+/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):
-    """Function to handle STEP file upload."""
+    """Handle STEP file upload."""
     return f"STEP file '{file.name}' uploaded successfully."
 
 def modify_apdl_script(modified_script):
-    """Function to modify and save the APDL script."""
+    """Modify and save the APDL script."""
     global apdl_script
     apdl_script = modified_script
     return "APDL script updated successfully."
 
 def execute_script():
-    """Function to execute the APDL script using ANSYS."""
+    """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)
+        f.write(apdl_script_with_cwd)
 
     # Define ANSYS execution command
     ansys_command = [
-        "C:\\Program Files\\ANSYS Inc\\v211\\ansys.exe",  # Replace with actual path
+        "C:\\Program Files\\ANSYS Inc\\v211\\ansys\\bin\\winx64\\ansys.exe",  # Replace with your actual path
         "-b",
         "-i", script_file,
         "-o", "output.log"
@@ -68,23 +75,29 @@ def execute_script():
     try:
         # Run the ANSYS command
         subprocess.run(ansys_command, check=True)
-        
-        # Check if 2D and 3D images exist
-        if os.path.exists("2D_View.jpg") and os.path.exists("3D_View.jpg"):
+
+        # 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:
-            return "APDL script executed, but 2D/3D images are missing. Check output.log for details."
-    
+            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."""
-    if os.path.exists("2D_View.jpg") and os.path.exists("3D_View.jpg"):
-        return "2D_View.jpg", "3D_View.jpg"
+    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