Deploy Takeoff AI Annotator

app.py

@@ -118,7 +118,7 @@ def _load_image_for_detect(image_path, pdf_path, pdf_page):
 # ---------------------------------------------------------------------------
 # Canvas helpers — HTML5 canvas replaces Plotly for the Review & Edit tab
 # ---------------------------------------------------------------------------
-_CANVAS_MAX_PX = 1400  # max dimension sent to canvas (keeps base64 small)
+_CANVAS_MAX_PX = 2400  # max dimension sent to canvas (large enough to stay sharp when zoomed)
 
 
 def _build_canvas_json(img_rgb, annotations, selected=-1):
@@ -171,23 +171,33 @@ def _build_canvas_json(img_rgb, annotations, selected=-1):
     )
 
 
+# The detections grid is a secondary "click row to select" aid — the canvas is
+# the primary selector. Gradio's Dataframe (Svelte Table) crashes
+# ("RangeError: Too many properties to enumerate") on dense pages with 100+
+# rows, which wedges the whole UI. Cap the rows it renders; all boxes still
+# show on the canvas and are selectable there.
+_REVIEW_DF_MAX_ROWS = 50
+
+
 def _build_review_df(annotations):
     if not PANDAS_SUPPORT or not annotations:
         return []
-    return pd.DataFrame(
-        [
-            {
-                "#": i + 1,
-                "x1": round(ann["x1"]),
-                "y1": round(ann["y1"]),
-                "x2": round(ann["x2"]),
-                "y2": round(ann["y2"]),
-                "conf": f"{ann.get('conf', 1.0):.2f}",
-                "status": ann.get("status", "pending"),
-            }
-            for i, ann in enumerate(annotations)
-        ]
-    )
+    rows = [
+        {
+            "#": i + 1,
+            "x1": round(ann["x1"]),
+            "y1": round(ann["y1"]),
+            "x2": round(ann["x2"]),
+            "y2": round(ann["y2"]),
+            "conf": f"{ann.get('conf', 1.0):.2f}",
+            "status": ann.get("status", "pending"),
+        }
+        for i, ann in enumerate(annotations)
+    ]
+    if len(rows) > _REVIEW_DF_MAX_ROWS:
+        # Keep the grid small enough to never crash; select the rest on the canvas.
+        rows = rows[:_REVIEW_DF_MAX_ROWS]
+    return pd.DataFrame(rows)
 
 
 # ---------------------------------------------------------------------------
@@ -773,6 +783,68 @@ def on_rv_detect(rv_pdf, rv_page, rv_annotations, rv_conf):
     )
 
 
