Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| 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 | |
| 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. | |
| """ | |
| # Handle optional parameters | |
| force = force if include_force else 0 | |
| load = load if include_load else 0 | |
| # Generate APDL script dynamically | |
| if use_case == "plate": | |
| hole_diameter = hole_diameter if include_hole else 0 | |
| apdl_path = generate_plate_apdl(thickness, length, width, hole_diameter, force) | |
| elif use_case == "beam": | |
| apdl_path = generate_beam_apdl(length, width, thickness, load) | |
| else: | |
| return "Invalid use case selected.", None, None, None | |
| # Read the generated APDL program | |
| with open(apdl_path, "r") as file: | |
| apdl_program = file.read() | |
| # Run simulation using Python-based solver | |
| stress, deformation = run_python_simulation(apdl_path, use_case, thickness, length, width, force, load, elastic_modulus) | |
| # Explanation for negligible deformation | |
| if deformation < 1e-6: | |
| deformation_note = "\nNote: Deformation is negligible due to the high rigidity of the material." | |
| else: | |
| deformation_note = "" | |
| # Generate 2D simulation visualization | |
| graph_path, _ = visualize_results("Python-Based Solver", length, width, thickness, stress, deformation) | |
| # Generate end-product visualization | |
| product_path = visualize_end_product(use_case, length, width, thickness, deformation) | |
| 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() | |