API Pointé IA

import os
import requests

API = "https://gprviewer.com/api"
HEADERS = {"Authorization": f"Bearer {os.environ['GPV_API_KEY']}"}


def check(r):
    """Return the JSON body, or raise with the API error code."""
    body = r.json()
    if r.status_code not in (200, 202):
        err = body.get("error", {})
        raise RuntimeError(f"HTTP {r.status_code} {err.get('code')}: {err.get('message')}")
    return body


def pick_rebars(path, epsr=None):
    params = {"filename": os.path.basename(path), "wait": 30}
    if epsr is not None:
        params["epsr"] = epsr

    # 1. Upload the raw file
    with open(path, "rb") as f:
        job = check(requests.post(
            f"{API}/v1/rebars", params=params, data=f,
            headers={**HEADERS, "Content-Type": "application/octet-stream"},
            timeout=120,
        ))

    # 2. 202 = still processing: poll the job
    while job["status"] not in ("done", "failed"):
        job = check(requests.get(
            f"{API}/v1/jobs/{job['id']}", params={"wait": 20},
            headers=HEADERS, timeout=60,
        ))

    if job["status"] == "failed":
        raise RuntimeError(f"Job {job['id']} failed: {job.get('error')}")
    return job["result"]


if __name__ == "__main__":
    result = pick_rebars("SCAN_001.DZT")
    for b in result["rebars"]:
        print(f"{b['distance_m']:7.2f} m  {b['depth_cm']:6.1f} cm  confidence {b['confidence']:.2f}")
    print(f"{result['count']} rebar(s) on {result['file']['length_m']} m")

// pick-rebars.mjs — Node.js 18+
import { readFile } from "node:fs/promises";
import { basename } from "node:path";

const API = "https://gprviewer.com/api";
const AUTH = { Authorization: `Bearer ${process.env.GPV_API_KEY}` };

// Return the JSON body, or throw with the API error code
async function call(url, init = {}) {
  const res = await fetch(url, { ...init, headers: { ...AUTH, ...init.headers } });
  const body = await res.json();
  if (!res.ok) {
    const e = body.error ?? {};
    throw new Error(`HTTP ${res.status} ${e.code}: ${e.message}`);
  }
  return body;
}

async function pickRebars(path, { epsr } = {}) {
  const params = new URLSearchParams({ filename: basename(path), wait: "30" });
  if (epsr !== undefined) params.set("epsr", String(epsr));

  // 1. Upload the raw file
  let job = await call(`${API}/v1/rebars?${params}`, {
    method: "POST",
    headers: { "Content-Type": "application/octet-stream" },
    body: await readFile(path),
  });

  // 2. 202 = still processing: poll the job
  while (job.status !== "done" && job.status !== "failed") {
    job = await call(`${API}/v1/jobs/${job.id}?wait=20`);
  }
  if (job.status === "failed") throw new Error(`Job ${job.id} failed`);
  return job.result;
}

const result = await pickRebars(process.argv[2] ?? "SCAN_001.DZT");
for (const b of result.rebars) {
  console.log(`${b.distance_m.toFixed(2)} m  ${b.depth_cm.toFixed(1)} cm  confidence ${b.confidence.toFixed(2)}`);
}
console.log(`${result.count} rebar(s) on ${result.file.length_m} m`);