+def _box_iou(a, b):
+    ix1, iy1 = max(a["x1"], b["x1"]), max(a["y1"], b["y1"])
+    ix2, iy2 = min(a["x2"], b["x2"]), min(a["y2"], b["y2"])
+    inter = max(0.0, ix2 - ix1) * max(0.0, iy2 - iy1)
+    if inter <= 0:
+        return 0.0
+    area_a = (a["x2"] - a["x1"]) * (a["y2"] - a["y1"])
+    area_b = (b["x2"] - b["x1"]) * (b["y2"] - b["y1"])
+    return inter / (area_a + area_b - inter)
+
+
+def on_rv_find_more(rv_pdf, rv_page, rv_annotations, rv_conf):
+    """Low-conf re-detect that MERGES: keeps every existing box (approved,
+    rejected, human, pending) and only adds detections that don't overlap one.
+    Lets the model fill audit gaps without the user redrawing by hand."""
+    anns = rv_annotations.get(rv_page, [])
+    if rv_pdf is None:
+        return gr.update(), _build_review_df(anns), rv_annotations, "No PDF loaded."
+    detect_rgb = _render_pdf_page(rv_pdf, rv_page, dpi=DETECTION_DPI)
+    if detect_rgb is None:
+        return gr.update(), _build_review_df(anns), rv_annotations, "Failed to render page."
+    from detection import _get_model
+
+    using_yolo = _get_model() is not None
+    detect_bgr = cv2.cvtColor(detect_rgb, cv2.COLOR_RGB2BGR)
+    conf = min(float(rv_conf), 0.10)  # always cast a wide net regardless of the slider
+    boxes = detect_rectangles(detect_bgr, conf=conf)
+    scale = 100.0 / DETECTION_DPI
+    rv_annotations = dict(rv_annotations)
+    new_anns = list(anns)
+    added = 0
+    for b in boxes:
+        cand = {
+            "x1": b["x1"] * scale,
+            "y1": b["y1"] * scale,
+            "x2": b["x2"] * scale,
+            "y2": b["y2"] * scale,
+        }
+        # an existing box of ANY status wins — including rejected, so boxes the
+        # user already threw out don't come back
+        if any(_box_iou(cand, a) > 0.30 for a in anns):
+            continue
+        new_anns.append(
+            dict(
+                cand,
+                id=len(new_anns),
+                conf=b["conf"],
+                status="pending",
+                source="yolo" if using_yolo else "opencv",
+            )
+        )
+        added += 1
+    rv_annotations[rv_page] = new_anns
+    _autosave(rv_pdf, rv_annotations)
+    return (
+        _build_canvas_json(None, new_anns),
+        _build_review_df(new_anns),
+        rv_annotations,
+        f"Find More @conf {conf:.2f}: +{added} new box(es) (existing {len(anns)} kept).",
+    )
+
+
 def on_rv_detect_all(rv_pdf, rv_page, rv_total, rv_annotations, rv_conf):
     if rv_pdf is None:
         return _build_canvas_json(None, []), _build_review_df([]), rv_annotations, "No PDF loaded."
@@ -938,6 +1010,30 @@ def on_rv_approve_selected(rv_pdf, rv_page, rv_annotations, ledger, selected_idx
     )
 
 
+def on_rv_reject_selected(rv_pdf, rv_page, rv_annotations, ledger, selected_idx):
+    anns = rv_annotations.get(rv_page, [])
+    if rv_pdf is None or not anns or selected_idx is None or selected_idx < 0:
+        return gr.update(), _build_review_df(anns), rv_annotations, ledger, "Select a box first."
+    idx = int(selected_idx)
+    if idx >= len(anns):
+        return gr.update(), _build_review_df(anns), rv_annotations, ledger, "Invalid selection."
+    rv_annotations = dict(rv_annotations)
+    new_anns = list(anns)
+    old = new_anns[idx]
+    new_anns[idx] = dict(old, status="rejected")
+    rv_annotations[rv_page] = new_anns
+    event = _make_event(rv_page, old, "rejected", before=_make_box(old), after=_make_box(new_anns[idx]))
+    ledger = _append_events(ledger, [event])
+    _autosave(rv_pdf, rv_annotations)
+    return (
+        _build_canvas_json(None, new_anns, selected=idx),
+        _build_review_df(new_anns),
+        rv_annotations,
+        ledger,
+        f"Rejected box #{idx + 1}. {_ledger_summary(ledger)}",
+    )
+
+
 def on_rv_delete_selected(rv_pdf, rv_page, rv_annotations, ledger, selected_idx):
     anns = rv_annotations.get(rv_page, [])
     if rv_pdf is None or not anns or selected_idx is None or selected_idx < 0:
