#!/usr/bin/env python3
"""CHAR-004 R3 候选质检 · Qwen-VL · 反乞丐滑坡版标准"""
import sys, os, json, base64, subprocess

sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
from secrets_loader import secret, endpoint

API_KEY = secret("SC-004")
EP = endpoint("EPT-002")
FALLBACK_EP = "https://dashscope.aliyuncs.com/api/v1/services/aigc/multimodal-generation/generation"

CANDIDATES_DIR = os.path.expanduser("~/guanghulab/video-ai-system/assets/candidates/D157-VA-05-001-R3/CHAR-004-ZhugeFeng")
CANDIDATES = [f"{CANDIDATES_DIR}/CHAR-004-R3-candidate-{i:02d}-1080x1920.jpg" for i in range(1, 5)]

QC_PROMPT = """你是动画资产质量检查员。检查这张CHAR-004诸葛风立绘候选图。规范要求：清贫但整洁、有骨气、不是乞丐不是流民。服装干净，无破洞无补丁无污渍。

逐项检查并输出严格JSON：

1. 裤子膝盖有没有破洞/破损？(hole_in_pants: true/false)
2. 衣服下摆有没有毛边/撕裂？(frayed_hem: true/false)
3. 鞋子干不干净？有泥污没？(dirty_shoes: true/false)
4. 衣服有没有污渍？(stains: true/false)
5. 有没有补丁？(patches: true/false)
6. 整体气质像不像乞丐/流浪汉？(beggar_look: true/false)
7. 五官清晰无崩坏吗？(face_intact: true/false)
8. 肢体比例正常吗？(anatomy_ok: true/false)
9. 画面清晰吗？(clear_image: true/false)
10. 全身从头到鞋完整吗？(full_body: true/false)
11. 和苏白CHAR-003的脸像不像？(similar_to_sb: true/false)

注意：
- "磨白"不等于"破洞"——裤子膝盖只有颜色变浅、没有破损开口，hole_in_pants就是false
- 鞋子没有明显泥污就是dirty_shoes=false
- 衣领轻微磨毛不等于毛边撕裂——frayed_hem特指明显破损/散开/撕裂

输出严格JSON，不要额外文字：
{"candidate":N,"pass":true/false,"score":0-100,"hole_in_pants":bool,"frayed_hem":bool,"dirty_shoes":bool,"stains":bool,"patches":bool,"beggar_look":bool,"face_intact":bool,"anatomy_ok":bool,"clear_image":bool,"full_body":bool,"similar_to_sb":bool,"summary":"一句话"}"""

def encode_image(path):
    with open(path, "rb") as f:
        b64 = base64.b64encode(f.read()).decode()
    ext = path.rsplit(".", 1)[-1].lower()
    mime = {"jpg": "jpeg", "jpeg": "jpeg", "png": "png"}.get(ext, "jpeg")
    return f"data:image/{mime};base64,{b64}"

def call_qwen(img):
    body = json.dumps({"model":"qwen-vl-max","input":{"messages":[{"role":"user","content":[{"image":img},{"text":QC_PROMPT}]}]}})
    for ep in [EP, FALLBACK_EP]:
        try:
            proc = subprocess.run(["curl","-s","-m","60","--noproxy","*","-X","POST",ep,"-H",f"Authorization: Bearer {API_KEY}","-H","Content-Type: application/json","--data-binary","@-"],input=body,capture_output=True,text=True,timeout=65)
            if not proc.stdout.strip(): continue
            resp = json.loads(proc.stdout)
            if "output" in resp:
                content = resp["output"]["choices"][0]["message"]["content"][0]["text"]
                if "```" in content:
                    content = content.split("```")[1]
                    if content.startswith("json"): content = content[4:]
                    content = content.split("```")[0]
                return json.loads(content.strip())
        except: continue
    return None

def main():
    print("CHAR-004 R3 QC · Qwen-VL")
    results = []
    for i, p in enumerate(CANDIDATES):
        print(f"[{i+1}/4]", end=" ", flush=True)
        qc = call_qwen(encode_image(p))
        if qc:
            s = qc.get("score","?")
            ok = qc.get("pass","?")
            h = qc.get("hole_in_pants","?")
            print(f"score={s} pass={ok} 破洞={h}")
            results.append(qc)
        else:
            print("✗")
            results.append({"candidate":i+1,"pass":False,"error":"api"})
    
    ok = sum(1 for r in results if r.get("pass"))
    print(f"\n通过: {ok}/4")
    
    for r in results:
        if not r.get("pass"):
            issues = []
            if r.get("hole_in_pants"): issues.append("破洞")
            if r.get("frayed_hem"): issues.append("毛边")
            if r.get("dirty_shoes"): issues.append("脏鞋")
            if r.get("stains"): issues.append("污渍")
            if r.get("beggar_look"): issues.append("乞丐化")
            print(f"  候选{r.get('candidate')} ✗: {', '.join(issues) if issues else '评分不足'} (score={r.get('score')})")
    
    report = {"spec":"VA-05-001-R3","asset":"CHAR-004","tool":"qwen-vl-max","results":results,"summary":{"passed":ok,"total":4}}
    with open(os.path.join(CANDIDATES_DIR,"QC-R3-BATCH.json"),"w") as f:
        json.dump(report,f,ensure_ascii=False,indent=2)
    print(f"\n报告: QC-R3-BATCH.json")
    return 0

if __name__=="__main__":
    sys.exit(main())