Update app.py

app.py

@@ -1,73 +1,52 @@
 import gradio as gr
-import numpy as np
-import matplotlib.pyplot as plt
-from matplotlib.patches import Rectangle, Ellipse
+import os
 
-# Function for basic G-code generation (simple circles and rectangles)
-def generate_gcode(shape_type, size):
+# Function to generate G-code based on user input
+def generate_gcode(length, diameter, feed_rate):
+    """
+    Generates a simple G-code string based on user inputs.
+    """
     gcode = []
-
-    if shape_type == "Circle":
-        # Generate a simple circle G-code (assuming radius as size)
-        gcode.append(f"G17 G21 G90")  # Select XY plane, units in mm, absolute positioning
-        gcode.append(f"G0 Z5.0")  # Move tool to safe height
-        gcode.append(f"G0 X0 Y0")  # Move to start position
-        gcode.append(f"G2 X0 Y0 I{size} J0")  # Circle in clockwise direction
-        gcode.append(f"G0 Z5.0")  # Return to safe height
-
-    elif shape_type == "Rectangle":
-        # Generate a simple rectangle G-code (using size for length and width)
-        width, height = size, size / 2
-        gcode.append(f"G17 G21 G90")  # Select XY plane, units in mm, absolute positioning
-        gcode.append(f"G0 Z5.0")  # Move tool to safe height
-        gcode.append(f"G0 X0 Y0")  # Move to start position
-        gcode.append(f"G1 X{width} Y0")  # Move to the right side of the rectangle
-        gcode.append(f"G1 X{width} Y{height}")  # Move upwards
-        gcode.append(f"G1 X0 Y{height}")  # Move to the left side
-        gcode.append(f"G1 X0 Y0")  # Close the rectangle
-        gcode.append(f"G0 Z5.0")  # Return to safe height
-
+    gcode.append(f"(Generated G-code for length={length}, diameter={diameter}, feed_rate={feed_rate})")
+    gcode.append("G21   ; Set units to millimeters")
+    gcode.append("G90   ; Absolute positioning mode")
+    gcode.append("G17   ; Select XY plane")
+    gcode.append("M03 S1000   ; Start spindle at 1000 RPM")
+    gcode.append("G00 Z5.0   ; Move to safe height")
+    gcode.append(f"G00 X0 Y0  ; Move to start position")
+    gcode.append(f"G01 Z-1.0 F{feed_rate}   ; Start cutting")
+    gcode.append(f"G01 X{length} F{feed_rate} ; Linear move along X-axis")
+    gcode.append(f"G01 Y{diameter} F{feed_rate} ; Linear move along Y-axis")
+    gcode.append("G00 Z5.0   ; Retract to safe height")
+    gcode.append("M05   ; Stop spindle")
+    gcode.append("M30   ; End of program")
     return "\n".join(gcode)
 
-# Function for 2D Visualization
-def visualize_2d(shape_type, size):
-    fig, ax = plt.subplots()
-
-    if shape_type == "Circle":
-        # Visualize a circle
-        circle = Ellipse((0, 0), width=2*size, height=2*size, edgecolor='blue', facecolor='none')
-        ax.add_patch(circle)
-
-    elif shape_type == "Rectangle":
-        # Visualize a rectangle
-        width, height = size, size / 2
-        rect = Rectangle((-width/2, -height/2), width, height, linewidth=2, edgecolor='green', facecolor='none')
-        ax.add_patch(rect)
-
-    ax.set_xlim(-size*1.5, size*1.5)
-    ax.set_ylim(-size*1.5, size*1.5)
-    ax.set_aspect('equal', 'box')
-    plt.grid(True)
-    plt.title(f"2D Visualization: {shape_type}")
-    return fig
-
-# Gradio Interface
-def interface(shape_type, size):
-    # Generate G-code and visualization
-    gcode = generate_gcode(shape_type, size)
-    fig = visualize_2d(shape_type, size)
-    return gcode, fig
-
-# Define Gradio Interface
-gr.Interface(
-    fn=interface, 
-    inputs=[
-        gr.Dropdown(choices=["Circle", "Rectangle"], label="Shape Type"),
-        gr.Slider(minimum=1, maximum=100, step=0.1, label="Size")
-    ], 
-    outputs=[
-        gr.Textbox(label="G-code Output", lines=20),
-        gr.Plot(label="2D Visualization")
-    ], 
-    live=True
-).launch()
+# Gradio UI setup
+def main_interface(length, diameter, feed_rate):
+    gcode = generate_gcode(length, diameter, feed_rate)
+    return gcode
+
+# Create Gradio Interface
+inputs = [
+    gr.Number(label="Length (mm)", value=50),
+    gr.Number(label="Diameter (mm)", value=20),
+    gr.Number(label="Feed Rate (mm/min)", value=100)
+]
+
+outputs = gr.Textbox(label="Generated G-code", lines=20)
+
+title = "CNC G-code Generator"
+description = "Enter the desired length, diameter, and feed rate to generate a basic G-code file."
+
+demo = gr.Interface(
+    fn=main_interface,
+    inputs=inputs,
+    outputs=outputs,
+    title=title,
+    description=description
+)
+
+# Launch the app
+if __name__ == "__main__":
+    demo.launch()