studyOverflow commited on
Commit
c8e1c40
·
verified ·
1 Parent(s): f7dcac9

feat: add colored category badge (Causal/Human/Spatial&Object) in bottom-left meta

Browse files
Files changed (1) hide show
  1. app.py +19 -0
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']}`",