Software Engineer & 3D Engineering Maker
desktop tools and 3D engineering

param(
    [string]$Path = ".",
    [string]$VlmEndpoint = "http://localhost:8080/vision"
)

# SQLite dedup database
$dbPath = "$PSScriptRoot/file_index.db"
$conn = New-Object System.Data.SQLite.SQLiteConnection
$conn.ConnectionString = "Data Source=$dbPath;Version=3;"
$conn.Open()

# Ensure schema
$cmd = $conn.CreateCommand()
$cmd.CommandText = @"
CREATE TABLE IF NOT EXISTS file_index (
    hash TEXT PRIMARY KEY,
    path TEXT,
    tag TEXT,
    category TEXT,
    processed_at TEXT DEFAULT (datetime('now'))
)
"@
$cmd.ExecuteNonQuery()

# Batch process files
$files = Get-ChildItem $Path -Recurse -File |
    Where-Object { $_.Extension -match '\.(jpg|png|mp4|avi)$' }

foreach ($file in $files) {
    $hash = (Get-FileHash $file.FullName).Hash

    # Skip duplicates
    $check = $conn.CreateCommand()
    $check.CommandText = "SELECT 1 FROM file_index WHERE hash = @h"
    $check.Parameters.AddWithValue("@h", $hash) | Out-Null
    $exists = $check.ExecuteScalar()
    if ($exists) { continue }

    # Send to local VLM for classification
    $bytes = [Convert]::ToBase64String(
        [IO.File]::ReadAllBytes($file.FullName)
    )
    $body = @{ image = $bytes } | ConvertTo-Json
    $result = Invoke-RestMethod -Uri $VlmEndpoint -Method Post -Body $body -ContentType "application/json"

    # Auto-organize into semantic folders
    $targetDir = Join-Path $Path $result.category
    if (-not (Test-Path $targetDir)) {
        New-Item -ItemType Directory -Path $targetDir
    }
    Move-Item $file.FullName (Join-Path $targetDir $file.Name)

    # Record in SQLite
    $ins = $conn.CreateCommand()
    $ins.CommandText = @"
INSERT INTO file_index (hash, path, tag, category)
VALUES (@h, @p, @t, @c)
"@
    $ins.Parameters.AddWithValue("@h", $hash) | Out-Null
    $ins.Parameters.AddWithValue("@p", $targetDir) | Out-Null
    $ins.Parameters.AddWithValue("@t", $result.tag) | Out-Null
    $ins.Parameters.AddWithValue("@c", $result.category) | Out-Null
    $ins.ExecuteNonQuery()
}
$conn.Close()

import asyncio
import json
import websockets

MOONRAKER_WS = "ws://127.0.0.1:7125/websocket"

async def send_gcode(ws, gcode: str):
    """Send a G-code command via Moonraker WebSocket."""
    msg = {
        "jsonrpc": "2.0",
        "method": "printer.gcode.script",
        "params": {"script": gcode},
        "id": None
    }
    await ws.send(json.dumps(msg))
    response = await ws.recv()
    return json.loads(response)

async def auto_calibrate():
    """Run full bed mesh + thermal calibration cycle."""
    async with websockets.connect(MOONRAKER_WS) as ws:
        # Home all axes
        await send_gcode(ws, "G28")

        # Heat nozzle for thermal expansion compensation
        await send_gcode(ws, "M109 S210")
        print("[+] Nozzle at 210C — measuring thermal drift...")

        # Run bed mesh calibration
        await send_gcode(ws, "BED_MESH_CALIBRATE")
        print("[+] Bed mesh calibrated")

        # Measure nozzle offset at temp
        await send_gcode(ws, "PROBE_CALIBRATE")
        print("[+] Probe offset captured at operating temp")

        # Save config profile
        await send_gcode(ws, "SAVE_CONFIG")
        print("[+] Configuration saved to printer.cfg")

        # Park toolhead
        await send_gcode(ws, "G1 X0 Y0 Z20 F3000")

asyncio.run(auto_calibrate())