"""
Step 2 helper: Verify that both GLM-OCR model files are present and print their sizes.
"""
import os, sys
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
from _ocr_cfg import OCR_DIR

if not OCR_DIR:
    print("ERROR: OCR_DIR not resolved. Run preflight_workdir.py first.")
    sys.exit(1)

model_dir = os.path.join(OCR_DIR, "models", "GLM-OCR-GGUF")

for fname in ["GLM-OCR-Q8_0.gguf", "mmproj-GLM-OCR-Q8_0.gguf"]:
    fpath = os.path.join(model_dir, fname)
    if os.path.exists(fpath):
        mb = round(os.path.getsize(fpath) / 1024**2)
        print(f"{fname}: {mb} MB")
    else:
        print(f"MISSING: {fname}")