| """ |
| ui/search_progress.py |
| --------------------- |
| Live checklist renderer for the SEARCHING stage: turns the set of completed |
| LangGraph node names + per-source paper counts into the step-by-step HTML |
| checklist markup. |
| """ |
|
|
| _SEARCH_STEPS = [ |
| ("arxiv", "Searching arXiv"), |
| ("semantic_scholar", "Searching Semantic Scholar"), |
| ("open_alex", "Searching OpenAlex"), |
| ("aggregation", "Merging & deduplicating"), |
| ("reranker", "Re-ranking by relevance (SPECTER)"), |
| ("clustering", "Clustering by approach"), |
| ] |
|
|
|
|
| def search_steps_html(completed: set, counts: dict) -> str: |
| sources_done = all(s in completed for s in ("arxiv", "semantic_scholar", "open_alex")) |
|
|
| def state_of(key): |
| if key in completed: |
| return "done" |
| if key in ("arxiv", "semantic_scholar", "open_alex"): |
| return "active" |
| if key == "aggregation": |
| return "active" if sources_done else "pending" |
| if key == "reranker": |
| return "active" if "aggregation" in completed else "pending" |
| if key == "clustering": |
| return "active" if "reranker" in completed else "pending" |
| return "pending" |
|
|
| rows = [] |
| for key, label in _SEARCH_STEPS: |
| s = state_of(key) |
| ic = "✓" if s == "done" else ("•" if s == "pending" else "") |
| cnt = f'<span class="step-count">{counts[key]} papers</span>' if key in counts else "" |
| rows.append(f'<div class="step-row {s}"><div class="step-ic">{ic}</div><div>{label}</div>{cnt}</div>') |
| return '<div class="search-wrap">' + "".join(rows) + "</div>" |
|
|
