# -*- coding: utf-8 -*-
"""Spot-check screenshot files with doubao vision: is this a real procurement page?"""
import base64, json, sys, requests
CFG = json.load(open('/data/www/files/LhyjdesignPro/config.json'))
KEY, BASE = CFG['api_key'], CFG['base_url']
Q = ("这是一张网页截图。请回答：(1) 这是哪个网站的页面（页眉/页脚可见的站点名称）？"
     "(2) 页面标题（项目名称）是什么？(3) 是否出现'预算金额'或金额数字？(4) 是否出现发布日期？"
     "(5) 页面是否为空白页/验证码页/登录页？请如实回答，不要猜测。")
for p in sys.argv[1:]:
    b = base64.b64encode(open(p, 'rb').read()).decode()
    msgs = [{"role": "user", "content": [
        {"type": "image_url", "image_url": {"url": "data:image/png;base64,%s" % b}},
        {"type": "text", "text": Q}]}]
    r = requests.post("%s/chat/completions" % BASE,
        headers={"Authorization": "Bearer %s" % KEY, "Content-Type": "application/json"},
        json={"model": CFG['models']['text_vision'], "messages": msgs, "max_tokens": 700},
        timeout=180).json()
    print("=" * 24, p.split('/')[-1])
    print(r.get('choices', [{}])[0].get('message', {}).get('content',
          json.dumps(r, ensure_ascii=False))[:900])
    sys.stdout.flush()