#!/usr/bin/env python3.12
"""brand_hotspots_0.jpg 豆包重生成（霸王茶姬×米菲桂花联名；正文图全为竖版不可用）"""
import sys, os
sys.path.insert(0, '/data/news/scripts')
from PIL import Image
import image_relevance_guard as G

DST = '/data/news/images/0929/brand_hotspots_0.jpg'
TITLE = '霸王茶姬 × 米菲「酿」出秋天桂花香'
SUM = '霸王茶姬联名米菲推出秋季桂花新品，把季节原料做成IP联名话术。'
NOP = '无人空镜，画面中不得出现任何人物、人脸、人影与人体局部'

PROMPTS = [
    f'配图：{TITLE}，秋季桂花主题的联名奶茶新品与纸杯包装摆放在暖金色调桌面上，'
    f'桂花与秋叶点缀，新闻配图风格，写实，横向构图，16:9，高清，{NOP}',
    f'配图：桂花风味奶茶与联名限定纸杯、纸袋的组合静物，秋季暖色调，桂花枝元素，'
    f'新闻配图风格，写实，横向构图，16:9，高清，{NOP}',
    f'配图：秋季桂花联名饮品海报风格静物，奶茶杯与桂花花朵、暖色背景，'
    f'新闻配图风格，写实，横向构图，16:9，高清，{NOP}',
]
for ci, pr in enumerate(PROMPTS, 1):
    try:
        tmp = f'/tmp/regen0929_brand0_{ci}.jpg'
        open(tmp, 'wb').write(G.gen_image(pr))
        G.strip_watermark(Image.open(tmp).convert('RGB')).save(DST, 'JPEG', quality=92)
        o = Image.open(DST)
        ok, why, desc = G.parse_relevance(G.relevance_ask(DST, TITLE, SUM))
        print(f'候选{ci} {o.size} {os.path.getsize(DST)//1024}KB 判定={ok} {why} | {desc[:130]}', flush=True)
        if ok:
            print(f'✅ 候选{ci} 通过 → {DST}', flush=True); break
    except Exception as e:
        print(f'候选{ci} 异常: {str(e)[:90]}', flush=True)
else:
    print('⚠️ 未达「是」，候选留在 /tmp/regen0929_brand0_*.jpg 待人工判定', flush=True)
o = Image.open(DST); print(f'最终: {o.format} {o.size} {os.path.getsize(DST)//1024}KB')