@@ -1015,9 +1111,12 @@ _model_version = get_model_version()
 
 _CANVAS_HTML = """
 <div id="rv-canvas-outer" style="width:100%;min-height:600px;background:#0f0f0f;border-radius:8px;overflow:hidden;position:relative;user-select:none;-webkit-user-select:none;">
+  <label style="position:absolute;top:8px;left:8px;z-index:10;background:rgba(0,0,0,0.6);color:#fff;font-family:monospace;font-size:12px;padding:4px 8px;border-radius:4px;cursor:pointer;">
+    <input type="checkbox" id="rv-hide-rejected" style="vertical-align:middle;margin-right:4px;">hide rejected <kbd style="opacity:0.6;">H</kbd>
+  </label>
   <canvas id="rv-ann-canvas" style="display:block;width:100%;cursor:crosshair;touch-action:none;"></canvas>
   <div id="rv-canvas-msg" style="position:absolute;top:50%;left:50%;transform:translate(-50%,-50%);color:#555;font-family:monospace;font-size:13px;pointer-events:none;text-align:center;line-height:1.8;">
-    Upload a PDF to begin<br><span style="color:#333;font-size:11px;">Drag on canvas to draw &bull; click box to select &bull; drag to move &bull; drag corner to resize</span>
+    Upload a PDF to begin<br><span style="color:#333;font-size:11px;">Scroll to zoom &bull; right-click + drag to pan &bull; double-click to reset view<br>Drag on canvas to draw &bull; click box to select &bull; drag to move &bull; drag corner to resize</span>
   </div>
 </div>
 <div id="rv-ann-tooltip" style="position:fixed;display:none;background:rgba(20,20,20,0.93);color:#fff;border-radius:6px;padding:8px 12px;font-family:monospace;font-size:12px;pointer-events:none;z-index:9999;box-shadow:0 2px 10px rgba(0,0,0,0.5);line-height:1.7;max-width:280px;"></div>
@@ -1037,6 +1136,9 @@ _CANVAS_JS = """
   var anns = [], selected = -1;
   var drag = null;
   var lastRender = '';
+  // View transform: zoom is multiplied on top of the fit-to-width base scale;
+  // pan offsets are in canvas pixels. zoom 1 + pan 0 == whole page fits the column.
+  var zoom = 1, panX = 0, panY = 0, spaceDown = false;
 
   // ── Boot ────────────────────────────────────────────────────────────────
   var _iv = setInterval(function() {
@@ -1048,6 +1150,12 @@ _CANVAS_JS = """
       canvas.addEventListener('mousemove', onMove);
       canvas.addEventListener('mouseup', onUp);
       canvas.addEventListener('mouseleave', function(e) { hideTooltip(); if (drag) { onUp(e); } });
+      canvas.addEventListener('wheel', onWheel, { passive: false });
+      canvas.addEventListener('dblclick', function() { resetView(); draw(); });
+      canvas.addEventListener('auxclick', function(e) { e.preventDefault(); });
+      canvas.addEventListener('contextmenu', function(e) { e.preventDefault(); });
+      var hr = document.getElementById('rv-hide-rejected');
+      if (hr) hr.addEventListener('change', draw);
       new ResizeObserver(refit).observe(canvas.parentElement);
       setInterval(checkRender, 150);
     }
@@ -1068,6 +1176,7 @@ _CANVAS_JS = """
     var msg = document.getElementById('rv-canvas-msg');
     if (d.image && d.image !== cachedImgSrc) {
       imgW = d.imageW; imgH = d.imageH;
+      resetView();
       var im = new Image();
       im.onload = function() { cachedImg = im; cachedImgSrc = d.image; refit(); };
       im.onerror = function() { console.warn('Canvas: failed to load image'); };
@@ -1084,20 +1193,56 @@ _CANVAS_JS = """
     var w = canvas.parentElement.clientWidth || 800;
     canvas.width  = Math.round(w);
     canvas.height = Math.round(imgH * (w / imgW));
+    clampPan();
     draw();
   }
 
-  function s() { return canvas && imgW ? canvas.width / imgW : 1; }
+  function s() { return canvas && imgW ? (canvas.width / imgW) * zoom : 1; }
+
+  // Rejected boxes stay in the data (training hard-negatives) but can be
+  // hidden from the canvas; hidden boxes are also skipped by hit-testing.
+  function rejectedHidden() {
+    var cb = document.getElementById('rv-hide-rejected');
+    return !!(cb && cb.checked);
+  }
+
+  function isHidden(ann) { return rejectedHidden() && ann.status === 'rejected'; }
+
+  // ── Zoom / pan ───────────────────────────────────────────────────────────
+  function resetView() { zoom = 1; panX = 0; panY = 0; }
+
+  function clampPan() {
+    if (!canvas || !imgW) return;
+    var sc = s();
+    panX = Math.min(0, Math.max(canvas.width  - imgW * sc, panX));
+    panY = Math.min(0, Math.max(canvas.height - imgH * sc, panY));
+  }
+
+  function onWheel(e) {
+    if (!imgW) return;
+    e.preventDefault();
+    var r = canvas.getBoundingClientRect();
+    var mx = (e.clientX - r.left) * (canvas.width  / r.width);
+    var my = (e.clientY - r.top)  * (canvas.height / r.height);
+    var nz = Math.min(16, Math.max(1, zoom * Math.pow(1.0015, -e.deltaY)));
+    panX = mx - (mx - panX) * (nz / zoom);
+    panY = my - (my - panY) * (nz / zoom);
+    zoom = nz;
+    if (zoom <= 1.001) resetView();
+    clampPan();
+    draw();
+  }
 
   // ── Draw ─────────────────────────────────────────────────────────────────
   function draw() {
     if (!canvas) return;
     var sc = s();
     ctx.clearRect(0, 0, canvas.width, canvas.height);
-    if (cachedImg) ctx.drawImage(cachedImg, 0, 0, canvas.width, canvas.height);
+    if (cachedImg) ctx.drawImage(cachedImg, panX, panY, imgW * sc, imgH * sc);
 
     anns.forEach(function(ann, i) {
-      var x1 = ann.x1*sc, y1 = ann.y1*sc, x2 = ann.x2*sc, y2 = ann.y2*sc;
+      if (isHidden(ann)) return;
+      var x1 = ann.x1*sc+panX, y1 = ann.y1*sc+panY, x2 = ann.x2*sc+panX, y2 = ann.y2*sc+panY;
       var c = COLORS[ann.status] || COLORS.pending;
       var sel = (i === selected);
 
@@ -1121,14 +1266,26 @@ _CANVAS_JS = """
       }
     });
 
