feat: add TOPIK track labels

Author: dixonfrederick

src/nokaman/cli.py

@@ -2,6 +2,7 @@
 
 import json
 from pathlib import Path
+from typing import Optional
 
 import typer
 from rich.console import Console
@@ -30,6 +31,10 @@
 console = Console()
 
 
+def _print_json(data: object) -> None:
+    console.print_json(data=data, ensure_ascii=True)
+
+
 @app.command("version")
 def version_cmd() -> None:
     console.print(f"NokaMan {__version__}")
@@ -50,7 +55,7 @@ def stats_cmd() -> None:
         skill = parts[1] if len(parts) > 1 else "?"
         by_lang[lang] += 1
         by_skill[skill] += 1
-    console.print_json(
+    _print_json(
         data={
             "version": __version__,
             "n_samples": len(list_sample_files()),
@@ -65,7 +70,7 @@ def stats_cmd() -> None:
 def demo_cmd(lang: str = typer.Option("en", "--lang", "-l")) -> None:
     """Full multi-skill demo for a language (end-to-end runnable)."""
     result = evaluate_demo(lang)
-    console.print_json(data=result)
+    _print_json(data=result)
     OUT_DIR.mkdir(parents=True, exist_ok=True)
     path = OUT_DIR / f"demo_{lang}.json"
     path.write_text(json.dumps(result, indent=2) + "\n", encoding="utf-8")
@@ -85,7 +90,7 @@ def languages_list() -> None:
 
 
 @rubrics_app.command("list")
-def rubrics_list(lang: str | None = typer.Option(None, "--lang", "-l")) -> None:
+def rubrics_list(lang: Optional[str] = typer.Option(None, "--lang", "-l")) -> None:
     files = list_rubric_files()
     if lang:
         files = [p for p in files if p.stem == lang.strip().lower()]
@@ -105,8 +110,8 @@ def rubrics_list(lang: str | None = typer.Option(None, "--lang", "-l")) -> None:
 @eval_app.command("text")
 def eval_text(
     lang: str = typer.Option("en", "--lang", "-l"),
-    text: str | None = typer.Option(None, "--text", "-t"),
-    file: Path | None = typer.Option(None, "--file", "-f", exists=True, dir_okay=False),
+    text: Optional[str] = typer.Option(None, "--text", "-t"),
+    file: Optional[Path] = typer.Option(None, "--file", "-f", exists=True, dir_okay=False),
     skill: str = typer.Option("writing", "--skill", "-s"),
 ) -> None:
     if file is not None:
@@ -116,12 +121,12 @@ def eval_text(
     else:
         console.print("[red]Provide --text or --file[/red]")
         raise typer.Exit(code=1)
-    console.print_json(data=result)
+    _print_json(data=result)
 
 
 @eval_app.command("demo")
 def eval_demo(lang: str = typer.Option("en", "--lang", "-l")) -> None:
-    console.print_json(data=evaluate_demo(lang))
+    _print_json(data=evaluate_demo(lang))
 
 
 @eval_app.command("samples")
@@ -150,7 +155,7 @@ def eval_samples() -> None:
 
 @eval_app.command("batch")
 def eval_batch(
-    out: Path | None = typer.Option(None, "--out", "-o"),
+    out: Optional[Path] = typer.Option(None, "--out", "-o"),
     table: bool = typer.Option(True, "--table/--json-only"),
 ) -> None:
     report = batch_evaluate()
@@ -188,7 +193,7 @@ def eval_summary() -> None:
         lang = stem.split("_")[0] if "_" in stem else "?"
         by_lang[lang] = by_lang.get(lang, 0) + 1
     report = batch_evaluate()
-    console.print_json(
+    _print_json(
         data={
             "version": __version__,
             "n_samples": len(files),
@@ -205,7 +210,7 @@ def eval_placement(
     answer: list[str] = typer.Option(..., "--answer", "-a", help="Repeat for multiple answers"),
 ) -> None:
     result = placement_test(lang, answer)
-    console.print_json(data=result)
+    _print_json(data=result)
 
 
 @train_app.command("toy")

src/nokaman/models/toy.py

@@ -123,15 +123,17 @@ def _framework_bands(language: str, cefr: str, score: float) -> dict:
     c = (cefr or "A1").upper()
     # rough ladders for demo UX
     jlpt = {"A1": "N5", "A2": "N4", "B1": "N3", "B2": "N2", "C1": "N1", "C2": "N1+"}.get(c, "N5")
-    topik = {"A1": "1", "A2": "2", "B1": "3", "B2": "4", "C1": "5", "C2": "6"}.get(c, "1")
+    topik = _topik_band(c)
     hsk = {"A1": "1", "A2": "2", "B1": "3", "B2": "4", "C1": "5", "C2": "6"}.get(c, "1")
     ielts = round(3.0 + (score / 100.0) * 6.0, 1)  # ~3.0–9.0
     toeic = int(200 + score * 7)  # ~200–900
     out = {"cefr": c}
     if lang == "ja":
         out["jlpt"] = jlpt
     if lang == "ko":
-        out["topik"] = topik
+        out["topik"] = topik["level"]
+        out["topik_track"] = topik["track"]
+        out["topik_label"] = topik["label"]
     if lang == "zh":
         out["hsk"] = hsk
     if lang == "en":
@@ -140,6 +142,19 @@ def _framework_bands(language: str, cefr: str, score: float) -> dict:
     return out
 
 
+def _topik_band(cefr: str) -> dict[str, str]:
+    level = {"A1": "1", "A2": "2", "B1": "3", "B2": "4", "C1": "5", "C2": "6"}.get(
+        cefr,
+        "1",
+    )
+    track = "TOPIK I" if level in {"1", "2"} else "TOPIK II"
+    return {
+        "level": level,
+        "track": track,
+        "label": f"{track} Level {level}",
+    }
+
+
 def _script_bonus(text: str, language: str) -> float:
     if not text:
         return 0.0

tests/test_framework_bands.py

@@ -15,3 +15,22 @@ def test_demo_includes_frameworks() -> None:
     d = evaluate_demo("ja")
     assert "framework_bands" in d
     assert d["framework_bands"].get("jlpt")
+
+
+def test_ko_framework_bands_include_topik_track() -> None:
+    r = ToyAbilityModel("ko").score_text(
+        "I study Korean every morning and write short practice notes.",
+        skill="writing",
+    )
+    bands = r["framework_bands"]
+    assert bands["topik"] in {"1", "2", "3", "4", "5", "6"}
+    assert bands["topik_track"] in {"TOPIK I", "TOPIK II"}
+    assert bands["topik_label"] == f"{bands['topik_track']} Level {bands['topik']}"
+
+
+def test_ko_demo_includes_topik_field() -> None:
+    d = evaluate_demo("ko")
+    bands = d["framework_bands"]
+    assert bands.get("topik")
+    assert bands.get("topik_track")
+    assert bands.get("topik_label")