import json, random, collections, os
random.seed(1234)
MANI="/scratch/prep/manifests"; OUT="/root/devhard"; os.makedirs(OUT,exist_ok=True)
def load(p): return [json.loads(l) for l in open(p,encoding="utf-8")]
dev_all=[]
for L in ["lin","sna"]:
    rows=load(f"{MANI}/waxal_{L}_train.jsonl")
    by=collections.defaultdict(list)
    for r in rows: by[r.get("speaker")].append(r)
    cand=[s for s,v in by.items() if 8<=len(v)<=70]
    random.shuffle(cand)
    target=min(400,int(0.05*len(rows)))
    hold=set(); n=0
    for s in cand:
        if n>=target: break
        hold.add(s); n+=len(by[s])
    dev=[r for r in rows if r.get("speaker") in hold]
    tmin=[r for r in rows if r.get("speaker") not in hold]
    for r in dev: r["lang"]=L
    dev_all+=dev
    with open(f"{OUT}/devhard_{L}.jsonl","w",encoding="utf-8") as f:
        for r in dev: f.write(json.dumps(r,ensure_ascii=False)+"\n")
    with open(f"{OUT}/train_{L}_min.jsonl","w",encoding="utf-8") as f:
        for r in tmin: f.write(json.dumps(r,ensure_ascii=False)+"\n")
    print(f"{L}: holdout_spk={len(hold)} devhard={len(dev)} train_min={len(tmin)} (was {len(rows)})")
with open(f"{OUT}/devhard_linsna.jsonl","w",encoding="utf-8") as f:
    for r in dev_all: f.write(json.dumps(r,ensure_ascii=False)+"\n")
print("DEVHARD_LINSNA_DONE total",len(dev_all))