-    // Draw-in-progress preview
+    // Draw-in-progress preview (drag coords are in image space)
     if (drag && drag.mode === 'draw') {
+      var ox = drag.ox*sc+panX, oy = drag.oy*sc+panY, dcx = drag.cx*sc+panX, dcy = drag.cy*sc+panY;
       ctx.strokeStyle = '#2563EB'; ctx.lineWidth = 2;
       ctx.setLineDash([5, 4]);
-      ctx.strokeRect(drag.ox, drag.oy, drag.cx - drag.ox, drag.cy - drag.oy);
+      ctx.strokeRect(ox, oy, dcx - ox, dcy - oy);
       ctx.setLineDash([]);
       ctx.fillStyle = 'rgba(37,99,235,0.1)';
-      ctx.fillRect(drag.ox, drag.oy, drag.cx - drag.ox, drag.cy - drag.oy);
+      ctx.fillRect(ox, oy, dcx - ox, dcy - oy);
+    }
+
+    // Zoom badge
+    if (zoom > 1.001) {
+      var badge = Math.round(zoom * 100) + '%  \\u00B7 right-drag = pan \\u00B7 dbl-click = reset';
+      ctx.font = '12px monospace';
+      var bw = ctx.measureText(badge).width + 16;
+      ctx.fillStyle = 'rgba(0,0,0,0.6)';
+      ctx.fillRect(canvas.width - bw - 8, 8, bw, 22);
+      ctx.fillStyle = '#fff';
+      ctx.fillText(badge, canvas.width - bw, 23);
     }
   }
 
