#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
路径C立面组装：apiyi 三图融合（风格 + 外立面 + 窗户字汇表）→ 指定立面。
在 Python 内部用绝对路径调 cmd_fuse，避开 Windows shell 传中文路径给 argv 的 GBK 乱码坑。
注意：不在这里重定向 stdout/stderr —— image_gen.py 已做，重复包装会关闭底层 buffer。

用法（英文 key，避开中文 argv）:
  python _run_pathc_fuse.py           # 默认 south
  python _run_pathc_fuse.py south     # 南立面
  python _run_pathc_fuse.py east      # 东立面
  python _run_pathc_fuse.py west      # 西立面
"""
import sys
from pathlib import Path

HERE = Path(__file__).resolve().parent            # tools/
RESEARCH = HERE.parent                             # ai-art-research/
sys.path.insert(0, str(HERE))
from image_gen import get_client, cmd_fuse

# 三张参考图（南/东/西立面通用）
STYLE = RESEARCH / "参考图" / "游戏截图" / "深夏时光" / "深夏时光-下午-街景.png"
CONTENT = [
    RESEARCH / "参考图" / "真实场景" / "东井村" / "QQ20260616-174350.png",
    RESEARCH / "生成结果" / "窗户标准件" / "标准件合集图.png",
]

# 英文 key → (prompt 文件名, 输出子目录)
ELEVATIONS = {
    "south": ("居民楼-单元南立面-路径C-窗户字汇表-v2.txt", "路径C组装"),
    "north": ("居民楼-单元北立面-路径C-窗户字汇表-v1.txt", "路径C北立面"),
    "east":  ("居民楼-单元东立面-路径C-窗户字汇表-v1.txt", "路径C东立面"),
    "west":  ("居民楼-单元西立面-路径C-窗户字汇表-v1.txt", "路径C西立面"),
}

key = sys.argv[1] if len(sys.argv) > 1 else "south"
if key not in ELEVATIONS:
    print(f"❌ 未知立面 key: {key}  (可用: {' '.join(ELEVATIONS)})")
    sys.exit(1)
PROMPT_NAME, SUBDIR = ELEVATIONS[key]
PROMPT_FILE = RESEARCH / "prompts" / PROMPT_NAME

print(f"=== 立面 [{key}]  prompt: {PROMPT_NAME}  →  生成结果/{SUBDIR}/ ===")

for p in [STYLE] + CONTENT + [PROMPT_FILE]:
    if not p.exists():
        print(f"❌ 缺失: {p}")
        sys.exit(1)

prompt = PROMPT_FILE.read_text(encoding="utf-8")
client = get_client()

# cmd_fuse 上传顺序 = [style, *content]，与 prompt 里 图1=风格/图2=外立面/图3=窗户表 一致
cmd_fuse(
    client,
    style_path=str(STYLE),
    content_paths=[str(c) for c in CONTENT],
    prompt=prompt,
    model=None,
    subdir=SUBDIR,
)