# -*- coding: utf-8 -*-
import json, ssl, pathlib, urllib.request, urllib.error

BASE = "https://192.168.31.88:8042"
CFG = pathlib.Path(r"C:\Users\admin\.zcode\cli\config.json")
pat = json.loads(CFG.read_text(encoding="utf-8-sig"))["mcp"]["servers"]["sorb"]["env"]["SORB_PAT"]
ctx = ssl._create_unverified_context()

PROMPT = """【锁死吕布与赤兔】吕布必须100%按参考图1：黑色高束发髻、金色凤翅束发冠、两侧双根鲜红超长翎羽；面部与参考图1完全一致——剑眉浓黑上扬、眼窝深、眼神锐利、鼻梁高挺、下颌线清晰、年轻英武；兽面吞头连环铠（双肩兽头吞甲、胸前双圆护胸铜片、黑色护腕、腰部干净高级灰皮毛裙边）、红色长披风；兵器方天画戟按参考图2：月牙双刃+中央枪尖+戟尾配重、红色戟缨缠绕杆身。赤兔必须100%按参考图3：红鬃、深枣红马身、黑色皮质马鞍+金色鞍饰+红色缨穗+完整鞍布，毛色与马具形制不得改动。参考图只锁造型，机位与构图重新设计。
【方向与构图锁死】赤兔必须**马头在画面左侧、马尾在画面右侧**，吕布与赤兔都面朝画面左前方；吕布位于画面中央偏右，赤兔完整马身横贯画面中下部。红披风与双红翎向**画面右后方**飘扬。城门只在画面**右后方**远处，辽阔野外在画面左前方。禁止把马头放到画面右侧，禁止人物或马朝右。
【镜头】超低角度仰拍英雄全貌：完整吕布骑马全身与赤兔完整马身入画，赤兔四蹄稳稳踏在薄雪上，四腿数量正确、关节自然，禁止畸变。方天画戟从画面右下前景斜贯向上、戟尖刺向画面左上天空；红披风向画面右侧后方翻卷，双红翎向右后方猎猎。清晨低角度侧逆光自画面左后方射来，铠甲边缘、红翎、披风、马鬃勾出金色轮廓光；前景虚化雪尘粒子。背景：右后方远处关隘与城墙、左前方辽阔雪原与远山，冷青蓝晨雾。电影级调色：冷青阴影+暖金高光，胶片颗粒，边缘暗角，浅景深，16:9，无文字无水印无旗帜。"""

payload = {
    "client_namespace": "lubu_pv_board",
    "idempotency_key": "lubu_s2_full_reveal_20261027_v2_left",
    "project_id": 30,
    "model_config_id": 62,
    "task_type": "image",
    "references": [
        {"source_type": "media_asset", "id": 9089, "asset_type": "image", "role": "reference_image"},
        {"source_type": "media_asset", "id": 9057, "asset_type": "image", "role": "reference_image"},
        {"source_type": "media_asset", "id": 9047, "asset_type": "image", "role": "reference_image"},
        {"source_type": "media_asset", "id": 9048, "asset_type": "image", "role": "reference_image"},
    ],
    "parameters": {"prompt": PROMPT, "aspect_ratio": "16:9", "image_size": "2K",
                   "output_format": "png", "watermark": False},
}
data = json.dumps(payload, ensure_ascii=False).encode("utf-8")
req = urllib.request.Request(BASE + "/api/external/v1/tasks", data=data, method="POST",
                             headers={"Authorization": "Bearer " + pat, "Content-Type": "application/json"})
try:
    with urllib.request.urlopen(req, context=ctx, timeout=120) as r:
        o = json.loads(r.read().decode("utf-8"))
    print("task_id=", o.get("task", {}).get("id"), "status=", o.get("task", {}).get("status"))
except urllib.error.HTTPError as e:
    print("HTTP", e.code, e.read().decode("utf-8")[:800])