import json, sys, time
sys.path.insert(0, "/tank/spikes/parakeet-ab/code")
from bench import post
url = sys.argv[1]
rows = {json.loads(l)["id"]: json.loads(l) for l in open("/tank/spikes/parakeet-ab/data/lat.jsonl")}
from bench import hostpath; w = lambda i: open(hostpath(rows[i]["wav"]), "rb").read()
def show(tag, i):
    r = post(url, w(i)); print(f"{tag:10s} {i:12s} dur {rows[i]['dur']:6.2f}s e2e {r['e2e_ms']:8.1f} server {r['server_ms']}", flush=True)
for k in range(5): show("repeat", "b3_8_10")
for k in range(5): show("fresh", f"b3_8_{k:02d}")
for k in range(5): show("fresh1-3", f"b1_3_{k:02d}")
for k in range(3): show("repeat", "b1_3_00")
for k in range(3): show("fresh20", f"b20_60_{k:02d}")
for k in range(3): show("repeat", "b20_60_00")