"""Recognize one image locally and write UTF-8 text and JSON."""
import argparse
import json
from pathlib import Path
from PIL import Image
from runtime.engine import OCR


def main():
    p=argparse.ArgumentParser(description=__doc__)
    p.add_argument('image',type=Path)
    p.add_argument('--backend',choices=['cpu','gpu','auto'],default='cpu')
    p.add_argument('--output',type=Path,default=Path('result.txt'))
    args=p.parse_args()
    with Image.open(args.image) as image:r=OCR(backend=args.backend).recognize(image,include_images=False)
    args.output.write_text(r['full_text']+'\n',encoding='utf-8')
    args.output.with_suffix('.json').write_text(json.dumps(r,ensure_ascii=False,indent=2)+'\n',encoding='utf-8')
    print(f'Saved {args.output}; backend={r["backend"]}; lines={len(r["lines"])}')


if __name__=='__main__':main()