Update app.py

app.py

@@ -1,5 +1,7 @@
 import os
 import gradio as gr
+from ansys.mapdl.core import launch_mapdl
+
 
 def generate_apdl_script(step_file_path, material_e, material_nu, mesh_size, load_value):
     """
@@ -59,10 +61,51 @@ prnsol, u, comp
 
     return apdl_file_path
 
+
+def run_simulation(step_file, material_e, material_nu, mesh_size, load_value):
+    """
+    Generate the APDL script, run the simulation, and fetch results.
+    """
+    # Generate the APDL script
+    apdl_file_path = generate_apdl_script(step_file.name, material_e, material_nu, mesh_size, load_value)
+
+    # Launch ANSYS MAPDL (ensure it's installed and licensed)
+    mapdl = launch_mapdl()
+
+    # Upload STEP file and APDL script to MAPDL
+    mapdl.upload(step_file.name)
+    mapdl.input(apdl_file_path)
+
+    # Run the APDL script
+    mapdl.run("/SOLU")
+    mapdl.solve()
+
+    # Post-process results
+    stress_output = mapdl.result.principal_stress()
+    displacement_output = mapdl.result.nodal_displacement()
+
+    # Save outputs
+    stress_file = "/tmp/stress_results.txt"
+    with open(stress_file, "w") as f:
+        f.write("Principal Stress Results:\n")
+        f.write(str(stress_output))
+
+    displacement_file = "/tmp/displacement_results.txt"
+    with open(displacement_file, "w") as f:
+        f.write("Nodal Displacement Results:\n")
+        f.write(str(displacement_output))
+
+    # Stop MAPDL
+    mapdl.exit()
+
+    return stress_file, displacement_file
+
+
 # Gradio interface
 def process_step_file(step_file, material_e, material_nu, mesh_size, load_value):
-    apdl_file = generate_apdl_script(step_file.name, material_e, material_nu, mesh_size, load_value)
-    return apdl_file
+    stress_file, displacement_file = run_simulation(step_file, material_e, material_nu, mesh_size, load_value)
+    return stress_file, displacement_file
+
 
 # Define Gradio Interface
 with gr.Blocks() as app:
@@ -74,12 +117,13 @@ with gr.Blocks() as app:
         load_value = gr.Number(label="Load Value (N)", value=1000)
 
     submit_button = gr.Button("Submit")
-    output = gr.File(label="Download Generated APDL Script")
+    stress_output = gr.File(label="Download Stress Results")
+    displacement_output = gr.File(label="Download Displacement Results")
 
     submit_button.click(
         process_step_file,
         inputs=[step_file, material_e, material_nu, mesh_size, load_value],
-        outputs=output,
+        outputs=[stress_output, displacement_output],
     )
 
 if __name__ == "__main__":