@@ -1155,38 +1312,58 @@ _CANVAS_JS = """
     hoveredIdx = -1;
   }
 
-  // ── Hit testing ──────────────────────────────────────────────────────────
-  function hitHandle(ann, cx, cy) {
-    var sc = s(), h = HANDLE;
+  // ── Hit testing (image-space coords) ─────────────────────────────────────
+  function hitHandle(ann, ix, iy) {
+    var h = HANDLE / s();
     var pts = [
-      [ann.x1*sc, ann.y1*sc, 'nw'], [ann.x2*sc, ann.y1*sc, 'ne'],
-      [ann.x2*sc, ann.y2*sc, 'se'], [ann.x1*sc, ann.y2*sc, 'sw'],
+      [ann.x1, ann.y1, 'nw'], [ann.x2, ann.y1, 'ne'],
+      [ann.x2, ann.y2, 'se'], [ann.x1, ann.y2, 'sw'],
     ];
     for (var i = 0; i < pts.length; i++) {
-      if (Math.abs(cx - pts[i][0]) <= h && Math.abs(cy - pts[i][1]) <= h)
+      if (Math.abs(ix - pts[i][0]) <= h && Math.abs(iy - pts[i][1]) <= h)
         return pts[i][2];
     }
     return null;
   }
 
-  function hitBox(ann, cx, cy) {
-    var sc = s();
-    return cx >= ann.x1*sc && cx <= ann.x2*sc && cy >= ann.y1*sc && cy <= ann.y2*sc;
+  function hitBox(ann, ix, iy) {
+    return ix >= ann.x1 && ix <= ann.x2 && iy >= ann.y1 && iy <= ann.y2;
   }
 
+  // Mouse position in IMAGE pixels (pan/zoom already removed)
   function pos(e) {
     var r = canvas.getBoundingClientRect();
-    var sx = canvas.width  / r.width;
-    var sy = canvas.height / r.height;
-    return { x: (e.clientX - r.left) * sx, y: (e.clientY - r.top) * sy };
+    var cx = (e.clientX - r.left) * (canvas.width  / r.width);
+    var cy = (e.clientY - r.top)  * (canvas.height / r.height);
+    var sc = s();
+    return { x: (cx - panX) / sc, y: (cy - panY) / sc };
   }
 
-  function fi(v) { return v / s(); }
-
   // ── Mouse ────────────────────────────────────────────────────────────────
   function onDown(e) {
     hideTooltip();
+    // Pan: right or middle mouse drag, or hold Space + left-drag
+    if (e.button === 2 || e.button === 1 || spaceDown) {
+      e.preventDefault();
+      drag = { mode:'pan', sx:e.clientX, sy:e.clientY, px:panX, py:panY };
+      canvas.style.cursor = 'grabbing';
+      return;
+    }
+    if (e.button !== 0) return;
     var p = pos(e);
+    // Shift+click: stamp a copy of the selected box centered at the cursor
+    // (the original stays selected so you can keep stamping down a row)
+    if (e.shiftKey && selected >= 0 && selected < anns.length) {
+      var tpl = anns[selected];
+      var hw = (tpl.x2 - tpl.x1) / 2, hh = (tpl.y2 - tpl.y1) / 2;
+      anns.push(Object.assign({}, tpl, {
+        x1: p.x - hw, y1: p.y - hh, x2: p.x + hw, y2: p.y + hh,
+        status: 'pending', source: 'human_added'
+      }));
+      draw();
+      sync('created');
+      return;
+    }
     // Resize handles on selected box take priority
     if (selected >= 0 && selected < anns.length) {
       var dir = hitHandle(anns[selected], p.x, p.y);
@@ -1198,6 +1375,7 @@ _CANVAS_JS = """
     }
     // Click inside any box to select + move
     for (var i = anns.length - 1; i >= 0; i--) {
+      if (isHidden(anns[i])) continue;
       if (hitBox(anns[i], p.x, p.y)) {
         selected = i;
         drag = { mode:'move', idx:i,
@@ -1214,11 +1392,20 @@ _CANVAS_JS = """
   }
 
   function onMove(e) {
+    if (drag && drag.mode === 'pan') {
+      var r = canvas.getBoundingClientRect();
+      panX = drag.px + (e.clientX - drag.sx) * (canvas.width  / r.width);
+      panY = drag.py + (e.clientY - drag.sy) * (canvas.height / r.height);
+      clampPan();
+      draw();
+      return;
+    }
     var p = pos(e);
     // Tooltip on hover (only when not dragging)
     if (!drag) {
       var hovered = -1;
       for (var j = anns.length - 1; j >= 0; j--) {
+        if (isHidden(anns[j])) continue;
         if (hitBox(anns[j], p.x, p.y)) { hovered = j; break; }
       }
       if (hovered >= 0) {
@@ -1228,7 +1415,7 @@ _CANVAS_JS = """
       }
       return;
     }
-    var dx = fi(p.x - drag.ox), dy = fi(p.y - drag.oy);
+    var dx = p.x - drag.ox, dy = p.y - drag.oy;
     if (drag.mode === 'move') {
       var o = drag.orig;
       anns[drag.idx] = Object.assign({}, o,
@@ -1248,13 +1435,18 @@ _CANVAS_JS = """
 
   function onUp(e) {
     if (!drag) return;
+    if (drag.mode === 'pan') {
+      drag = null;
+      canvas.style.cursor = spaceDown ? 'grab' : 'crosshair';
+      return;
+    }
     var action = drag.mode;
     if (drag.mode === 'draw') {
       var p = pos(e);
       var x1c = Math.min(drag.ox, p.x), y1c = Math.min(drag.oy, p.y);
       var x2c = Math.max(drag.ox, p.x), y2c = Math.max(drag.oy, p.y);
-      if (x2c - x1c > 5 && y2c - y1c > 5) {
-        anns.push({ x1:fi(x1c), y1:fi(y1c), x2:fi(x2c), y2:fi(y2c),
+      if ((x2c - x1c) * s() > 5 && (y2c - y1c) * s() > 5) {
+        anns.push({ x1:x1c, y1:y1c, x2:x2c, y2:y2c,
                     conf:1.0, status:'pending', source:'human_added' });
         selected = anns.length - 1;
         action = 'created';
@@ -1314,8 +1506,25 @@ _CANVAS_JS = """
       case 'c': case 'C':      copySelected(); break;
       case 'Delete': case 'Backspace': deleteSelected(); break;
       case 'a': case 'A':     triggerBtn('rv-approve-sel'); break;
+      case 'r': case 'R':     triggerBtn('rv-reject-sel'); break;
+      case 'h': case 'H': {
+        var cb = document.getElementById('rv-hide-rejected');
+        if (cb) { cb.checked = !cb.checked; draw(); }
+        break;
+      }
       case 'ArrowLeft':        e.preventDefault(); triggerBtn('rv-prev'); break;
       case 'ArrowRight':       e.preventDefault(); triggerBtn('rv-next'); break;
+      case ' ':                e.preventDefault();
+                               if (!spaceDown && canvas) { spaceDown = true; if (!drag) canvas.style.cursor = 'grab'; }
+                               break;
+      case '0':                resetView(); draw(); break;
+    }
+  });
+
+  document.addEventListener('keyup', function(e) {
+    if (e.key === ' ') {
+      spaceDown = false;
+      if (canvas && !drag) canvas.style.cursor = 'crosshair';
     }
   });
 })();
