feat: 新增作息报时器功能并替换短链接前缀为/to/
- 新增作息报时器独立模块，包含完整的日程管理、语音播报和待机功能 - 短链接正式前缀由/l/改为/to/，解决手机小屏下与I、i、1字形混淆问题 - 保留旧前缀/l/、/L/、/I/、/i/兼容已分发的旧短链 - 新增短码大小写兜底匹配，仅唯一匹配时生效避免歧义 - 新增防爆紧急替换功能，可轮换首页鉴权并替换短链接 - 新增多个调试和集成测试脚本，保障数据安全和功能回归
This commit is contained in:
yangxiangyuan
@@ -0,0 +1,151 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
# ============================================================
|
||||
# tool_qa_mall_logos.py - 宝龙美食打卡 lab · logo 裁切抽检（方案 C）
|
||||
#
|
||||
# 用视觉大模型 qwen-vl-plus-latest（Rule 11 视觉/多模态选型）对方案 A 的裁切结果做抽检：
|
||||
# 按行拼版（每行 11 格，格顶标注 id），连同 id->店名 清单一起发给模型，
|
||||
# 让它逐格检查：logo 是否被切边 / 串入邻格 / 带店名文字条 / 基本空白 / 与店名明显不符。
|
||||
# 模型只读抽检、不改图；结果落 report.json 供人工复核决定是否重裁。
|
||||
#
|
||||
# 凭据：从 ~/Toolbox_local_creds.env.local 读 DASHSCOPE_API_KEY / DASHSCOPE_API_HOST_FOR_OPENAI
|
||||
# （零硬编码，Rule 13）
|
||||
# 用法：python tool_qa_mall_logos.py
|
||||
# ============================================================
|
||||
import base64
|
||||
import json
|
||||
import os
|
||||
import sys
|
||||
import urllib.request
|
||||
|
||||
import cv2
|
||||
import numpy as np
|
||||
|
||||
HERE = os.path.dirname(os.path.abspath(__file__))
|
||||
IMAGES_DIR = r'd:\Trae_Files\TRAE-Toolbox\public\tools\thought_lab\labs\mall_food_checkin\images'
|
||||
SEED = r'd:\Trae_Files\TRAE-Toolbox\src\server\thought_lab\labs\mall_food_checkin\shops_seed.json'
|
||||
REPORT = r'd:\Trae_Files\TRAE-Toolbox\dev_test_scripts\debug\mall_qa_report.json'
|
||||
CREDS = os.path.join(os.path.expanduser('~'), 'Toolbox_local_creds.env.local')
|
||||
# 视觉模型：Rule 11 推荐 qwen-vl-plus-latest，但本 key 实测 403；
|
||||
# 与项目现网口径（plant_home/style_check/yuanzhupai）一致用 qwen-vl-plus，可用 --model 覆盖
|
||||
MODEL = 'qwen-vl-plus'
|
||||
ROWS, COLS = 7, 11
|
||||
|
||||
|
||||
def load_creds():
|
||||
env = {}
|
||||
try:
|
||||
for line in open(CREDS, encoding='utf-8'):
|
||||
line = line.strip()
|
||||
if not line or line.startswith('#') or '=' not in line:
|
||||
continue
|
||||
k, v = line.split('=', 1)
|
||||
env[k.strip()] = v.strip().strip('"').strip("'")
|
||||
except OSError:
|
||||
pass
|
||||
return env
|
||||
|
||||
|
||||
def imread_u(path):
|
||||
data = np.fromfile(path, dtype=np.uint8)
|
||||
return cv2.imdecode(data, cv2.IMREAD_COLOR)
|
||||
|
||||
|
||||
def imencode_b64(img):
|
||||
ok, buf = cv2.imencode('.jpg', img, [cv2.IMWRITE_JPEG_QUALITY, 85])
|
||||
if not ok:
|
||||
return ''
|
||||
return base64.b64encode(buf.tobytes()).decode('ascii')
|
||||
|
||||
|
||||
def build_row_sheet(ids, crops):
|
||||
cw, ch, lab_h = 400, 260, 26
|
||||
sheet = np.full((ch + lab_h, cw * len(ids), 3), 255, np.uint8)
|
||||
for i, sid in enumerate(ids):
|
||||
x = i * cw
|
||||
cv2.putText(sheet, sid, (x + 8, 19), cv2.FONT_HERSHEY_SIMPLEX, 0.65, (0, 0, 0), 2)
|
||||
c = crops.get(sid)
|
||||
if c is None:
|
||||
cv2.putText(sheet, 'MISS', (x + 150, 150), cv2.FONT_HERSHEY_SIMPLEX, 1.0, (0, 0, 255), 3)
|
||||
continue
|
||||
sc = min((cw - 10) / float(c.shape[1]), (ch - 10) / float(c.shape[0]))
|
||||
cc = cv2.resize(c, (max(1, int(c.shape[1] * sc)), max(1, int(c.shape[0] * sc))))
|
||||
sheet[lab_h:lab_h + cc.shape[0], x + 5:x + 5 + cc.shape[1]] = cc
|
||||
return sheet
|
||||
|
||||
|
||||
def ask_vl(api_base, api_key, sheet, names_text):
|
||||
body = {
|
||||
'model': MODEL,
|
||||
'messages': [{
|
||||
'role': 'user',
|
||||
'content': [
|
||||
{'type': 'text', 'text': (
|
||||
'图中是若干张饭店 logo 裁切图，从左到右每格顶部标注了 id。'
|
||||
'id 与店名对照：' + names_text + '。'
|
||||
'判定口径（严格，宁漏报勿误报）：允许轻微白边、允许画面偏暗或过曝、允许构图不完美；'
|
||||
'只有以下明显缺陷才报告：cut(logo 主体被裁掉超过 1/4) / neighbor(明显串入邻格的另一张图) / '
|
||||
'label(底部带白底黑字店名文字条) / blank(整格基本是空白板面) / mismatch(画面内容与店名完全对不上，'
|
||||
'例如咖啡店格子里是火锅)。拿不准的一律视为合格。'
|
||||
'只输出 JSON 数组，元素形如 {"id":"r0c1","problem":"cut"}；全部合格则输出 []。'
|
||||
'不要输出任何其他文字。')},
|
||||
{'type': 'image_url', 'image_url': {'url': 'data:image/jpeg;base64,' + imencode_b64(sheet)}}
|
||||
]
|
||||
}],
|
||||
'temperature': 0.1
|
||||
}
|
||||
req = urllib.request.Request(
|
||||
api_base.rstrip('/') + '/chat/completions',
|
||||
data=json.dumps(body).encode('utf-8'),
|
||||
headers={'Content-Type': 'application/json', 'Authorization': 'Bearer ' + api_key},
|
||||
method='POST')
|
||||
with urllib.request.urlopen(req, timeout=120) as r:
|
||||
data = json.loads(r.read().decode('utf-8'))
|
||||
text = data['choices'][0]['message']['content']
|
||||
text = text.strip()
|
||||
if text.startswith('```'):
|
||||
text = text.strip('`')
|
||||
if text.startswith('json'):
|
||||
text = text[4:]
|
||||
return json.loads(text.strip())
|
||||
|
||||
|
||||
def main():
|
||||
global MODEL
|
||||
if '--model' in sys.argv:
|
||||
MODEL = sys.argv[sys.argv.index('--model') + 1]
|
||||
creds = load_creds()
|
||||
api_key = creds.get('DASHSCOPE_API_KEY', '') or os.environ.get('DASHSCOPE_API_KEY', '')
|
||||
api_base = creds.get('DASHSCOPE_API_HOST_FOR_OPENAI', '') or os.environ.get('DASHSCOPE_API_HOST_FOR_OPENAI', '')
|
||||
if not api_key or not api_base:
|
||||
print('缺少 DASHSCOPE_API_KEY / DASHSCOPE_API_HOST_FOR_OPENAI（检查 ~/Toolbox_local_creds.env.local）')
|
||||
sys.exit(1)
|
||||
|
||||
seed = json.load(open(SEED, encoding='utf-8'))
|
||||
name_of = {s['id']: s['name'] for s in seed['shops']}
|
||||
crops = {}
|
||||
for sid in name_of:
|
||||
p = os.path.join(IMAGES_DIR, sid + '.jpg')
|
||||
if os.path.exists(p):
|
||||
crops[sid] = imread_u(p)
|
||||
|
||||
all_ids = ['r%dc%d' % (ri, ci) for ri in range(ROWS) for ci in range(COLS)]
|
||||
report = {'model': MODEL, 'batches': []}
|
||||
for bi in range(0, len(all_ids), 4):
|
||||
ids = all_ids[bi:bi + 4]
|
||||
names_text = '、'.join('%s=%s' % (i, name_of.get(i, '?')) for i in ids)
|
||||
sheet = build_row_sheet(ids, crops)
|
||||
try:
|
||||
bad = ask_vl(api_base, api_key, sheet, names_text)
|
||||
except Exception as e: # noqa: BLE001
|
||||
print('%s 抽检失败: %s' % (ids[0], e))
|
||||
report['batches'].append({'ids': ids, 'error': str(e)})
|
||||
continue
|
||||
print('%s -> %s' % (ids[0], json.dumps(bad, ensure_ascii=False)))
|
||||
report['batches'].append({'ids': ids, 'bad': bad})
|
||||
json.dump(report, open(REPORT, 'w', encoding='utf-8'), ensure_ascii=False, indent=1)
|
||||
total = sum(len(b.get('bad', [])) for b in report['batches'] if isinstance(b.get('bad'), list))
|
||||
print('抽检完成，问题格合计 %d，报告: %s' % (total, REPORT))
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
Reference in New Issue
Block a user