import argparse, json, time, numpy as np
def main():
    p = argparse.ArgumentParser(prog="edge-embed")
    p.add_argument("--profile", default="edge_gpu")
    p.add_argument("--device", default="auto")
    sub = p.add_subparsers(dest="cmd")
    ep = sub.add_parser("embed"); ep.add_argument("input"); ep.add_argument("--modality", default="auto")
    cp = sub.add_parser("compare"); cp.add_argument("a"); cp.add_argument("b")
    bp = sub.add_parser("benchmark"); bp.add_argument("-n", type=int, default=50)
    sub.add_parser("info")
    args = p.parse_args()
    if not args.cmd: p.print_help(); return
    import logging; logging.basicConfig(level=logging.INFO)
    from edge_multimodal_embeddings.core.config import EmbedderConfig, HardwareProfile
    from edge_multimodal_embeddings.core.embedder import EdgeMultimodalEmbedder
    cfg = EmbedderConfig.from_profile(HardwareProfile(args.profile))
    if args.device != "auto": cfg.device = args.device
    emb = EdgeMultimodalEmbedder(cfg)
    if args.cmd == "embed":
        t0 = time.time(); e = emb.embed(args.input, modality=args.modality)
        print(f"Shape: {e.shape} | Norm: {np.linalg.norm(e):.4f} | {(time.time()-t0)*1000:.0f}ms")
    elif args.cmd == "compare":
        print(f"Similarity: {emb.similarity(emb.embed(args.a), emb.embed(args.b)):.4f}")
    elif args.cmd == "benchmark": print(json.dumps(emb.benchmark(args.n), indent=2, default=str))
    elif args.cmd == "info": print(json.dumps(emb.get_status(), indent=2, default=str))
    emb.unload()
if __name__ == "__main__": main()