Update app.py

app.py

@@ -3,13 +3,11 @@ from apdl_generator.apdl_plate import generate_plate_apdl
 from apdl_generator.apdl_beam import generate_beam_apdl
 from simulators.python_simulation import run_python_simulation
 from visualization import visualize_results, visualize_end_product
-import os
 
 def simulation_workflow(tool_type, use_case, include_hole, include_force, include_load, thickness, length, width, hole_diameter, force, load, elastic_modulus):
     """
-    Main simulation workflow that generates multiple formats for APDL, Python code, and simulation results.
+    Main simulation workflow.
     """
-
     # Handle optional parameters
     force = force if include_force else 0
     load = load if include_load else 0
@@ -42,14 +40,54 @@ def simulation_workflow(tool_type, use_case, include_hole, include_force, includ
     # Generate end-product visualization
     product_path = visualize_end_product(use_case, length, width, thickness, deformation)
 
-    # Save Python code for simulation
-    python_code = f"""
-import numpy as np
-# Define material properties
-elastic_modulus = {elastic_modulus}
-thickness = {thickness}
-length = {length}
-width = {width}
-# Simulate stress and deformation (simplified example)
-def simulate():
-    stress = elastic
+    return (
+        f"{tool_type} Simulation\nStress: {stress:.2f} MPa\nDeformation: {deformation:.6f} mm{deformation_note}",
+        graph_path,
+        product_path,
+        apdl_program
+    )
+
+# Gradio interface setup
+with gr.Blocks() as interface:
+    gr.Markdown("# Punch and Die Simulation Tool")
+    
+    with gr.Row():
+        tool_type = gr.Radio(["Punch", "Die"], label="Select Tool Type")
+        use_case = gr.Radio(["plate", "beam"], label="Select Use Case")
+    
+    with gr.Row():
+        include_hole = gr.Checkbox(label="Include Hole for Plate Simulation")
+        include_force = gr.Checkbox(label="Include Force")
+        include_load = gr.Checkbox(label="Include Load")
+    
+    with gr.Row():
+        thickness = gr.Slider(10, 50, step=1, label="Thickness (mm)")
+        length = gr.Slider(100, 500, step=10, label="Length (mm)")
+        width = gr.Slider(50, 200, step=10, label="Width (mm)")
+    
+    with gr.Row():
+        hole_diameter = gr.Slider(5, 25, step=1, label="Hole Diameter (mm)")
+        force = gr.Slider(1000, 10000, step=500, label="Force (N)")
+        load = gr.Slider(1000, 20000, step=1000, label="Load (N)")
+        elastic_modulus = gr.Slider(5e10, 3e11, step=1e10, label="Elastic Modulus (Pa)")
+    
+    submit_button = gr.Button("Run Simulation")
+    
+    with gr.Row():
+        result_text = gr.Textbox(label="Simulation Results")
+        result_graph = gr.Image(label="2D Simulation Visualization")
+        result_product = gr.Image(label="End Product Visualization")
+        apdl_output = gr.Code(language="python", label="Generated APDL Program")
+    
+    # Link the button to the simulation function
+    submit_button.click(
+        simulation_workflow,
+        inputs=[
+            tool_type, use_case, include_hole, include_force, include_load,
+            thickness, length, width, hole_diameter, force, load, elastic_modulus
+        ],
+        outputs=[result_text, result_graph, result_product, apdl_output]
+    )
+
+# Launch interface
+interface.launch()