@@ -1398,6 +1607,7 @@ with gr.Blocks(title="Takeoff AI Annotator", head=_CANVAS_JS) as demo:
                 rv_page_jump = gr.Textbox(value="", show_label=False, scale=1, min_width=80, placeholder="Go to page…")
                 rv_conf_slider = gr.Slider(0.05, 0.95, value=0.25, step=0.05, label="Conf", scale=3)
                 rv_detect_btn = gr.Button("Detect Page", variant="primary", scale=2)
+                rv_find_more_btn = gr.Button("🔎 Find More", scale=2)
                 rv_detect_all_btn = gr.Button("Detect All", scale=2)
 
             with gr.Row():
@@ -1415,6 +1625,8 @@ with gr.Blocks(title="Takeoff AI Annotator", head=_CANVAS_JS) as demo:
                         rv_approve_sel_btn = gr.Button(
                             "✓ Approve", variant="secondary", elem_id="rv-approve-sel", scale=2
                         )
+                        rv_reject_sel_btn = gr.Button("✗ Reject", variant="stop", elem_id="rv-reject-sel", scale=2)
+                    with gr.Row():
                         rv_copy_sel_btn = gr.Button("⧉ Copy", scale=1)
                         rv_delete_sel_btn = gr.Button("🗑 Delete", variant="stop", elem_id="rv-delete-sel", scale=2)
 
