Update app.py

app.py

@@ -2,74 +2,79 @@ import gradio as gr
 import cadquery as cq
 from cadquery import exporters
 import tempfile
-import pyvista as pv
 import trimesh
 
-def load_and_display_step(file):
+# Generate APDL Script
+def generate_apdl_script(file, press_force, material):
     try:
+        # Save uploaded STEP file temporarily
         with tempfile.NamedTemporaryFile(delete=False, suffix=".step") as temp_file:
             temp_file.write(file.read())
-            temp_path = temp_file.name
+            step_file_path = temp_file.name
 
-        model = cq.importers.importStep(temp_path)
+        # Load STEP file with CadQuery
+        model = cq.importers.importStep(step_file_path)
+
+        # Export as STL for meshing
         stl_file = "temp_model.stl"
         exporters.export(model, stl_file)
 
-        # Render with PyVista
-        mesh = pv.read(stl_file)
-        plotter = pv.Plotter(off_screen=True)
-        image_file = "rendered_output.png"
-        plotter.add_mesh(mesh, color="lightblue")
-        plotter.screenshot(image_file)
-        plotter.close()
+        # Load STL with trimesh
+        mesh = trimesh.load(stl_file)
+        volume = mesh.volume
+        surface_area = mesh.area
 
-        return f"STEP file processed successfully! Render saved as {image_file}"
-    except Exception as e:
-        return f"Error: {str(e)}"
+        # Placeholder stress calculation
+        stress = press_force / surface_area if surface_area > 0 else "Undefined (Area is zero)"
 
-def analyze_stl(file):
-    try:
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".stl") as temp_file:
-            temp_file.write(file.read())
-            temp_path = temp_file.name
+        # Generate APDL script
+        apdl_script = f"""\
+/PREP7
+! Importing Geometry
+/import, '{step_file_path}', geom, STEP
+! Material Properties
+MP, EX, 1, {material['elastic_modulus']}
+MP, PRXY, 1, {material['poisson']}
+! Load and Boundary Conditions
+F, NODE, ALL, FX, {press_force}
+! Solve
+/SOLU
+SOLVE
+/POST1
+*GET, stress, NODE, 0, S, MAX
+/EXIT, SAVE
+"""
+        # Save APDL script to file
+        apdl_file_path = "simulation_script.inp"
+        with open(apdl_file_path, "w") as apdl_file:
+            apdl_file.write(apdl_script)
 
-        mesh = trimesh.load(temp_path)
-        return f"Volume: {mesh.volume:.3f} m³\nSurface Area: {mesh.area:.3f} m²"
+        return apdl_script, apdl_file_path
     except Exception as e:
-        return f"Error: {str(e)}"
-
-def stress_analysis(press_force, area):
-    try:
-        if area <= 0:
-            return "Error: Area must be greater than zero."
-        stress = press_force / area
-        return f"Calculated Stress: {stress:.2f} N/m²"
-    except Exception as e:
-        return f"Error: {str(e)}"
+        return f"Error: {str(e)}", None
 
+# Gradio App
 def main():
     with gr.Blocks() as app:
-        gr.Markdown("# Press Tool Design Simulation")
+        gr.Markdown("# APDL Script Generator for Press Tool Design")
 
         with gr.Row():
             step_input = gr.File(label="Upload STEP File")
-            step_output = gr.Textbox(label="STEP File Status")
-            step_submit = gr.Button("Load and Display STEP File")
-
-        with gr.Row():
-            stl_input = gr.File(label="Upload STL File")
-            stl_output = gr.Textbox(label="STL Analysis Result")
-            stl_submit = gr.Button("Analyze STL File")
-
-        with gr.Row():
             force_input = gr.Number(label="Press Force (N)", value=1000)
-            area_input = gr.Number(label="Contact Area (m²)", value=0.01)
-            stress_output = gr.Textbox(label="Stress Analysis Result")
-            stress_submit = gr.Button("Calculate Stress")
+            material_input = gr.Textbox(
+                label="Material Properties (JSON)",
+                value='{"elastic_modulus": 2e11, "poisson": 0.3}'
+            )
+            script_output = gr.Textbox(label="Generated APDL Script")
+            download_button = gr.File(label="Download Script")
+            generate_button = gr.Button("Generate APDL Script")
 
-        step_submit.click(fn=load_and_display_step, inputs=step_input, outputs=step_output)
-        stl_submit.click(fn=analyze_stl, inputs=stl_input, outputs=stl_output)
-        stress_submit.click(fn=stress_analysis, inputs=[force_input, area_input], outputs=stress_output)
+        # Events
+        generate_button.click(
+            fn=generate_apdl_script,
+            inputs=[step_input, force_input, material_input],
+            outputs=[script_output, download_button],
+        )
 
     app.launch()