53 lines
1.6 KiB
Python
53 lines
1.6 KiB
Python
|
"""Data controls tab: model-improvement switch (read-only otherwise).
|
||
|
|
|
||
|
|
The switch row label is not yet pinned from recon, so resolution
|
||
|
|
prefers the single switch on the tab and falls back to keyword
|
||
|
|
match. Import/Delete rows are inventoried, never touched.
|
||
|
|
"""
|
||
|
|
from hatch_menu import controls, dialog
|
||
|
|
|
||
|
|
TAB = "Data controls"
|
||
|
|
_KEYWORDS = ("improv", "train", "model", "data", "usage")
|
||
|
|
|
||
|
|
|
||
|
|
def _resolve(ws):
|
||
|
|
"""The improvement switch dict, or None when not resolvable."""
|
||
|
|
if not dialog.goto_tab(ws, TAB):
|
||
|
|
return None
|
||
|
|
switches = controls.list_switches(ws)
|
||
|
|
if len(switches) == 1:
|
||
|
|
return switches[0]
|
||
|
|
for kw in _KEYWORDS:
|
||
|
|
for s in switches:
|
||
|
|
blob = ((s.get("label") or "") + " "
|
||
|
|
+ (s.get("aria") or "")).lower()
|
||
|
|
if kw in blob:
|
||
|
|
return s
|
||
|
|
return None
|
||
|
|
|
||
|
|
|
||
|
|
def ai_improvement(ws):
|
||
|
|
"""Improvement-switch state: True/False, None when unreadable."""
|
||
|
|
sw = _resolve(ws)
|
||
|
|
return None if sw is None else bool(sw.get("checked"))
|
||
|
|
|
||
|
|
|
||
|
|
def set_ai_improvement(ws, on):
|
||
|
|
"""Set the improvement switch; verify-then-fallback. Bool."""
|
||
|
|
sw = _resolve(ws)
|
||
|
|
if sw is None:
|
||
|
|
return False
|
||
|
|
label = (sw.get("aria") or "").strip() \
|
||
|
|
or (sw.get("label") or "")[:40].strip()
|
||
|
|
if not label:
|
||
|
|
return False
|
||
|
|
return controls.set_switch(ws, label, on)
|
||
|
|
|
||
|
|
|
||
|
|
def describe(ws):
|
||
|
|
"""Inventory: switch state + row names (import/delete read-only)."""
|
||
|
|
state = ai_improvement(ws)
|
||
|
|
return {"ai_improvement":
|
||
|
|
("on" if state else "off") if state is not None else None,
|
||
|
|
"rows": dialog.describe_rows(ws, TAB)}
|