from app.database import SessionLocal
from app.services import rerank_service as rs
from sqlalchemy import text
db=SessionLocal()
ids=[797,1704,1380,3273,1389]
rows={r[0]:" ".join((r[1] or "").split()) for r in db.execute(text("select id,question_text from questions where id = any(:i)"),{"i":ids}).fetchall()}
docs=[rows[i] for i in ids]
for q in ["what causes croup","croup","barking cough in a toddler"]:
    s=rs.rerank(q,docs)
    print(q, [(i,round(x,4)) for i,x in zip(ids,s)])