Add center/scale post-processing, increase diffusion steps to 50

app.py

@@ -419,6 +419,86 @@ def clean_gcode(gcode: str) -> str:
     return result
 
 
+def center_and_scale_gcode(gcode: str) -> str:
+    """Center the drawing on the workplane and scale to fill 80% of it."""
+    lines = gcode.split("\n")
+    
+    # Extract all coordinates
+    coords = []
+    for line in lines:
+        x_match = re.search(r"X([-\d.]+)", line, re.IGNORECASE)
+        y_match = re.search(r"Y([-\d.]+)", line, re.IGNORECASE)
+        if x_match and y_match:
+            try:
+                x = float(x_match.group(1))
+                y = float(y_match.group(1))
+                coords.append((x, y))
+            except ValueError:
+                pass
+    
+    if len(coords) < 2:
+        return gcode  # Nothing to transform
+    
+    # Find bounding box
+    xs = [c[0] for c in coords]
+    ys = [c[1] for c in coords]
+    min_x, max_x = min(xs), max(xs)
+    min_y, max_y = min(ys), max(ys)
+    
+    # Current dimensions
+    width = max_x - min_x
+    height = max_y - min_y
+    
+    if width < 1 or height < 1:
+        return gcode  # Degenerate case
+    
+    # Target: 80% of workplane, centered
+    target_width = (BOUNDS["right"] - BOUNDS["left"]) * 0.8
+    target_height = (BOUNDS["top"] - BOUNDS["bottom"]) * 0.8
+    
+    # Scale to fit (maintain aspect ratio)
+    scale = min(target_width / width, target_height / height)
+    
+    # Center of current drawing
+    cx = (min_x + max_x) / 2
+    cy = (min_y + max_y) / 2
+    
+    # Center of workplane
+    target_cx = (BOUNDS["left"] + BOUNDS["right"]) / 2
+    target_cy = (BOUNDS["bottom"] + BOUNDS["top"]) / 2
+    
+    print(f"Centering: bbox=({min_x:.0f},{min_y:.0f})-({max_x:.0f},{max_y:.0f}), scale={scale:.2f}")
+    
+    # Transform each line
+    result = []
+    for line in lines:
+        new_line = line
+        x_match = re.search(r"X([-\d.]+)", line, re.IGNORECASE)
+        y_match = re.search(r"Y([-\d.]+)", line, re.IGNORECASE)
+        
+        if x_match:
+            try:
+                x = float(x_match.group(1))
+                new_x = (x - cx) * scale + target_cx
+                new_x = max(BOUNDS["left"], min(BOUNDS["right"], new_x))
+                new_line = re.sub(r"X[-\d.]+", f"X{new_x:.2f}", new_line, count=1, flags=re.IGNORECASE)
+            except ValueError:
+                pass
+        
+        if y_match:
+            try:
+                y = float(y_match.group(1))
+                new_y = (y - cy) * scale + target_cy
+                new_y = max(BOUNDS["bottom"], min(BOUNDS["top"], new_y))
+                new_line = re.sub(r"Y[-\d.]+", f"Y{new_y:.2f}", new_line, count=1, flags=re.IGNORECASE)
+            except ValueError:
+                pass
+        
+        result.append(new_line)
+    
+    return "\n".join(result)
+
+
 def validate_gcode(gcode: str) -> str:
     """Clamp coordinates to machine bounds."""
     lines = []
@@ -710,6 +790,8 @@ def generate(prompt: str, temperature: float, max_tokens: int, num_steps: int, g
             # Clean up the gcode
             gcode = clean_gcode(gcode)
         
+        # Center and scale to fill workplane
+        gcode = center_and_scale_gcode(gcode)
         gcode = validate_gcode(gcode)
         line_count = len([l for l in gcode.split("\n") if l.strip()])
         svg = gcode_to_svg(gcode)
@@ -830,10 +912,10 @@ with gr.Blocks(css=css, theme=gr.themes.Base()) as demo:
             )
             
             with gr.Accordion("settings", open=False):
-                temperature = gr.Slider(0.3, 1.2, value=0.5, label="temperature", step=0.1)
+                temperature = gr.Slider(0.3, 1.2, value=0.7, label="temperature", step=0.1)
                 max_tokens = gr.Slider(256, 2048, value=2048, step=256, label="max tokens")
-                num_steps = gr.Slider(20, 50, value=35, step=5, label="diffusion steps")
-                guidance = gr.Slider(5.0, 20.0, value=10.0, step=0.5, label="guidance")
+                num_steps = gr.Slider(20, 75, value=50, step=5, label="diffusion steps")
+                guidance = gr.Slider(5.0, 20.0, value=12.0, step=0.5, label="guidance")
                 seed = gr.Number(value=-1, label="seed (-1 = random)", precision=0)
             
             generate_btn = gr.Button("generate", variant="secondary")