# -*- coding: utf-8 -*-
"""原创离线合成样例；不连接模型、不读取账户。输出位于脚本旁。
已有输入保留，修改输入后可以重跑。失败边界由 --negative 单独演示。
"""
from pathlib import Path
import sys, json, csv, math, io, hashlib, html, base64, re
sys.stdout.reconfigure(encoding='utf-8')
sys.stderr.reconfigure(encoding='utf-8')
BASE = Path(__file__).resolve().parent
NEG = '--negative' in sys.argv
def write(name, value):
    p=BASE/name; p.parent.mkdir(parents=True,exist_ok=True)
    if isinstance(value,bytes): p.write_bytes(value)
    else: p.write_text(value,encoding='utf-8',newline='')
def seed(name,value):
    if not (BASE/name).exists(): write(name,value)
def jwrite(name,value): write(name,json.dumps(value,ensure_ascii=False,indent=2))
def jseed(name,value):
    if not (BASE/name).exists(): jwrite(name,value)
def load(name): return json.loads((BASE/name).read_text(encoding='utf-8'))
def table(name,heads,rows):
    f=io.StringIO(newline='');w=csv.writer(f);w.writerow(heads);w.writerows(rows);write(name,f.getvalue())
def page(name,title,body):
    write(name,'<!doctype html><html lang="zh-CN"><meta charset="utf-8"><meta name="viewport" content="width=device-width,initial-scale=1"><title>'+html.escape(title)+'</title><style>body{max-width:1000px;margin:2rem auto;padding:0 1rem;font:17px/1.7 system-ui}table{border-collapse:collapse}td,th{border:1px solid #aaa;padding:.5rem}pre{white-space:pre-wrap;overflow-wrap:anywhere}img,svg{max-width:100%;height:auto}a{color:#125caa}</style><h1>'+html.escape(title)+'</h1>'+body+'</html>')
def emit(data): print(json.dumps(data,ensure_ascii=False))

from PIL import Image,ImageDraw,ImageOps
if not (BASE/'source.png').exists():
    im=Image.new('RGBA',(96,64),'#d7e6f2');ImageDraw.Draw(im).rectangle((15,15,70,50),fill='#30679c');im.save(BASE/'source.png')
if not (BASE/'selection.png').exists():
    s=Image.new('L',(96,64),0);ImageDraw.Draw(s).ellipse((30,15,60,45),fill=255);s.save(BASE/'selection.png')
source=Image.open(BASE/'source.png').convert('RGBA');sel=Image.open(BASE/'selection.png').convert('L')
if NEG:sel=sel.resize((48,32))
if source.size!=sel.size:raise ValueError('源图和选区尺寸不一致')
mask=Image.new('RGBA',source.size,(0,0,0,255));mask.putalpha(ImageOps.invert(sel));mask.save(BASE/'mask.png');red=Image.new('RGBA',source.size,(255,30,20,0));red.putalpha(sel.point(lambda x:round(x*.45)));Image.alpha_composite(source,red).save(BASE/'overlay.png')
jwrite('request-shape.json',{'note':'只示意JSON引用路径，不发送；先上传真实文件并替换占位ID与模型','model':'REPLACE_WITH_SUPPORTED_IMAGE_MODEL','images':[{'file_id':'REPLACE_SOURCE_FILE_ID'}],'mask':{'file_id':'REPLACE_MASK_FILE_ID'},'prompt':'修改红色预览标出的区域；保留其余主要结构'})
emit({'size':list(mask.size),'center_alpha':mask.getpixel((45,30))[3],'corner_alpha':mask.getpixel((0,0))[3],'selection_convention':'white selected -> transparent alpha'})
