feat: confidence override checkbox — reviewer can mark perfect pages as 100%

smoke_signal_tab.py

@@ -2584,7 +2584,7 @@ def get_review_item(idx: int) -> tuple:
     return img_path, item.get("raw_ocr",""), info, len(q_df) - len(pending), len(q_df)
 
 
-def save_review_decision(idx: int, final_text: str, action: str, reviewer: str, reason: str) -> tuple:
+def save_review_decision(idx: int, final_text: str, action: str, reviewer: str, reason: str, conf_override: bool = False) -> tuple:
     q_df = load_queue_df()
     d_df = load_decisions_df()
     if q_df.empty:
@@ -2642,12 +2642,16 @@ def save_review_decision(idx: int, final_text: str, action: str, reviewer: str,
     except Exception:
         punct_flags_count = 0
 
+    final_confidence = 1.0 if conf_override else float(item.get("confidence", 0))
+    if conf_override:
+        decision["confidence_override"] = True
+
     with open(GOLD_FILE, "a", encoding="utf-8") as f:
         f.write(json.dumps({
             **decision,
             "region_class":  region_class,
-            "confidence":    float(item.get("confidence", 0)),
-            "conf_class":    item.get("confidence_class",""),
+            "confidence":    final_confidence,
+            "conf_class":    "verified-100" if conf_override else item.get("confidence_class",""),
             "punct_score":   punct_score,
             "punct_flags_count": punct_flags_count,
         }) + "\n")
@@ -3086,6 +3090,10 @@ def smoke_signal_tab():
                             value="",
                         )
 
+                        conf_override = gr.Checkbox(
+                            label="Override confidence to 100% (page is perfect)",
+                            value=False,
+                        )
                         with gr.Row():
                             accept_btn = gr.Button("✓ Accept", elem_classes=["ss-btn-accept"])
                             edit_btn   = gr.Button("✎ Save Edit", elem_classes=["ss-btn-edit"])
@@ -3109,9 +3117,11 @@ def smoke_signal_tab():
                     img, raw, info, done, total = get_review_item(new_idx)
                     return new_idx, img, raw, raw, info
 
-                def do_accept(idx, final, reviewer, reason):
+                def do_accept(idx, final, reviewer, reason, conf_ov=False):
                     action = "edited" if final.strip() != "" else "accepted"
-                    fb, img, raw, info, _, _ = save_review_decision(idx, final, action, reviewer, reason)
+                    fb, img, raw, info, _, _ = save_review_decision(
+                        idx, final, action, reviewer, reason, conf_override=bool(conf_ov)
+                    )
                     status = _review_status_html()
                     new_idx = idx + 1
                     img2, raw2, info2, _, _ = get_review_item(new_idx)
@@ -3144,8 +3154,8 @@ def smoke_signal_tab():
                 ocr_done.change(load_review, outputs=review_load_outputs)
                 pipeline_done.change(load_review, outputs=review_load_outputs)
 
-                accept_btn.click(do_accept, inputs=[current_idx, final_text_box, reviewer_name, reason_code], outputs=action_outputs)
-                edit_btn.click(do_accept, inputs=[current_idx, final_text_box, reviewer_name, reason_code], outputs=action_outputs)
+                accept_btn.click(do_accept, inputs=[current_idx, final_text_box, reviewer_name, reason_code, conf_override], outputs=action_outputs)
+                edit_btn.click(do_accept, inputs=[current_idx, final_text_box, reviewer_name, reason_code, conf_override], outputs=action_outputs)
                 reject_btn.click(do_reject, inputs=[current_idx, final_text_box, reviewer_name, reason_code], outputs=action_outputs)
                 quar_btn.click(do_quarantine, inputs=[current_idx, final_text_box, reviewer_name, reason_code], outputs=action_outputs)