Update app.py

app.py

@@ -1,47 +1,30 @@
+from utils.cad_utils import generate_cad_model
+from utils.simulation_utils import run_simulation
+from utils.gcode_utils import generate_gcode
 import gradio as gr
-import FreeCAD, Part
-from ansys.mapdl.core import launch_mapdl
 
-def generate_and_simulate(part_name, length, width, height, material_property):
-    # Step 1: Create CAD Model with FreeCAD
-    try:
-        doc = FreeCAD.newDocument(part_name)
-        box = Part.makeBox(length, width, height)
-        Part.show(box)
-        step_file = f"{part_name}.step"
-        box.exportStep(step_file)
-    except Exception as e:
-        return f"Error in FreeCAD CAD generation: {e}"
+def automate_workflow(dimensions, material, tolerances):
+    # Step 1: Generate CAD model
+    cad_file = generate_cad_model(dimensions)
 
-    # Step 2: Simulate in ANSYS
-    try:
-        mapdl = launch_mapdl()
-        mapdl.prep7()
-        mapdl.et(1, "SOLID186")
-        mapdl.mp("EX", 1, material_property)  # Elastic modulus
-        mapdl.input(step_file)
-        mapdl.vmesh("ALL")
-        mapdl.solve()
-        displacement = mapdl.post_processing.nodal_displacement()
-        mapdl.exit()
-    except Exception as e:
-        return f"Error in ANSYS simulation: {e}"
+    # Step 2: Run simulation
+    simulation_results = run_simulation(cad_file)
 
-    return f"Simulation completed! Displacement results: {displacement}"
+    # Step 3: Generate G-Code
+    gcode_file = generate_gcode(cad_file)
 
-# Gradio Interface
-iface = gr.Interface(
-    fn=generate_and_simulate,
+    return f"Simulation Results: {simulation_results}", gcode_file
+
+interface = gr.Interface(
+    fn=automate_workflow,
     inputs=[
-        gr.Textbox(label="Part Name", placeholder="Enter the part name"),
-        gr.Number(label="Length (mm)"),
-        gr.Number(label="Width (mm)"),
-        gr.Number(label="Height (mm)"),
-        gr.Number(label="Material Elastic Modulus (Pa)"),
+        gr.Textbox(label="Dimensions (e.g., 100x50x30 mm)"),
+        gr.Textbox(label="Material (e.g., Steel)"),
+        gr.Textbox(label="Tolerances (e.g., ±0.05 mm)")
     ],
-    outputs="text",
-    title="FreeCAD and ANSYS Automation",
-    description="Generate CAD models and perform ANSYS simulations automatically!"
+    outputs=["text", "file"],
+    title="End-to-End Automation Workflow"
 )
 
-iface.launch()
+if __name__ == "__main__":
+    interface.launch()