Update app.py

app.py

@@ -1,4 +1,31 @@
 import gradio as gr
+import subprocess
+import os
+
+def execute_script():
+    """Function to execute the APDL script using ANSYS."""
+    global apdl_script
+
+    # Save the APDL script to a file
+    script_file = "script.dat"
+    with open(script_file, "w") as f:
+        f.write(apdl_script)
+    
+    # Define ANSYS execution command
+    ansys_command = [
+        "ansys",  # Replace with the full path to your ANSYS executable, e.g., "C:\\Program Files\\ANSYS Inc\\v211\\ansys.exe"
+        "-b",     # Batch mode
+        "-i", script_file,  # Input script
+        "-o", "output.log"  # Output log
+    ]
+    
+    try:
+        # Run the command
+        subprocess.run(ansys_command, check=True)
+        return "APDL script executed. Check generated views (2D_View.jpg, 3D_View.jpg)."
+    except subprocess.CalledProcessError as e:
+        return f"Error executing APDL script: {e}"
+
 
 # APDL Script for ANSYS
 apdl_script = """/CLEAR            ! Clear any existing database
@@ -88,5 +115,23 @@ with gr.Blocks() as app:
     modify_button.click(modify_apdl_script, inputs=apdl_view, outputs=step_output)
     execute_button.click(execute_script, outputs=execution_output)
 
+
+if os.path.exists("2D_View.jpg") and os.path.exists("3D_View.jpg"):
+    print("Images generated successfully.")
+else:
+    print("Error: Images not found.")
+def get_generated_images():
+    """Return the generated 2D and 3D view images."""
+    return "2D_View.jpg", "3D_View.jpg"
+
+# Add this to the Gradio interface
+with gr.Row():
+    gr.Markdown("## Generated Images")
+    view_2d = gr.Image(label="2D View")
+    view_3d = gr.Image(label="3D View")
+    refresh_button = gr.Button("Refresh Images")
+
+refresh_button.click(get_generated_images, outputs=[view_2d, view_3d])
+
 # Launch the app
 app.launch()