Update app.py

app.py

@@ -2,25 +2,52 @@ import gradio as gr
 import sys
 import os
 
-
-# Update with the correct paths for your system
-sys.path.append(r"C:\Program Files\FreeCAD 1.0\bin")  # FreeCAD bin path
-sys.path.append(r"C:\Program Files\FreeCAD 1.0\lib")  # FreeCAD lib path
+# 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
-    print("FreeCAD module imported successfully.")
+    import FreeCAD, Fem, FemGui
+    from FreeCAD import Gui
+    print("FreeCAD modules imported successfully.")
 except ImportError as e:
-    print(f"Error importing FreeCAD module: {e}")
-
-
-def run_fea(step_file, output_dir):
-    """Run FEA using FreeCAD."""
+    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
+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)
+    obj.Shape = FreeCAD.importShape(step_file.name)
     doc.recompute()
 
     # Assign material properties
@@ -67,7 +94,7 @@ def run_fea(step_file, output_dir):
     results_obj.NodeCount = len(results.Mesh.Nodes)
     doc.recompute()
 
-    # Save views as images
+    # 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")
 
@@ -75,33 +102,47 @@ def run_fea(step_file, output_dir):
     Gui.ActiveDocument.ActiveView.viewIsometric()
     Gui.ActiveDocument.ActiveView.saveImage(three_d_image, 1024, 768, "White")
 
-    doc.saveAs(os.path.join(output_dir, "FEM_Analysis.FCStd"))
+    # 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
+    return two_d_image, three_d_image, apdl_file
 
-def fea_workflow(file):
-    """Gradio wrapper for FEA."""
+# 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)
-    step_file = file.name
 
     try:
-        two_d_image, three_d_image = run_fea(step_file, output_dir)
-        return two_d_image, three_d_image
+        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
+        return f"Error during FEA: {e}", None, None, None
 
 # Gradio Interface
 with gr.Blocks() as app:
-    gr.Markdown("# FreeCAD FEA for 2D and 3D Views")
-
+    gr.Markdown("# FreeCAD and APDL-Based FEA Workflow")
+    
     with gr.Row():
         step_file = gr.File(label="Upload STEP File")
-        view_2d = gr.Image(label="2D View")
-        view_3d = gr.Image(label="3D View")
+        apdl_editor = gr.Textbox(value=apdl_script, label="APDL Script", lines=20)
 
-    step_file.upload(fea_workflow, inputs=step_file, outputs=[view_2d, view_3d])
+    with gr.Row():
+        execute_button = gr.Button("Submit")
 
-app.launch()
+    with gr.Row():
+        status = gr.Text(label="Execution Status")
+        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])
 
+app.launch()