| """CLI dispatcher for `python3 -m data_label_factory.identify <command>`. |
| |
| Subcommands: |
| index → build_index.main |
| verify → verify_index.main |
| train → train.main |
| serve → serve.main |
| |
| Each is lazy-loaded so users only pay the import cost for the command they |
| actually invoke. |
| """ |
|
|
| from __future__ import annotations |
|
|
| import sys |
|
|
|
|
| HELP = """\ |
| data_label_factory.identify — open-set image retrieval |
| |
| usage: python3 -m data_label_factory.identify <command> [options] |
| |
| commands: |
| index Build a CLIP retrieval index from a folder of reference images |
| verify Self-test an index and report margin / confusable pairs |
| train Contrastive fine-tune a projection head (improves accuracy) |
| scrape_prices Scrape live market prices and cache them as a sidecar to the index |
| serve Run an HTTP server that exposes the index as /api/falcon |
| mcp Run an MCP server that bridges the agent gateway over stdio |
| webui-mcp Run an MCP server for website UI labeling over stdio |
| |
| run any command with --help for its options. The full blueprint is in |
| data_label_factory/identify/README.md. |
| """ |
|
|
|
|
| def main(argv: list[str] | None = None) -> int: |
| args = list(argv) if argv is not None else sys.argv[1:] |
| if not args or args[0] in ("-h", "--help", "help"): |
| print(HELP) |
| return 0 |
|
|
| cmd = args[0] |
| rest = args[1:] |
|
|
| if cmd == "index": |
| from .build_index import main as _main |
| return _main(rest) |
| if cmd == "verify": |
| from .verify_index import main as _main |
| return _main(rest) |
| if cmd == "train": |
| from .train import main as _main |
| return _main(rest) |
| if cmd in ("scrape_prices", "scrape-prices", "prices"): |
| from .scrape_prices import main as _main |
| return _main(rest) |
| if cmd == "serve": |
| from .serve import main as _main |
| return _main(rest) |
| if cmd in ("mcp", "mcp-server", "mcp_server"): |
| from .mcp_server import main as _main |
| return _main() |
| if cmd in ("webui-mcp", "webui_mcp"): |
| from .webui_mcp import main as _main |
| return _main() |
|
|
| print(f"unknown command: {cmd}\n", file=sys.stderr) |
| print(HELP, file=sys.stderr) |
| return 1 |
|
|
|
|
| if __name__ == "__main__": |
| sys.exit(main()) |
|
|
