ExistedYear commited on
Commit
c2dc23a
·
1 Parent(s): 5ec3383

three-tier inbox tags: SCAM / SUSPICIOUS / SAFE matching detail card

Browse files
Files changed (1) hide show
  1. app.py +15 -6
app.py CHANGED
@@ -533,20 +533,29 @@ def mob_scan_all(progress=gr.Progress()):
533
  for i, m in enumerate(SEED_SMS):
534
  progress(i / len(SEED_SMS), desc=f"Scanning {i+1}/{len(SEED_SMS)} · {m['n']}")
535
  try:
536
- spam = _run(m["b"]).get("label") == "spam"
537
  except Exception:
538
- spam = False
 
539
  threats += spam
540
- colour = "var(--hi)" if spam else "var(--lo)"
541
- soft = "var(--hi-s)" if spam else "var(--lo-s)"
542
- brd = "var(--hi-b)" if spam else "var(--lo-b)"
 
 
 
 
 
 
 
 
543
  hit = "expected scam" if (spam == (m["e"] == "spam")) else "expected safe"
544
  rows.append(
545
  f'<div class="in-r" onclick="void 0"><span class="in-d" style="background:{colour}"></span>'
546
  f'<div class="in-b"><div class="in-s">{esc(m["n"])}<em>{hit}</em></div>'
547
  f'<div class="in-x">{esc(m["b"])}</div></div>'
548
  f'<span class="in-l" style="background:{soft};color:{colour};border:1px solid {brd}">'
549
- f'{"SCAM" if spam else "SAFE"}</span></div>'
550
  )
551
  progress(1, desc="done")
552
  stats = ('<div class="stats">'
 
533
  for i, m in enumerate(SEED_SMS):
534
  progress(i / len(SEED_SMS), desc=f"Scanning {i+1}/{len(SEED_SMS)} · {m['n']}")
535
  try:
536
+ res = _run(m["b"])
537
  except Exception:
538
+ res = None
539
+ spam = bool(res) and res.get("label") == "spam"
540
  threats += spam
541
+ if res is None:
542
+ tier, colour, soft, brd, tag = "lo", "var(--faint)", "var(--panel2)", "var(--line2)", "—"
543
+ else:
544
+ # Same three tiers as the detail card: high → SCAM, medium → SUSPICIOUS.
545
+ tier, colour, _, _ = risk(res.get("confidence", 0), res.get("risk_level", ""))
546
+ if tier == "hi":
547
+ soft, brd, tag = "var(--hi-s)", "var(--hi-b)", "SCAM"
548
+ elif tier == "md":
549
+ soft, brd, tag = "var(--md-s)", "var(--md-b)", "SUSPICIOUS"
550
+ else:
551
+ soft, brd, tag = "var(--lo-s)", "var(--lo-b)", "SAFE"
552
  hit = "expected scam" if (spam == (m["e"] == "spam")) else "expected safe"
553
  rows.append(
554
  f'<div class="in-r" onclick="void 0"><span class="in-d" style="background:{colour}"></span>'
555
  f'<div class="in-b"><div class="in-s">{esc(m["n"])}<em>{hit}</em></div>'
556
  f'<div class="in-x">{esc(m["b"])}</div></div>'
557
  f'<span class="in-l" style="background:{soft};color:{colour};border:1px solid {brd}">'
558
+ f'{tag}</span></div>'
559
  )
560
  progress(1, desc="done")
561
  stats = ('<div class="stats">'