# -*- coding: utf-8 -*-
"""A/B 对比: 稳定版(VisionParser1) vs 交叉版(VisionParserCross) 同图实测
检查: 输出差异 / 耗时 / page_wrong一致性 / 调用次数(按日志行数)"""
import sys
import time
from pathlib import Path

ROOT = Path(__file__).parent
sys.path.insert(0, str(ROOT))

from ai_helper_vision1 import VisionParser1
from ai_helper_vision_cross import VisionParserCross
from PIL import Image

IMAGES = [
    (r"screenshots\OV7T8PV8GEHYSCNB\step3\step3_b0.png", "甲巯咪唑片"),
    (r"screenshots\OV7T8PV8GEHYSCNB\step3\step3_b1.png", "甲巯咪唑片"),
    (r"screenshots\OV7T8PV8GEHYSCNB\step3\step3_b2.png", "甲巯咪唑片"),
    (r"screenshots\T4VK4LM7AAUOV8AY\step3\step3_b0.png", "甲巯咪唑片"),
]

CROSS = VisionParserCross()
STABLE = VisionParser1()


def run(parser, img, kw, w, h):
    t0 = time.time()
    shops, status = parser.parse_shops(img, screen_size=(w, h), keyword=kw, crop_x=287)
    dt = time.time() - t0
    return shops, status, dt


def sig(shops):
    return sorted((s[0], str(price_num := __import__("re").search(r"[\d.]+", s[2]).group(0)), s[1][:18])
                  for s in shops)


for rel, kw in IMAGES:
    img = str(ROOT / rel)
    w, h = Image.open(img).size
    print("=" * 78)
    print(f"{rel.split(chr(92))[-2]}_b{Path(img).stem[-1]}  ({w}x{h})")

    s_shops, s_status, s_dt = run(STABLE, img, kw, w, h)
    print(f"  [稳定] {s_status} {len(s_shops)}卡 {s_dt:.0f}s")
    for s in s_shops:
        print(f"     {s[0][:14]} | {s[2]} | {s[1][:26]}")

    c_shops, c_status, c_dt = run(CROSS, img, kw, w, h)
    print(f"  [交叉] {c_status} {len(c_shops)}卡 {c_dt:.0f}s")
    for s in c_shops:
        print(f"     {s[0][:14]} | {s[2]} | {s[1][:26]}")

    same = (s_status == c_status) and (sig(s_shops) == sig(c_shops))
    print(f"  → {'两版结果一致' if same else '★ 两版结果不同(上面对比)'}  耗时比 {c_dt/s_dt:.1f}x")