Update app.py

app.py

@@ -1,132 +1,47 @@
 import gradio as gr
-import os
-import subprocess
 
-def process_cad_with_solidworks(cad_file_path):
+# G-code Generator Function
+def generate_gcode(length, radius1, radius2, feed_rate, depth):
     """
-    Converts the CAD file to STEP format using SolidWorks automation.
-    This requires SolidWorks to be installed and accessible via the Windows COM interface.
+    Generates G-code for a turning operation with arcs and linear moves.
+    Inputs:
+    - length: Length of the part to machine
+    - radius1: Radius of first arc (R = 0.25)
+    - radius2: Radius of second arc (R = 0.125)
+    - feed_rate: Feed rate for the CNC machine
+    - depth: Cutting depth
     """
-    import win32com.client
-
-    # Open SolidWorks
-    swApp = win32com.client.Dispatch("SldWorks.Application")
-    swApp.Visible = True
-
-    # Open the CAD file
-    model = swApp.OpenDoc6(cad_file_path, 1, 0, "", None, None)
-
-    # Save as STEP format
-    step_file_path = cad_file_path.replace(".sldprt", ".step")
-    success = model.Extension.SaveAs(step_file_path, 0, 1, None, None, None)
-
-    # Close the model
-    swApp.CloseDoc(model.GetTitle())
-    swApp.Quit()
-
-    if success:
-        return step_file_path
-    else:
-        raise Exception("Failed to convert CAD file to STEP format using SolidWorks.")
-
-def run_ansys_simulation(step_file_path):
-    """
-    Runs a simulation in ANSYS using the provided STEP file.
-    Requires ANSYS to be installed and accessible via command-line or API.
-    """
-    # Example APDL script
-    apdl_script = f"""
-    /PREP7
-    /IMPORT,STEP,{step_file_path}
-    /SOLU
-    ! Apply loads and boundary conditions here
-    SOLVE
-    FINISH
-    /POST1
-    ! Post-process results
-    """
-
-    # Save the APDL script to a temporary file
-    apdl_script_path = "simulation.inp"
-    with open(apdl_script_path, "w") as f:
-        f.write(apdl_script)
-
-    # Run ANSYS using the APDL script
-    ansys_path = "C:\\Program Files\\ANSYS Inc\\v201\\ansys\\bin\\winx64\\ansys201.exe"  # Replace with your ANSYS installation path
-    command = f'"{ansys_path}" -b -i {apdl_script_path} -o simulation_output.txt'
-
-    try:
-        subprocess.run(command, shell=True, check=True)
-        with open("simulation_output.txt", "r") as f:
-            simulation_results = f.read()
-        return simulation_results
-    except Exception as e:
-        raise Exception(f"Failed to run ANSYS simulation: {e}")
-
-def process_cad_and_generate_gcode_with_simulation(cad_file):
-    """
-    Processes the uploaded CAD file, converts it to STEP using SolidWorks,
-    runs ANSYS simulation, and generates G-code for CNC simulation.
-    """
-    # Check if CAD file is provided
-    if not cad_file:
-        return "No file provided. Please upload a valid CAD file.", None, None
-
-    # Save the uploaded CAD file to a temporary location
-    cad_file_path = cad_file.name
-    with open(cad_file_path, "wb") as f:
-        f.write(cad_file.read())
-
-    # Process CAD file with SolidWorks
-    step_file_path = process_cad_with_solidworks(cad_file_path)
-
-    # Run ANSYS simulation
-    simulation_results = run_ansys_simulation(step_file_path)
-
-    # Generate G-code
     gcode = []
     gcode.append("G21 ; Set units to millimeters")
     gcode.append("G90 ; Absolute positioning")
     gcode.append("G28 U0 W0 ; Return to home position")
-    gcode.append("M3 S1000 ; Start spindle at 1000 RPM")
-
-    # Example toolpath for simulation
-    gcode.append("G00 X0 Z3.125 ; Rapid move to start position")
-    gcode.append("G01 X0 Z0 F100 ; Linear cut to start the main contour")
-    gcode.append("G02 X0.5 Z0.5 R0.25 ; Arc (R=0.25)")
-    gcode.append("G01 X0.5 Z1.125 ; Linear move along Z")
-    gcode.append("G02 X1 Z1.5 R0.5 ; Arc (R=0.5)")
-    gcode.append("G01 X1 Z2 ; Linear move to next segment")
-    gcode.append("G03 X1.125 Z2.75 R0.125 ; Arc (R=0.125)")
-    gcode.append("G01 X1.25 Z3 ; Linear cut to next step")
-    gcode.append("G01 X2 Z3 ; Straight cut")
-    gcode.append("G02 X2.75 Z4.125 R0.5 ; Arc (R=0.5)")
-    gcode.append("G01 X3 Z5 ; Linear move to final step")
-    gcode.append("G00 X100 Z100 ; Rapid move to safe position")
+    gcode.append(f"M3 S1000 ; Start spindle at 1000 RPM")
+    gcode.append(f"G00 X0 Z{depth} ; Rapid move to start position")
+
+    # Simulate moves based on reference image
+    gcode.append(f"G01 X0 Z-{length} F{feed_rate} ; Linear move")
+    gcode.append(f"G02 X{radius1} Z-{length / 2} R{radius1} ; Arc with R={radius1}")
+    gcode.append(f"G01 X{radius2} Z-{(3 * length) / 4} ; Linear move to smaller radius")
+    gcode.append(f"G03 X{radius2 + 0.5} Z-{length} R{radius2} ; Arc with R={radius2}")
+    gcode.append("G00 Z5 ; Retract tool")
     gcode.append("M5 ; Stop spindle")
     gcode.append("M30 ; End of program")
-
-    # Save the G-code to a file
-    gcode_filename = "generated_gcode.txt"
-    with open(gcode_filename, "w") as f:
-        f.write("\n".join(gcode))
-
-    return "CAD file processed successfully.", simulation_results, gcode_filename
-
-# Create Gradio interface
-interface = gr.Interface(
-    fn=process_cad_and_generate_gcode_with_simulation,
-    inputs=gr.File(label="Upload CAD File (SolidWorks or STEP)"),
-    outputs=[
-        gr.Text(label="Processing Status"),
-        gr.Text(label="Simulation Results"),
-        gr.File(label="Download Generated G-Code")
+    return "\n".join(gcode)
+
+# Gradio Interface
+iface = gr.Interface(
+    fn=generate_gcode,
+    inputs=[
+        gr.Slider(2, 10, step=0.5, label="Length of Part (mm)", value=5),
+        gr.Slider(0.1, 1.0, step=0.05, label="First Arc Radius (mm)", value=0.25),
+        gr.Slider(0.1, 1.0, step=0.05, label="Second Arc Radius (mm)", value=0.125),
+        gr.Slider(50, 500, step=50, label="Feed Rate (mm/min)", value=100),
+        gr.Slider(0.5, 5, step=0.5, label="Cutting Depth (mm)", value=3.125),
     ],
-    title="CAD to G-Code with Simulation",
-    description="Upload a CAD file (SolidWorks or STEP) to process it using SolidWorks, "
-                "simulate using ANSYS, and generate G-code for CNC simulation."
+    outputs=gr.Textbox(label="Generated G-Code"),
+    title="CNC G-code Generator",
+    description="Generate G-code for CNC turning operations. Customize inputs for toolpath and cutting parameters."
 )
 
-# Launch the app
-if __name__ == "__main__":
-    interface.launch()
+# Launch the interface
+iface.launch(share=True)