@@ -1452,8 +1664,13 @@ with gr.Blocks(title="Takeoff AI Annotator", head=_CANVAS_JS) as demo:
                     )
 
             gr.Markdown(
-                "`A` approve &nbsp; `C` copy &nbsp; `Del` delete &nbsp; "
-                "`← →` pages &nbsp; — draw box by dragging on empty canvas"
+                "`A` approve &nbsp; `R` reject &nbsp; `H` hide/show rejected &nbsp; `C` copy &nbsp; `Del` delete &nbsp; "
+                "`← →` pages &nbsp; — draw box by dragging on empty canvas  \n"
+                "⚡ **Shift+click = stamp a copy** of the selected box at the cursor (select one good box, "
+                "then Shift+click each identical frame) &nbsp; 🔎 **Find More** = re-detect at low confidence, "
+                "adds only boxes you don't already have  \n"
+                "🔍 **Scroll wheel = zoom** &nbsp; **right-click + drag = pan** &nbsp; "
+                "double-click or `0` = reset view"
                 + (
                     f"  \n💾 **Autosave on** — every change is checkpointed to `{_AUTOSAVE_DIR}`; "
                     "a refresh/crash loses nothing (re-import from there to resume)."
@@ -1510,6 +1727,11 @@ with gr.Blocks(title="Takeoff AI Annotator", head=_CANVAS_JS) as demo:
                 inputs=[rv_pdf_state, rv_page_state, rv_annotations_state, rv_conf_slider],
                 outputs=[rv_canvas_render_txt, rv_df, rv_annotations_state, rv_status],
             )
+            rv_find_more_btn.click(
+                fn=on_rv_find_more,
+                inputs=[rv_pdf_state, rv_page_state, rv_annotations_state, rv_conf_slider],
+                outputs=[rv_canvas_render_txt, rv_df, rv_annotations_state, rv_status],
+            )
             rv_detect_all_btn.click(
                 fn=on_rv_detect_all,
                 inputs=[rv_pdf_state, rv_page_state, rv_total_state, rv_annotations_state, rv_conf_slider],
@@ -1543,6 +1765,11 @@ with gr.Blocks(title="Takeoff AI Annotator", head=_CANVAS_JS) as demo:
                 inputs=[rv_pdf_state, rv_page_state, rv_annotations_state, rv_ledger_state, rv_selected_state],
                 outputs=[rv_canvas_render_txt, rv_df, rv_annotations_state, rv_ledger_state, rv_status],
             )
+            rv_reject_sel_btn.click(
+                fn=on_rv_reject_selected,
+                inputs=[rv_pdf_state, rv_page_state, rv_annotations_state, rv_ledger_state, rv_selected_state],
+                outputs=[rv_canvas_render_txt, rv_df, rv_annotations_state, rv_ledger_state, rv_status],
+            )
             rv_copy_sel_btn.click(
                 fn=on_rv_copy_selected,
                 inputs=[rv_pdf_state, rv_page_state, rv_annotations_state, rv_ledger_state, rv_selected_state],