| """ae3-tl: serve the image service, or read one image.""" |
|
|
| import argparse |
| import json |
| import sys |
|
|
|
|
| def _serve(module: str, host: str, port: int, log_level: str) -> None: |
| import uvicorn |
|
|
| uvicorn.run(module, host=host, port=port, log_level=log_level) |
|
|
|
|
| def main(argv: list[str] | None = None) -> int: |
| parser = argparse.ArgumentParser(prog="ae3-tl", description=__doc__) |
| commands = parser.add_subparsers(dest="command", required=True) |
| for name, target, default_port in (("serve", "ae3_tl.serve.app:app", 8000),): |
| command = commands.add_parser(name) |
| command.add_argument("--host", default="0.0.0.0") |
| command.add_argument("--port", type=int, default=default_port) |
| command.add_argument("--log-level", default="info") |
| command.set_defaults(target=target) |
| extract = commands.add_parser("extract", help="read one page image and print the document") |
| extract.add_argument("image") |
| extract.add_argument("--words") |
| extract.add_argument("--format", default="json", choices=("json", "markdown", "html", "csv")) |
| extract.add_argument("--include-crops", action="store_true") |
| args = parser.parse_args(argv) |
|
|
| if args.command == "serve": |
| _serve(args.target, args.host, args.port, args.log_level) |
| return 0 |
|
|
| from ae3_tl.backends import create_engine |
| from ae3_tl.image import service |
| from ae3_tl.serve.settings import load_settings |
|
|
| settings = load_settings() |
| engine = create_engine(settings) |
| words = open(args.words, "rb").read() if args.words else None |
| document = service.extract(engine, settings, open(args.image, "rb").read(), words, |
| fmt=args.format, include_crops=args.include_crops) |
| print(json.dumps(document, indent=2)) |
| return 0 |
|
|
|
|
| if __name__ == "__main__": |
| sys.exit(main()) |
|
|
