Spaces:
Running
Running
feat: add colored category badge (Causal/Human/Spatial&Object) in bottom-left meta
Browse files
app.py
CHANGED
|
@@ -264,10 +264,29 @@ def _stats_md() -> str:
|
|
| 264 |
)
|
| 265 |
|
| 266 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 267 |
def _meta_md(model: str, task: dict[str, Any], state: dict) -> str:
|
| 268 |
done_by_me = state.get("submitted_count", 0)
|
| 269 |
claimed_n = state.get("claimed_n", 0)
|
|
|
|
| 270 |
lines = [
|
|
|
|
| 271 |
f"**本次会话已提交**:{done_by_me} 条(共已领取 {claimed_n} 条)",
|
| 272 |
f"**模型**:`{model}`",
|
| 273 |
f"**task_id**:`{task['task_id']}`",
|
|
|
|
| 264 |
)
|
| 265 |
|
| 266 |
|
| 267 |
+
CATEGORY_LABELS = {
|
| 268 |
+
"casual": ("Causal", "#f59e0b"), # amber
|
| 269 |
+
"human": ("Human", "#3b82f6"), # blue
|
| 270 |
+
"spatial_object": ("Spatial & Object", "#10b981"), # emerald
|
| 271 |
+
}
|
| 272 |
+
|
| 273 |
+
|
| 274 |
+
def _category_badge(task: dict[str, Any]) -> str:
|
| 275 |
+
cat = task.get("category") or "?"
|
| 276 |
+
label, color = CATEGORY_LABELS.get(cat, (cat, "#6b7280"))
|
| 277 |
+
return (
|
| 278 |
+
f'<span style="display:inline-block;padding:4px 12px;border-radius:999px;'
|
| 279 |
+
f'background:{color};color:#fff;font-weight:600;font-size:14px;'
|
| 280 |
+
f'letter-spacing:0.3px;">{label}</span>'
|
| 281 |
+
)
|
| 282 |
+
|
| 283 |
+
|
| 284 |
def _meta_md(model: str, task: dict[str, Any], state: dict) -> str:
|
| 285 |
done_by_me = state.get("submitted_count", 0)
|
| 286 |
claimed_n = state.get("claimed_n", 0)
|
| 287 |
+
badge = _category_badge(task)
|
| 288 |
lines = [
|
| 289 |
+
f"**类别**:{badge}",
|
| 290 |
f"**本次会话已提交**:{done_by_me} 条(共已领取 {claimed_n} 条)",
|
| 291 |
f"**模型**:`{model}`",
|
| 292 |
f"**task_id**:`{task['task_id']}`",
|