Update app.py

app.py

@@ -1,148 +1,140 @@
 import gradio as gr
-import sys
+import subprocess
 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)
-
-# Default APDL Script
-apdl_script = """
-/CLEAR
-/PREP7
-MP, EX, 1, 210E3
-MP, PRXY, 1, 0.3
-RECTNG, 0, 100, 0, 50
-ET, 1, PLANE183
-ESIZE, 5
-AMESH, ALL
-D, NODE(0, 0), ALL, 0
-D, NODE(100, 0), UY, 0
-F, NODE(50, 50), FX, 1000
-FINISH
-/SOLU
-ANTYPE, 0
-SOLVE
-FINISH
-/POST1
-PLNSOL, U, SUM
-/IMAGE, SAVE, '2D_View', JPG
-/VIEW, 1, 1, 1, 1
-PLNSOL, S, EQV
-/IMAGE, SAVE, '3D_View', JPG
+# 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 run_fea_with_apdl(step_file, apdl_script, output_dir):
-    """Run FEA using FreeCAD and APDL script."""
-    doc = FreeCAD.newDocument("FEM_Analysis")
-    
-    # Import STEP file
-    obj = doc.addObject("Part::Feature", "ImportedPart")
-    obj.Shape = FreeCAD.importShape(step_file.name)
-    doc.recompute()
-
-    # Assign material properties
-    mat = doc.addObject("App::MaterialObjectPython", "Material")
-    mat.Material = {
-        "Name": "Steel",
-        "YoungsModulus": 210e9,  # Pa
-        "PoissonRatio": 0.3,
-    }
-    doc.addObject("Fem::ConstraintMaterial", "MaterialConstraint").References = [(obj, "Face1")]
-    doc.recompute()
-
-    # Create a mesh
-    mesh = doc.addObject("Fem::FemMeshShapeNetgenObject", "FEMMesh")
-    mesh.Shape = obj.Shape
-    mesh.MaxSize = 5
-    doc.recompute()
-
-    # Apply boundary conditions
-    fixed_constraint = doc.addObject("Fem::ConstraintFixed", "FixedConstraint")
-    fixed_constraint.References = [(obj, "Face1")]
-    doc.recompute()
-
-    # Apply a force
-    force_constraint = doc.addObject("Fem::ConstraintForce", "ForceConstraint")
-    force_constraint.References = [(obj, "Face2")]
-    force_constraint.Force = 1000  # N
-    doc.recompute()
-
-    # Add solver
-    solver = doc.addObject("Fem::SolverCalculixCcxTools", "Solver")
-    solver.InputFile = os.path.join(output_dir, "input.inp")
-    solver.WorkingDir = output_dir
-    doc.recompute()
-
-    # Solve
-    solver.write_input_file()
-    solver.run()
-
-    # 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/3D views as images
-    two_d_image = os.path.join(output_dir, "2D_View.png")
-    three_d_image = os.path.join(output_dir, "3D_View.png")
-
-    Gui.ActiveDocument.ActiveView.saveImage(two_d_image, 1024, 768, "White")
-    Gui.ActiveDocument.ActiveView.viewIsometric()
-    Gui.ActiveDocument.ActiveView.saveImage(three_d_image, 1024, 768, "White")
-
-    # Save APDL Script to File
-    apdl_file = os.path.join(output_dir, "script.dat")
-    with open(apdl_file, "w") as f:
-        f.write(apdl_script)
-
-    return two_d_image, three_d_image, apdl_file
-
-# Gradio Workflow
-def fea_workflow(file, apdl_script):
-    """Gradio workflow for FEA with APDL integration."""
-    output_dir = "output"
-    os.makedirs(output_dir, exist_ok=True)
+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:
-        two_d_image, three_d_image, apdl_file = run_fea_with_apdl(file, apdl_script, output_dir)
-        return (
-            "Analysis completed successfully!",
-            two_d_image,
-            three_d_image,
-            f"APDL script saved to: {apdl_file}"
-        )
-    except Exception as e:
-        return f"Error during FEA: {e}", None, None, None
-
-# Gradio Interface
+        # 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."
+
+# Gradio interface
 with gr.Blocks() as app:
-    gr.Markdown("# FreeCAD and APDL-Based FEA Workflow")
-    
+    gr.Markdown("# ANSYS APDL Script for 2D and 3D Views")
+
     with gr.Row():
         step_file = gr.File(label="Upload STEP File")
-        apdl_editor = gr.Textbox(value=apdl_script, label="APDL Script", lines=20)
+        step_output = gr.Text(label="Upload Status")
+        step_file.upload(upload_step_file, inputs=step_file, outputs=step_output)
 
     with gr.Row():
-        execute_button = gr.Button("Submit")
+        apdl_view = gr.Textbox(apdl_script, lines=20, label="APDL Script")
+        modify_button = gr.Button("Save Changes")
+        modify_button.click(modify_apdl_script, inputs=apdl_view, outputs=step_output)
 
     with gr.Row():
-        status = gr.Text(label="Execution Status")
+        execute_button = gr.Button("Run Script")
+        execution_output = gr.Text(label="Execution Status")
+        log_output = gr.Textbox(label="Output Log", lines=10, interactive=False)
+        execute_button.click(execute_script, outputs=[execution_output])
+        execute_button.click(read_output_log, outputs=log_output)
+
+    with gr.Row():
+        gr.Markdown("## Generated Images")
         view_2d = gr.Image(label="2D View", interactive=False)
         view_3d = gr.Image(label="3D View", interactive=False)
-        apdl_path = gr.Text(label="APDL Script Path")
-
-    execute_button.click(fea_workflow, inputs=[step_file, apdl_editor], outputs=[status, view_2d, view_3d, apdl_path])
+        refresh_button = gr.Button("Refresh Images")
+        refresh_button.click(get_generated_images, outputs=[view_2d, view_3d])
 
 app.launch()