import json, urllib.request, base64

KEY = "sk-aRNj3UwKSLPsDfh15QNTPwbHxahblfaO"
BASE = "https://token.sensenova.cn/v1"
MODEL = "sensenova-6.7-flash-lite"

img_b64 = base64.b64encode(open('/tmp/test_shot.jpg', 'rb').read()).decode()
payload = json.dumps({
    "model": MODEL,
    "messages": [{
        "role": "user",
        "content": [
            {"type": "image_url", "image_url": {"url": f"data:image/jpeg;base64,{img_b64}"}},
            {"type": "text", "text": "请识别这张图片中的所有文字内容，包括数字、股票名称、金额、日期。用中文回复。"}
        ]
    }],
    "max_tokens": 1500,
}).encode()
req = urllib.request.Request(f"{BASE}/chat/completions", data=payload,
    headers={"Content-Type": "application/json", "Authorization": f"Bearer {KEY}"})
try:
    resp = urllib.request.urlopen(req, timeout=90)
    data = json.loads(resp.read().decode())
    msg = data.get('choices', [{}])[0].get('message', {})
    text = msg.get('content', '') or msg.get('reasoning', '')
    print('OCR OK:')
    print(text[:1500])
except Exception as e:
    print('OCR FAIL:', e)