fix: address verify findings - add visualize subcommand, curl healthcheck, IoU in JSON, 50KB threshold

ts_anomaly_td/cli.py

@@ -79,6 +79,33 @@ def cmd_e2e(args):
     return run_e2e(args)
 
 
+def cmd_visualize(args):
+    """visualize 子命令：渲染双栏对比图。"""
+    logging.basicConfig(level=logging.INFO, format="%(asctime)s [%(levelname)s] %(message)s")
+
+    from ts_anomaly_td.io_csv import read_csv
+    from ts_anomaly_td.visualize import render_contrast
+
+    ts, vals, labels = read_csv(args.gt_csv)
+
+    conn = TDConnection(args.url)
+    try:
+        db = args.db or "ts_anomaly"
+        conn.execute_no_result(f"USE {db}")
+
+        if args.algos:
+            from ts_anomaly_td.detection import detect_all_algos
+            results = detect_all_algos(conn, args.stable, algos=args.algos.split(","))
+        else:
+            from ts_anomaly_td.detection import detect_all_algos
+            results = detect_all_algos(conn, args.stable)
+
+        render_contrast(ts, vals, labels, results, args.output, title=args.stable)
+        logger.info("visualize: %s", args.output)
+    finally:
+        conn.close()
+
+
 def cmd_watch(args):
     """watch 子命令：TMQ 准实时监控。"""
     logging.basicConfig(level=logging.INFO, format="%(asctime)s [%(levelname)s] %(message)s")
@@ -132,6 +159,15 @@ def build_parser() -> argparse.ArgumentParser:
     p_e2e.add_argument("--log-dir", default="logs", help="JSON 日志目录 (default: logs)")
     p_e2e.add_argument("--url", default="ws://root:taosdata@localhost:6041", help="TDengine WebSocket URL")
 
+    # visualize
+    p_viz = sub.add_parser("visualize", help="渲染 GT 对比可视化图")
+    p_viz.add_argument("--gt-csv", required=True, help="GT CSV 文件路径")
+    p_viz.add_argument("--stable", required=True, help="supertable 名称")
+    p_viz.add_argument("--db", default="ts_anomaly", help="数据库名")
+    p_viz.add_argument("--algos", default=None, help="逗号分隔算法列表 (默认全 6 个)")
+    p_viz.add_argument("--output", required=True, help="PNG 输出路径")
+    p_viz.add_argument("--url", default="ws://root:taosdata@localhost:6041", help="TDengine WebSocket URL")
+
     # watch (Iter-2)
     p_watch = sub.add_parser("watch", help="TMQ 准实时异常监控")
     p_watch.add_argument("--stable", required=True, help="supertable 名称")
@@ -151,6 +187,7 @@ def main():
         "detect-batch": cmd_detect_batch,
         "forecast-anomaly": cmd_forecast_anomaly,
         "e2e": cmd_e2e,
+        "visualize": cmd_visualize,
         "watch": cmd_watch,
     }