server.py 48 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005
  1. # -*- coding: utf-8 -*-
  2. """搜索评估案例查看 server。
  3. 沿用 图文排版搜索评估.html 的版式(卡片 + dialog 详情 + rubric 评分条),
  4. 数据实时扫描 runs_full/*/form_*.json —— runs_full 下每新增一个 q 文件夹,刷新即出现。
  5. 分页:query → 三种形式(A/B/C) → 三个渠道 三行从上到下。
  6. 用法:python server.py [port] 默认 8770,浏览器开 http://0.0.0.0:8770
  7. """
  8. import json, re, glob, sys, pathlib, subprocess, threading
  9. from datetime import datetime
  10. from http.server import BaseHTTPRequestHandler, ThreadingHTTPServer
  11. from urllib.parse import urlparse, parse_qs
  12. try: # Windows 控制台默认 cp1252,中文 print 会崩,统一切 utf-8
  13. sys.stdout.reconfigure(encoding="utf-8")
  14. except Exception:
  15. pass
  16. HERE = pathlib.Path(__file__).parent
  17. sys.path.insert(0, str(HERE))
  18. sys.path.insert(0, str(HERE.parent)) # fixed_query_eval:让兄弟模块 batch_extract_procedures 可 import
  19. PORT = int(sys.argv[1]) if len(sys.argv) > 1 else 8770
  20. PLAT = {"xhs": "小红书", "gzh": "公众号", "zhihu": "知乎", "x": "X", "bili": "B站", "douyin": "抖音",
  21. "sph": "视频号", "youtube": "YouTube", "github": "GitHub", "toutiao": "头条", "weibo": "微博"}
  22. KT = {"procedure": "工序", "step": "步骤", "tool": "工具"}
  23. # 从 taxonomy 取动作叶子/类型名,用于把 original_q 解析回原始维度(动作×类型 正交)
  24. # 路径优先级:search_eval/evaluation/(主源,IDE 编辑那份就是 runtime 实际读的)
  25. # → test_script/evaluation/(历史副本兜底)→ script/evaluation/(更老兜底)
  26. # 谁也找不到时整目录扫空,server 仍能起。
  27. EVALDIR = HERE / "evaluation"
  28. if not EVALDIR.exists():
  29. EVALDIR = HERE.parent.parent / "test_script" / "evaluation"
  30. if not EVALDIR.exists():
  31. EVALDIR = HERE.parent / "evaluation"
  32. try:
  33. _jm = json.load(open(EVALDIR / "judged_matrix.json", encoding="utf-8"))
  34. ACT_L1 = {a["name"]: a["l1"] for a in _jm["actions"]}
  35. ACTION_SET = set(ACT_L1)
  36. TYPE_SET = {t["name"] for t in _jm["types"]}
  37. ACTIONS_TAX = [{"name": a["name"], "l1": a["l1"], "l2": a.get("l2", "")} for a in _jm["actions"]]
  38. TYPES_TAX = [{"name": t["name"], "l1": t["l1"]} for t in _jm["types"]]
  39. # taxonomy 顺序沿用 judged_matrix(严格版);矩阵分值改用 type_action_scores(宽松版) —
  40. # 两份是同一组 27×50 cell 的独立 gemini judging,前者只 53 格到 tier3,后者 156 格到 score3
  41. _tas = json.load(open(EVALDIR / "type_action_scores.json", encoding="utf-8"))["scores"]
  42. _MATRIX = []
  43. for a in _jm["actions"]:
  44. row = []
  45. for t in _jm["types"]:
  46. rec = _tas.get(t["name"], {}).get(a["name"])
  47. row.append({"tier": rec["score"], "r": rec.get("reason", "")} if rec else {})
  48. _MATRIX.append(row)
  49. except Exception:
  50. ACT_L1, ACTION_SET, TYPE_SET, ACTIONS_TAX, TYPES_TAX, _MATRIX = {}, set(), set(), [], [], []
  51. ACTIVE_TASKS = {}
  52. ACTIVE_REEVALS = {}
  53. from batch_extract_procedures import _short_case, _source_to_dsl_input, _write_meta, _composite_score
  54. def run_extraction_task(q, folder_name, src_path, out_dir, engine, model):
  55. task_key = f"{q}/{folder_name}"
  56. log_path = out_dir / "_extract.log"
  57. try:
  58. out_dir.mkdir(parents=True, exist_ok=True)
  59. if engine == "cyber_runner":
  60. script_path = HERE / "procedure-dsl" / "run_cyber.py"
  61. else:
  62. script_path = HERE / "procedure-dsl" / "run_procedure_dsl.py"
  63. cmd = [
  64. sys.executable, "-u", str(script_path),
  65. str(src_path),
  66. "--out-dir", str(out_dir),
  67. "--model", model,
  68. "--max-turns", "300"
  69. ]
  70. if engine != "cyber_runner":
  71. cmd.extend(["--max-retries", "3"])
  72. flags = subprocess.CREATE_NEW_PROCESS_GROUP if sys.platform == "win32" else 0
  73. with open(log_path, "w", encoding="utf-8", buffering=1) as log_fh:
  74. proc = subprocess.Popen(cmd, stdout=log_fh, stderr=subprocess.STDOUT,
  75. cwd=str(HERE), creationflags=flags)
  76. ACTIVE_TASKS[task_key]["pid"] = proc.pid
  77. proc.wait()
  78. if proc.returncode == 0:
  79. try:
  80. import build_workflows
  81. build_workflows.write_one(q, folder_name, runs_dir=HERE / "runs_full")
  82. ACTIVE_TASKS[task_key]["status"] = "success"
  83. except Exception as ex:
  84. ACTIVE_TASKS[task_key]["status"] = "failed"
  85. ACTIVE_TASKS[task_key]["error"] = f"Workflow compilation failed: {ex}"
  86. with open(log_path, "a", encoding="utf-8") as f_err:
  87. f_err.write(f"\n[server error] Workflow compilation failed: {ex}\n")
  88. else:
  89. ACTIVE_TASKS[task_key]["status"] = "failed"
  90. ACTIVE_TASKS[task_key]["error"] = f"Runner failed with exit code {proc.returncode}"
  91. except Exception as e:
  92. ACTIVE_TASKS[task_key]["status"] = "failed"
  93. ACTIVE_TASKS[task_key]["error"] = str(e)
  94. try:
  95. with open(log_path, "a", encoding="utf-8") as f_err:
  96. f_err.write(f"\n[server error] Extraction failed: {e}\n")
  97. except Exception:
  98. pass
  99. # ── 工具解构(fixed_query_eval 新增)──────────────────────────────────────────────
  100. # 单次 LLM 调用(gemini-3.1-flash-lite)即可,比工序解构(多轮 agent)轻得多。
  101. # 仍走 subprocess(tool_extract.py),让 LLM 重依赖留在子进程,server 本身保持轻量。
  102. TOOL_TASKS = {} # {q: {"status", "case_ids", "error", "start_time"}}
  103. def run_tool_extraction_task(q, case_ids, force):
  104. log_path = HERE / "runs_full" / q / "tools" / "_extract.log"
  105. try:
  106. log_path.parent.mkdir(parents=True, exist_ok=True)
  107. cmd = [sys.executable, "-u", str(HERE / "tool_extract.py"),
  108. "--q", q, "--case-ids", ",".join(case_ids)]
  109. if force:
  110. cmd.append("--force")
  111. flags = subprocess.CREATE_NEW_PROCESS_GROUP if sys.platform == "win32" else 0
  112. with open(log_path, "w", encoding="utf-8", buffering=1) as log_fh:
  113. proc = subprocess.Popen(cmd, stdout=log_fh, stderr=subprocess.STDOUT,
  114. cwd=str(HERE), creationflags=flags)
  115. TOOL_TASKS[q]["pid"] = proc.pid
  116. proc.wait()
  117. TOOL_TASKS[q]["status"] = "success" if proc.returncode == 0 else "failed"
  118. if proc.returncode != 0:
  119. TOOL_TASKS[q]["error"] = f"tool_extract exit code {proc.returncode}"
  120. except Exception as e:
  121. TOOL_TASKS[q]["status"] = "failed"
  122. TOOL_TASKS[q]["error"] = str(e)
  123. MODSET = {"文", "图", "视频", "音频"}
  124. TOOLQUAL = {"AI": "AI 模型", "软件": "桌面 APP", "电脑端": "桌面 APP", "在线": "云端 Web",
  125. "网页版": "云端 Web", "代码": "API·CLI", "命令行": "API·CLI", "插件": "插件扩展"}
  126. def parse_dims(oq):
  127. """把组合 query(如 '文 元素生成 提示词 教程')解析回 {动作, 类型, 动作L1, 约束}。"""
  128. toks = (oq or "").split()
  129. action = next((t for t in toks if t in ACTION_SET), None)
  130. type_ = next((t for t in toks if t in TYPE_SET), None)
  131. cons = None
  132. if toks:
  133. t0 = toks[0]
  134. if t0 in MODSET:
  135. cons = {"kind": "模态", "value": t0}
  136. elif t0 in TOOLQUAL:
  137. cons = {"kind": "工具类型", "value": TOOLQUAL[t0]}
  138. return {"action": action, "type": type_, "action_l1": ACT_L1.get(action, ""), "constraint": cons}
  139. def flat_scores(sc):
  140. f = {}
  141. for k, v in (sc or {}).items():
  142. if isinstance(v, dict):
  143. for kk, vv in v.items():
  144. try: f[kk] = int(vv)
  145. except Exception: pass
  146. else:
  147. try: f[k] = int(v)
  148. except Exception: pass
  149. return f
  150. def _recency_hard(date_str):
  151. """按 publish_timestamp 头 10 字符(YYYY-MM-DD)算硬时效:半年内=3 / 两年内=2 / 更早=1。
  152. 取代原 LLM 评的 recency 维度——脚本算更稳,发布时间在帖子抓取时就有,无需 LLM token。
  153. """
  154. try:
  155. d = datetime.strptime((date_str or "")[:10], "%Y-%m-%d")
  156. except (ValueError, TypeError):
  157. return None
  158. days = (datetime.now() - d).days
  159. if days <= 180: return 3
  160. if days <= 730: return 2
  161. return 1
  162. def adapt(r, run, form_name=None):
  163. p = r.get("post", {}); e = r.get("llm_evaluation", {})
  164. # 1. 解析 知识类型 (knowledge_type)
  165. kt = []
  166. kt_raw = e.get("知识类型") or e.get("knowledge_type") or []
  167. for k in kt_raw:
  168. if k in ("工序", "procedure"): kt.append("procedure")
  169. elif k in ("能力", "步骤", "step"): kt.append("step")
  170. elif k in ("工具", "tool"): kt.append("tool")
  171. fs = {}
  172. score_reasons = {}
  173. # 检测是否为 eval_prompt_sample-mod 里的新版 0-10 分数 schema
  174. is_mod_schema = "相关性" in e and isinstance(e["相关性"], dict) and ("和内容制作知识相关" in e["相关性"] or "和 query 相关" in e["相关性"])
  175. if is_mod_schema:
  176. # 新版 0-10 分数格式解析
  177. # 1. 相关性
  178. rel = e.get("相关性") or {}
  179. for subkey, item in rel.items():
  180. if isinstance(item, dict):
  181. score_val = item.get("得分")
  182. reason_val = item.get("理由")
  183. code_key = None
  184. if "内容制作" in subkey or "知识" in subkey:
  185. code_key = "relevance_production"
  186. elif "query" in subkey or "检索" in subkey:
  187. code_key = "relevance_query"
  188. if code_key and score_val is not None:
  189. try:
  190. fs[code_key] = float(score_val)
  191. if reason_val:
  192. score_reasons[code_key] = reason_val
  193. except Exception:
  194. pass
  195. # 2. 质量
  196. q_block = e.get("质量") or {}
  197. fixed = q_block.get("固定维度") or {}
  198. # 固定维度
  199. fixed_keys = {
  200. "时效性": "recency",
  201. "热度性": "popularity",
  202. "评论反馈": "feedback"
  203. }
  204. for cn, code in fixed_keys.items():
  205. item = fixed.get(cn)
  206. if isinstance(item, dict):
  207. score_val = item.get("得分")
  208. reason_val = item.get("理由")
  209. if score_val is not None:
  210. try:
  211. fs[code] = float(score_val)
  212. if reason_val:
  213. score_reasons[code] = reason_val
  214. except Exception:
  215. pass
  216. # 用例 (真实感, 表现力)
  217. usecase = fixed.get("用例") or {}
  218. usecase_keys = {
  219. "真实感": "realism",
  220. "表现力": "expressiveness"
  221. }
  222. for cn, code in usecase_keys.items():
  223. item = usecase.get(cn)
  224. if isinstance(item, dict):
  225. score_val = item.get("得分")
  226. reason_val = item.get("理由")
  227. if score_val is not None:
  228. try:
  229. fs[code] = float(score_val)
  230. if reason_val:
  231. score_reasons[code] = reason_val
  232. except Exception:
  233. pass
  234. # 动态维度
  235. dynamic = q_block.get("动态维度") or {}
  236. # 工序
  237. proc = dynamic.get("工序") or {}
  238. if proc:
  239. item = proc.get("流程完整性")
  240. if isinstance(item, dict):
  241. score_val = item.get("得分")
  242. reason_val = item.get("理由")
  243. if score_val is not None:
  244. try:
  245. fs["procedure_completeness"] = float(score_val)
  246. if reason_val:
  247. score_reasons["procedure_completeness"] = reason_val
  248. except Exception:
  249. pass
  250. field = proc.get("字段完整性") or {}
  251. field_keys = {
  252. "输入完整性": "procedure_input",
  253. "实现完整性": "procedure_implementation",
  254. "输出完整性": "procedure_output"
  255. }
  256. for cn, code in field_keys.items():
  257. item = field.get(cn)
  258. if isinstance(item, dict):
  259. score_val = item.get("得分")
  260. reason_val = item.get("理由")
  261. if score_val is not None:
  262. try:
  263. fs[code] = float(score_val)
  264. if reason_val:
  265. score_reasons[code] = reason_val
  266. except Exception:
  267. pass
  268. item = proc.get("泛化性")
  269. if isinstance(item, dict):
  270. score_val = item.get("得分")
  271. reason_val = item.get("理由")
  272. if score_val is not None:
  273. try:
  274. fs["procedure_generality"] = float(score_val)
  275. if reason_val:
  276. score_reasons["procedure_generality"] = reason_val
  277. except Exception:
  278. pass
  279. # 能力
  280. cap = dynamic.get("能力") or dynamic.get("步骤") or {}
  281. if cap:
  282. field = cap.get("字段完整性") or {}
  283. field_keys = {
  284. "输入完整性": "step_input",
  285. "实现完整性": "step_implementation",
  286. "输出完整性": "step_output"
  287. }
  288. for cn, code in field_keys.items():
  289. item = field.get(cn)
  290. if isinstance(item, dict):
  291. score_val = item.get("得分")
  292. reason_val = item.get("理由")
  293. if score_val is not None:
  294. try:
  295. fs[code] = float(score_val)
  296. if reason_val:
  297. score_reasons[code] = reason_val
  298. except Exception:
  299. pass
  300. item = cap.get("泛化性")
  301. if isinstance(item, dict):
  302. score_val = item.get("得分")
  303. reason_val = item.get("理由")
  304. if score_val is not None:
  305. try:
  306. fs["step_generality"] = float(score_val)
  307. if reason_val:
  308. score_reasons["step_generality"] = reason_val
  309. except Exception:
  310. pass
  311. # 工具
  312. tool = dynamic.get("工具") or {}
  313. if tool:
  314. tool_keys = {
  315. "能力边界覆盖": "tool_boundary",
  316. "有效比较": "tool_comparison",
  317. "参数/接口具体性": "tool_specificity",
  318. "实操示例": "tool_example",
  319. "版本&限制": "tool_limits"
  320. }
  321. for cn, code in tool_keys.items():
  322. item = tool.get(cn)
  323. if isinstance(item, dict):
  324. score_val = item.get("得分")
  325. reason_val = item.get("理由")
  326. if score_val is not None:
  327. try:
  328. fs[code] = float(score_val)
  329. if reason_val:
  330. score_reasons[code] = reason_val
  331. except Exception:
  332. pass
  333. else:
  334. # 兼容老版 1-5 分数 schema (带 "评分" 或 old-style flatness)
  335. is_new_schema = "评分" in e or "知识类型" in e or "制作相关性" in e
  336. CN_TO_EN = {
  337. "相关性": "relevance",
  338. "成品质量": "result_quality",
  339. "可信度": "credibility",
  340. "具体用例": "concrete_use_case",
  341. "完整性": "completeness",
  342. "步骤结构": "step_structure",
  343. "步骤可复现": "step_reproducibility",
  344. "步骤可复现性": "step_reproducibility",
  345. "能力定义": "capability_definition",
  346. "实现深度": "implementation_depth",
  347. "边界失败": "boundary_failure_eval",
  348. "通用性": "generality",
  349. "能力覆盖": "capability_coverage",
  350. "有效对比": "effective_comparison",
  351. "参数具体": "param_specificity",
  352. "实操示例": "worked_example",
  353. "实操用例": "worked_example",
  354. "示例完整": "worked_example",
  355. "版本限制": "version_limits",
  356. "版本说明": "version_limits",
  357. "限制说明": "version_limits",
  358. }
  359. if is_new_schema:
  360. pf = e.get("评分") or {}
  361. for cat, metrics in pf.items():
  362. if isinstance(metrics, dict):
  363. for metric, val in metrics.items():
  364. en_key = CN_TO_EN.get(metric, metric)
  365. if isinstance(val, dict) and "得分" in val:
  366. try: fs[en_key] = int(val["得分"])
  367. except Exception: pass
  368. elif isinstance(val, (int, float)):
  369. fs[en_key] = int(val)
  370. if isinstance(val, dict) and "理由" in val:
  371. score_reasons[en_key] = val["理由"]
  372. else:
  373. fs = flat_scores(e.get("scores", {}))
  374. # 计算均分 (overall)
  375. if is_mod_schema:
  376. rel_keys = {"relevance_production", "relevance_query"}
  377. rel_vals = [v for k, v in fs.items() if k in rel_keys]
  378. qual_vals = [v for k, v in fs.items() if k not in rel_keys]
  379. rel_avg = sum(rel_vals) / len(rel_vals) if rel_vals else None
  380. qual_avg = sum(qual_vals) / len(qual_vals) if qual_vals else None
  381. if rel_avg is not None and qual_avg is not None:
  382. overall = round((rel_avg + qual_avg) / 2, 1)
  383. elif rel_avg is not None:
  384. overall = round(rel_avg, 1)
  385. elif qual_avg is not None:
  386. overall = round(qual_avg, 1)
  387. else:
  388. overall = 0.0
  389. else:
  390. overall = round(sum(fs.values()) / len(fs), 1) if fs else 0
  391. anomaly = bool(e.get("error")) or not fs
  392. grade = p.get("_quality_grade", "")
  393. fb = r.get("found_by_queries", [])
  394. # 4. 解析 制作相关性 (production_relevance)
  395. if is_mod_schema:
  396. # 新版使用 "相关性" 中的 "和内容制作知识相关" 代表制作相关性
  397. production_relevance = fs.get("relevance_production")
  398. else:
  399. if is_new_schema:
  400. pr_block = e.get("制作相关性") or {}
  401. pr_raw = pr_block.get("得分") if isinstance(pr_block, dict) else pr_block
  402. if isinstance(pr_block, dict) and "理由" in pr_block:
  403. score_reasons["production_relevance"] = pr_block["理由"]
  404. else:
  405. pr_raw = e.get("production_relevance")
  406. try: production_relevance = int(float(pr_raw)) if pr_raw is not None else None
  407. except (TypeError, ValueError): production_relevance = None
  408. recency_hard = _recency_hard(p.get("publish_timestamp", ""))
  409. # 5. 解析 判定决策 (decision) 和 理由 (reason)
  410. reason = e.get("判定理由") or e.get("reason") or ""
  411. # 根据过滤指标决定是否保留 (过滤指标判定逻辑优先,不依赖文字匹配)
  412. is_discard = False
  413. # 制作相关性低于阈值则丢弃 (新版 0-10 满分,因此低于 4 丢弃;老版低于 2 丢弃)
  414. if production_relevance is not None:
  415. threshold = 4 if is_mod_schema else 2
  416. if production_relevance < threshold:
  417. is_discard = True
  418. # 时效性低于 2 被丢弃(发布时间超两年的老帖)
  419. if recency_hard is not None and recency_hard < 2:
  420. is_discard = True
  421. # 综合均分低于阈值被丢弃 (新版低于 6 丢弃;老版低于 3 丢弃)
  422. if overall is not None:
  423. threshold_ov = 6 if is_mod_schema else 3
  424. if overall < threshold_ov:
  425. is_discard = True
  426. decision = "discard" if is_discard else "report"
  427. # Find matching procedure html
  428. procedure_html = None
  429. case_id = r.get("case_id", "")
  430. title = p.get("title", "")
  431. run_dir = HERE / "runs_full" / run
  432. if run_dir.is_dir():
  433. # 1. 优先扫描该帖子对应的文件夹下的任何 HTML 文件 (不限名称)
  434. # 文件夹名格式: {form}_{platform}_{channel_content_id[:8]}
  435. content_id = r.get("channel_content_id") or ""
  436. if not content_id and case_id and "_" in case_id:
  437. content_id = case_id.split("_", 1)[1]
  438. plat_key = r.get("platform") or ""
  439. if form_name and plat_key and content_id:
  440. folder_name = f"{form_name}_{plat_key}_{content_id[:8]}"
  441. case_dir = run_dir / "procedures" / folder_name
  442. if case_dir.is_dir():
  443. html_files = list(case_dir.glob("*.html"))
  444. if html_files:
  445. procedure_html = f"runs_full/{run}/procedures/{folder_name}/{html_files[0].name}"
  446. # 2. 其次匹配标准文件名: case-{case_id}.html 或 {case_id}.html
  447. candidate_dirs = [run_dir, run_dir / "procedures"]
  448. if not procedure_html and case_id:
  449. named_files = [f"case-{case_id}.html", f"{case_id}.html"]
  450. for d_dir in candidate_dirs:
  451. if d_dir.is_dir():
  452. for name in named_files:
  453. if (d_dir / name).is_file():
  454. procedure_html = f"runs_full/{run}/procedures/{name}" if d_dir.name == "procedures" else f"runs_full/{run}/{name}"
  455. break
  456. if procedure_html:
  457. break
  458. # 3. 再次匹配 HTML 内部的标准声明 (meta 标签或 HTML 注释)
  459. if not procedure_html and case_id:
  460. for d_dir in candidate_dirs:
  461. if d_dir.is_dir():
  462. for html_path in d_dir.glob("*.html"):
  463. try:
  464. content = html_path.read_text(encoding="utf-8")
  465. if f'name="case-id" content="{case_id}"' in content or \
  466. f'name="case_id" content="{case_id}"' in content or \
  467. f'<!-- case_id: {case_id} -->' in content or \
  468. f'<!-- case-id: {case_id} -->' in content:
  469. procedure_html = f"runs_full/{run}/procedures/{html_path.name}" if d_dir.name == "procedures" else f"runs_full/{run}/{html_path.name}"
  470. break
  471. except Exception:
  472. continue
  473. if procedure_html:
  474. break
  475. # 4. 最后使用标题作为兜底模糊匹配
  476. if not procedure_html and title:
  477. for d_dir in candidate_dirs:
  478. if d_dir.is_dir():
  479. for html_path in d_dir.glob("*.html"):
  480. try:
  481. content = html_path.read_text(encoding="utf-8")
  482. if title in content:
  483. procedure_html = f"runs_full/{run}/procedures/{html_path.name}" if d_dir.name == "procedures" else f"runs_full/{run}/{html_path.name}"
  484. break
  485. except Exception:
  486. continue
  487. if procedure_html:
  488. break
  489. return {
  490. "case_id": r.get("case_id", ""),
  491. "platform": PLAT.get(r.get("platform"), r.get("platform")), "platformKey": r.get("platform"),
  492. "title": p.get("title", "") or "(无标题)", "date": (p.get("publish_timestamp", "") or "")[:10],
  493. "url": r.get("source_url", ""), "engagement": f'{p.get("like_count", 0)} 赞',
  494. "knowledge_type": kt, "decision": decision,
  495. "tools": [KT.get(k, k) for k in kt] + ([f"质量 {grade}"] if grade else []), "found_by": fb,
  496. "images": (p.get("images") or [])[:6], "text": p.get("body_text", "") or "",
  497. "scores": fs, "overall": overall, "reason": reason, "score_reasons": score_reasons,
  498. "grade": grade, "qscore": p.get("_quality_score", 0), "anomaly": anomaly,
  499. "production_relevance": production_relevance, "recency_hard": recency_hard,
  500. "run": run, "procedure_html": procedure_html,
  501. }
  502. def scan_runs():
  503. runs = {}
  504. for f in sorted(glob.glob(str(HERE / "runs_full" / "*" / "form_*.json"))):
  505. try:
  506. d = json.load(open(f, encoding="utf-8"))
  507. except Exception:
  508. continue
  509. run = pathlib.Path(f).parent.name
  510. form_name = d.get("form") or ""
  511. results = [adapt(r, run, form_name) for r in d.get("results", [])]
  512. report_val = sum(1 for r in results if r.get("decision") == "report" and not r.get("anomaly"))
  513. discard_val = sum(1 for r in results if r.get("decision") == "discard" and not r.get("anomaly"))
  514. runs.setdefault(run, []).append({
  515. "form": d.get("form"), "query": d.get("query"), "original_q": d.get("original_q", ""),
  516. "requirement": d.get("requirement", ""),
  517. "platforms": d.get("platforms", []), "total": d.get("total"),
  518. "report": report_val, "discard": discard_val,
  519. "results": results,
  520. })
  521. # 数据库回退:仅当本地 runs_full 完全为空时才读库(本地清空后界面仍有数据)。
  522. # 本地有任何数据就信本地,避免每次 /api/data 都查远程 DB 拖慢。
  523. try:
  524. import db
  525. for q, g in (db.fetch_posts_grouped().items() if not runs else []):
  526. if q in runs:
  527. continue
  528. results = [adapt(r, q, "A") for r in g["results"]]
  529. runs[q] = [{
  530. "form": "A", "query": g.get("query_text"), "original_q": g.get("query_text") or "",
  531. "requirement": "", "platforms": [], "total": len(results),
  532. "report": sum(1 for r in results if r.get("decision") == "report" and not r.get("anomaly")),
  533. "discard": sum(1 for r in results if r.get("decision") == "discard" and not r.get("anomaly")),
  534. "results": results, "_from_db": True,
  535. }]
  536. except Exception as e:
  537. print(f"⚠️ DB 回退读取失败(仅用本地数据):{e}")
  538. for v in runs.values():
  539. v.sort(key=lambda x: x.get("form") or "")
  540. def _qnum(name): # "q156" → 156,按数字排,避免 "q156" < "q99" 的字符串误排
  541. m = re.search(r"\d+", name)
  542. return (int(m.group()) if m else 0, name)
  543. out = []
  544. for k, v in sorted(runs.items(), key=lambda kv: _qnum(kv[0])):
  545. oq = v[0].get("original_q") or v[0].get("query") or ""
  546. seen, hits = set(), 0 # 知识命中数 = 各形式采纳(report)且非异常、按 url 去重后的帖子数
  547. for f in v:
  548. for r in f.get("results", []):
  549. if r.get("decision") == "report" and not r.get("anomaly") and r.get("url") not in seen:
  550. seen.add(r.get("url")); hits += 1
  551. out.append({"key": k, "forms": v, "dims": parse_dims(oq), "original_q": oq,
  552. "hits": hits, "tot": sum((f.get("total") or 0) for f in v)})
  553. active_reevals = {k: v["status"] for k, v in ACTIVE_REEVALS.items()}
  554. return {"queries": out, "actions": ACTIONS_TAX, "types": TYPES_TAX, "matrix": _MATRIX, "active_reevals": active_reevals}
  555. class H(BaseHTTPRequestHandler):
  556. def _send(self, code, body, ctype):
  557. b = body.encode("utf-8") if isinstance(body, str) else body
  558. self.send_response(code)
  559. if ctype.startswith("text/") or ctype == "application/json" or ctype == "application/javascript":
  560. self.send_header("Content-Type", ctype + "; charset=utf-8")
  561. else:
  562. self.send_header("Content-Type", ctype)
  563. self.send_header("Content-Length", str(len(b))); self.end_headers(); self.wfile.write(b)
  564. def do_GET(self):
  565. parsed = urlparse(self.path)
  566. path = parsed.path
  567. params = parse_qs(parsed.query)
  568. if path in ("/", "/index.html"):
  569. try:
  570. page = (HERE / "index.html").read_text(encoding="utf-8")
  571. self._send(200, page, "text/html")
  572. except Exception as e:
  573. self._send(500, f"Error reading index.html: {e}", "text/plain")
  574. elif path == "/api/data":
  575. self._send(200, json.dumps(scan_runs(), ensure_ascii=False), "application/json")
  576. elif path == "/api/tools_status":
  577. # 工具解构状态:done=结果文件已存在;running=该 q 解构任务仍在跑
  578. q = (params.get("q") or [""])[0].strip()
  579. case_id = (params.get("case_id") or [""])[0].strip()
  580. if not q or not case_id:
  581. self._send(400, "missing q or case_id", "text/plain"); return
  582. done = (HERE / "runs_full" / q / "tools" / f"{case_id}.json").is_file()
  583. if not done: # 本地无 → 看库里有没有(本地清空后仍算已解构)
  584. try:
  585. import db
  586. done = db.has_tools(q, case_id)
  587. except Exception:
  588. pass
  589. task = TOOL_TASKS.get(q) or {}
  590. running = task.get("status") == "running" and case_id in (task.get("case_ids") or [])
  591. self._send(200, json.dumps({
  592. "done": done, "running": running, "error": task.get("error"),
  593. }, ensure_ascii=False), "application/json")
  594. elif path == "/api/tools_data":
  595. # 取某帖的工具解构结果。可选 version:指定则取该版本(只从库),否则取最新(本地优先)。
  596. # 始终附带 versions(全部历史版本,供前端下拉切换)。
  597. q = (params.get("q") or [""])[0].strip()
  598. case_id = (params.get("case_id") or [""])[0].strip()
  599. version = (params.get("version") or [""])[0].strip() or None
  600. if not q or not case_id:
  601. self._send(400, "missing q or case_id", "text/plain"); return
  602. try:
  603. import db
  604. versions = db.fetch_tool_versions(q, case_id)
  605. except Exception:
  606. db, versions = None, []
  607. def _emit(data):
  608. if data:
  609. data["exists"] = True
  610. data["versions"] = versions
  611. self._send(200, json.dumps(data, ensure_ascii=False), "application/json")
  612. else:
  613. self._send(200, json.dumps({"exists": False, "versions": versions}, ensure_ascii=False), "application/json")
  614. if version: # 指定历史版本 → 只能从库取
  615. _emit(db.fetch_tools(q, case_id, version) if db else None); return
  616. f = HERE / "runs_full" / q / "tools" / f"{case_id}.json" # 默认最新:本地优先
  617. if f.is_file():
  618. try:
  619. _emit(json.loads(f.read_text(encoding="utf-8"))); return
  620. except Exception as e:
  621. self._send(500, json.dumps({"error": f"read failed: {e}"}, ensure_ascii=False), "application/json"); return
  622. _emit(db.fetch_tools(q, case_id) if db else None) # 本地无 → 库最新
  623. elif path == "/api/procedure_status":
  624. q = (params.get("q") or [""])[0].strip()
  625. form = (params.get("form") or [""])[0].strip()
  626. case_id = (params.get("case_id") or [""])[0].strip()
  627. if not q or not form or not case_id:
  628. self._send(400, "missing q, form, or case_id", "text/plain")
  629. return
  630. folder_name = f"{form}_{_short_case(case_id)}"
  631. task_key = f"{q}/{folder_name}"
  632. if task_key in ACTIVE_TASKS:
  633. task = ACTIVE_TASKS[task_key]
  634. res = {
  635. "status": task["status"],
  636. "error": task["error"]
  637. }
  638. if task["status"] == "success":
  639. out_dir = HERE / "runs_full" / q / "procedures" / folder_name
  640. html_files = list(out_dir.glob("*.html")) if out_dir.is_dir() else []
  641. if html_files:
  642. res["procedure_html"] = f"runs_full/{q}/procedures/{folder_name}/{html_files[0].name}"
  643. self._send(200, json.dumps(res, ensure_ascii=False), "application/json")
  644. return
  645. out_dir = HERE / "runs_full" / q / "procedures" / folder_name
  646. html_files = list(out_dir.glob("*.html")) if out_dir.is_dir() else []
  647. if html_files:
  648. self._send(200, json.dumps({
  649. "status": "success",
  650. "procedure_html": f"runs_full/{q}/procedures/{folder_name}/{html_files[0].name}"
  651. }, ensure_ascii=False), "application/json")
  652. return
  653. log_path = out_dir / "_extract.log"
  654. if log_path.is_file():
  655. self._send(200, json.dumps({"status": "failed", "error": "Not running, but no HTML output found (possibly crashed)."}, ensure_ascii=False), "application/json")
  656. return
  657. self._send(200, json.dumps({"status": "not_started"}, ensure_ascii=False), "application/json")
  658. elif path == "/api/procedure_log":
  659. q = (params.get("q") or [""])[0].strip()
  660. form = (params.get("form") or [""])[0].strip()
  661. case_id = (params.get("case_id") or [""])[0].strip()
  662. if not q or not form or not case_id:
  663. self._send(400, "missing q, form, or case_id", "text/plain")
  664. return
  665. folder_name = f"{form}_{_short_case(case_id)}"
  666. log_path = HERE / "runs_full" / q / "procedures" / folder_name / "_extract.log"
  667. if not log_path.is_file():
  668. self._send(200, json.dumps({"log": ""}, ensure_ascii=False), "application/json")
  669. return
  670. try:
  671. content = log_path.read_text(encoding="utf-8", errors="replace")
  672. self._send(200, json.dumps({"log": content}, ensure_ascii=False), "application/json")
  673. except Exception as e:
  674. self._send(500, json.dumps({"error": str(e)}, ensure_ascii=False), "application/json")
  675. elif path == "/api/spec_content":
  676. file_name = (params.get("file") or [""])[0].strip()
  677. allowed = [
  678. "README.md",
  679. "tools.md",
  680. "extraction/phase1-skeleton.md",
  681. "extraction/phase2-normalize.md",
  682. "extraction/phase3-finalize.md",
  683. "taxonomy/type_suggestions.md"
  684. ]
  685. if file_name not in allowed:
  686. self._send(400, "invalid file parameter", "text/plain")
  687. return
  688. target_path = HERE / "procedure-dsl" / "spec" / file_name
  689. if not target_path.is_file():
  690. self._send(404, "spec file not found", "text/plain")
  691. return
  692. try:
  693. content = target_path.read_text(encoding="utf-8", errors="replace")
  694. self._send(200, json.dumps({"content": content}, ensure_ascii=False), "application/json")
  695. except Exception as e:
  696. self._send(500, json.dumps({"error": str(e)}, ensure_ascii=False), "application/json")
  697. elif path == "/api/reeval_status":
  698. q = (params.get("q") or [""])[0].strip()
  699. if not q:
  700. self._send(400, "missing q", "text/plain")
  701. return
  702. if q in ACTIVE_REEVALS:
  703. self._send(200, json.dumps({
  704. "status": ACTIVE_REEVALS[q]["status"],
  705. "error": ACTIVE_REEVALS[q].get("error")
  706. }, ensure_ascii=False), "application/json")
  707. else:
  708. self._send(200, json.dumps({"status": "not_started"}, ensure_ascii=False), "application/json")
  709. elif self.path.startswith("/runs_full/"):
  710. try:
  711. clean_path = self.path.split("?")[0]
  712. parts = clean_path.strip("/").split("/")
  713. target_file = HERE
  714. for part in parts:
  715. target_file = target_file / part
  716. runs_dir = HERE / "runs_full"
  717. if runs_dir.resolve() in target_file.resolve().parents and target_file.is_file():
  718. content = target_file.read_bytes()
  719. ext = target_file.suffix.lower()
  720. ctype = "text/html"
  721. if ext in (".png", ".webp"):
  722. ctype = f"image/{ext[1:]}"
  723. elif ext in (".jpg", ".jpeg"):
  724. ctype = "image/jpeg"
  725. elif ext == ".json":
  726. ctype = "application/json"
  727. elif ext == ".js":
  728. ctype = "application/javascript"
  729. elif ext == ".css":
  730. ctype = "text/css"
  731. self._send(200, content, ctype)
  732. else:
  733. self._send(404, "not found", "text/plain")
  734. except Exception as e:
  735. self._send(500, f"Error: {e}", "text/plain")
  736. else:
  737. self._send(404, "not found", "text/plain")
  738. def do_POST(self):
  739. if self.path == "/api/generate_procedure":
  740. length = int(self.headers.get("Content-Length") or 0)
  741. raw = self.rfile.read(length).decode("utf-8") if length > 0 else "{}"
  742. try:
  743. payload = json.loads(raw)
  744. except Exception as e:
  745. self._send(400, json.dumps({"error": f"bad json: {e}"}), "application/json"); return
  746. q = (payload.get("q") or "").strip()
  747. form = (payload.get("form") or "").strip()
  748. case_id = (payload.get("case_id") or "").strip()
  749. engine = (payload.get("engine") or "cyber_runner").strip()
  750. model = (payload.get("model") or "google/gemini-3.1-flash-lite").strip()
  751. if not re.match(r"^q\d+$", q):
  752. self._send(400, json.dumps({"error": f"bad q (expect 'qNN'): {q!r}"}, ensure_ascii=False), "application/json"); return
  753. if form not in ("A", "B", "C"):
  754. self._send(400, json.dumps({"error": f"bad form: {form!r}"}, ensure_ascii=False), "application/json"); return
  755. if not case_id:
  756. self._send(400, json.dumps({"error": "missing case_id"}, ensure_ascii=False), "application/json"); return
  757. q_dir = HERE / "runs_full" / q
  758. form_file = q_dir / f"form_{form}.json"
  759. if not form_file.is_file():
  760. self._send(404, json.dumps({"error": f"form file not found: {form_file.name}"}, ensure_ascii=False), "application/json"); return
  761. try:
  762. with open(form_file, encoding="utf-8") as f:
  763. form_data = json.load(f)
  764. except Exception as e:
  765. self._send(500, json.dumps({"error": f"failed to read form: {e}"}, ensure_ascii=False), "application/json"); return
  766. matching_result = None
  767. for r in form_data.get("results", []):
  768. if r.get("case_id") == case_id:
  769. matching_result = r
  770. break
  771. if not matching_result:
  772. self._send(404, json.dumps({"error": f"case_id {case_id} not found in form {form}"}, ensure_ascii=False), "application/json"); return
  773. folder_name = f"{form}_{_short_case(case_id)}"
  774. out_dir = q_dir / "procedures" / folder_name
  775. out_dir.mkdir(parents=True, exist_ok=True)
  776. src_path = out_dir / "_source.json"
  777. try:
  778. with open(src_path, "w", encoding="utf-8") as f:
  779. json.dump(_source_to_dsl_input(matching_result), f, ensure_ascii=False, indent=2)
  780. score = _composite_score(matching_result.get("llm_evaluation") or {})
  781. _write_meta(out_dir, case_id=case_id, from_q=q, form=form, score=score)
  782. except Exception as e:
  783. self._send(500, json.dumps({"error": f"failed to write inputs: {e}"}, ensure_ascii=False), "application/json"); return
  784. task_key = f"{q}/{folder_name}"
  785. ACTIVE_TASKS[task_key] = {
  786. "status": "running",
  787. "start_time": datetime.now().isoformat(),
  788. "pid": None,
  789. "error": None
  790. }
  791. t = threading.Thread(target=run_extraction_task, args=(q, folder_name, src_path, out_dir, engine, model))
  792. t.daemon = True
  793. t.start()
  794. self._send(200, json.dumps({
  795. "status": "started",
  796. "task_key": task_key,
  797. "log": f"runs_full/{q}/procedures/{folder_name}/_extract.log"
  798. }, ensure_ascii=False), "application/json")
  799. elif self.path == "/api/extract_tools":
  800. # 工具解构:body {q, case_ids:[...], force?} → 起 tool_extract.py 子进程
  801. length = int(self.headers.get("Content-Length") or 0)
  802. raw = self.rfile.read(length).decode("utf-8") if length > 0 else "{}"
  803. try:
  804. payload = json.loads(raw)
  805. except Exception as e:
  806. self._send(400, json.dumps({"error": f"bad json: {e}"}), "application/json"); return
  807. q = (payload.get("q") or "").strip()
  808. case_ids = payload.get("case_ids") or []
  809. force = bool(payload.get("force"))
  810. if not re.match(r"^q\d+$", q):
  811. self._send(400, json.dumps({"error": f"bad q (expect 'qNN'): {q!r}"}, ensure_ascii=False), "application/json"); return
  812. if not isinstance(case_ids, list) or not case_ids:
  813. self._send(400, json.dumps({"error": "case_ids must be a non-empty list"}, ensure_ascii=False), "application/json"); return
  814. if not (HERE / "runs_full" / q / "form_A.json").is_file():
  815. self._send(404, json.dumps({"error": f"runs_full/{q}/form_A.json not found"}, ensure_ascii=False), "application/json"); return
  816. TOOL_TASKS[q] = {
  817. "status": "running", "case_ids": case_ids,
  818. "start_time": datetime.now().isoformat(), "pid": None, "error": None,
  819. }
  820. t = threading.Thread(target=run_tool_extraction_task, args=(q, case_ids, force))
  821. t.daemon = True
  822. t.start()
  823. self._send(200, json.dumps({
  824. "status": "started", "q": q, "count": len(case_ids),
  825. "log": f"runs_full/{q}/tools/_extract.log",
  826. }, ensure_ascii=False), "application/json")
  827. elif self.path == "/api/reeval":
  828. length = int(self.headers.get("Content-Length") or 0)
  829. raw = self.rfile.read(length).decode("utf-8") if length > 0 else "{}"
  830. try:
  831. payload = json.loads(raw)
  832. except Exception as e:
  833. self._send(400, json.dumps({"error": f"bad json: {e}"}), "application/json"); return
  834. q = (payload.get("q") or "").strip()
  835. if not re.match(r"^q\d+$", q):
  836. self._send(400, json.dumps({"error": f"bad q (expect 'qNN'): {q!r}"},
  837. ensure_ascii=False), "application/json"); return
  838. q_dir = HERE / "runs_full" / q
  839. if not q_dir.is_dir():
  840. self._send(404, json.dumps({"error": f"runs_full/{q} not found"}, ensure_ascii=False),
  841. "application/json"); return
  842. log_path = q_dir / "_reeval.log"
  843. try:
  844. log_fh = open(log_path, "w", encoding="utf-8", buffering=1)
  845. cmd = [sys.executable, "-u", str(HERE / "batch_3forms.py"),
  846. "--reeval", "--reeval-q", q, "--output-dir", str(HERE / "runs_full")]
  847. flags = subprocess.CREATE_NEW_PROCESS_GROUP if sys.platform == "win32" else 0
  848. proc = subprocess.Popen(cmd, stdout=log_fh, stderr=subprocess.STDOUT,
  849. cwd=str(HERE), creationflags=flags)
  850. ACTIVE_REEVALS[q] = {
  851. "status": "running",
  852. "pid": proc.pid,
  853. "error": None
  854. }
  855. def wait_reeval(q_key, p_obj, fh):
  856. try:
  857. p_obj.wait()
  858. if p_obj.returncode == 0:
  859. ACTIVE_REEVALS[q_key]["status"] = "success"
  860. else:
  861. ACTIVE_REEVALS[q_key]["status"] = "failed"
  862. ACTIVE_REEVALS[q_key]["error"] = f"Subprocess exited with code {p_obj.returncode}"
  863. except Exception as ex:
  864. ACTIVE_REEVALS[q_key]["status"] = "failed"
  865. ACTIVE_REEVALS[q_key]["error"] = str(ex)
  866. finally:
  867. try:
  868. fh.close()
  869. except Exception:
  870. pass
  871. t = threading.Thread(target=wait_reeval, args=(q, proc, log_fh))
  872. t.daemon = True
  873. t.start()
  874. self._send(200, json.dumps(
  875. {"status": "started", "pid": proc.pid, "q": q,
  876. "log": str(log_path.relative_to(HERE))},
  877. ensure_ascii=False), "application/json")
  878. except Exception as e:
  879. self._send(500, json.dumps({"error": f"failed to start: {e}"},
  880. ensure_ascii=False), "application/json")
  881. elif self.path == "/api/save_spec":
  882. length = int(self.headers.get("Content-Length") or 0)
  883. raw = self.rfile.read(length).decode("utf-8") if length > 0 else "{}"
  884. try:
  885. payload = json.loads(raw)
  886. except Exception as e:
  887. self._send(400, json.dumps({"error": f"bad json: {e}"}), "application/json"); return
  888. file_name = (payload.get("file") or "").strip()
  889. content = payload.get("content") or ""
  890. allowed = [
  891. "README.md",
  892. "tools.md",
  893. "extraction/phase1-skeleton.md",
  894. "extraction/phase2-normalize.md",
  895. "extraction/phase3-finalize.md",
  896. "taxonomy/type_suggestions.md"
  897. ]
  898. if file_name not in allowed:
  899. self._send(400, json.dumps({"error": "invalid file parameter"}), "application/json"); return
  900. target_path = HERE / "procedure-dsl" / "spec" / file_name
  901. try:
  902. target_path.parent.mkdir(parents=True, exist_ok=True)
  903. target_path.write_text(content, encoding="utf-8")
  904. self._send(200, json.dumps({"status": "ok"}, ensure_ascii=False), "application/json")
  905. except Exception as e:
  906. self._send(500, json.dumps({"error": str(e)}, ensure_ascii=False), "application/json")
  907. else:
  908. self._send(404, json.dumps({"error": "not found"}), "application/json")
  909. def log_message(self, *a): pass
  910. if __name__ == "__main__":
  911. n = len(scan_runs()["queries"])
  912. print(f"搜索评估查看 server:http://0.0.0.0:{PORT} (runs_full/ 下 {n} 个 query,实时扫描)")
  913. ThreadingHTTPServer(("0.0.0.0", PORT), H).serve_forever()