import os
import sys
from pathlib import Path
sys.path.insert(0, str(Path(".").resolve()))
from PIL import Image
import src.common.vision_ocr as vision_ocr

images = [
    r"captures/pipeline_run/20260919_200603_398_STAGE2_MAP_DEVICE_HOVERED.png",
    r"captures/pipeline_run/20260919_200607_553_STAGE2_ATLAS_UI_PROBE_1.png",
    r"captures/pipeline_run/20260919_200618_832_STAGE2_ATLAS_UI_PROBE_RETRY_1.png",
]

for path in images:
    if not os.path.exists(path):
        print(f"File not found: {path}")
        continue
    img = Image.open(path)
    res = vision_ocr.safe_recognize_pil_sync(img)
    print(f"\n=== {path} ===")
    for line in res.get("lines", []):
        txt = line.get("text", "")
        box = line.get("box", [])
        print(f"  [{box}] {txt}")