395 lines
13 KiB
Python
395 lines
13 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
"""
|
||
CHAR-HERO-DESIGN-PACKER
|
||
生成/整理苏白主角资产包，让他不再像路人甲。
|
||
|
||
功能:
|
||
1. 读取候选参考图
|
||
2. 用 Seedance/Kling 生成多视角变体
|
||
3. 输出完整资产包到 approved/ 目录
|
||
4. 更新 manifest.hdlp
|
||
|
||
用法:
|
||
python char-hero-design-packer.py --character CHAR-003-SuBai --generate-all
|
||
python char-hero-design-packer.py --character CHAR-003-SuBai --view front_half_body
|
||
"""
|
||
|
||
import os
|
||
import sys
|
||
import json
|
||
import argparse
|
||
import shutil
|
||
from pathlib import Path
|
||
|
||
# 添加项目根目录到路径
|
||
PROJECT_ROOT = Path(__file__).parent.parent.parent
|
||
sys.path.insert(0, str(PROJECT_ROOT / "engines"))
|
||
|
||
try:
|
||
from image_api_adapter import generate_image
|
||
except Exception as exc:
|
||
print(f"⚠️ 图片生成适配器未接通: {exc}")
|
||
print("💡 桥接需要 Node.js: image-api-bridge.js + image-api-adapter.js")
|
||
generate_image = None
|
||
|
||
try:
|
||
from hldp_prompt import expand_prompt
|
||
except Exception:
|
||
expand_prompt = None
|
||
|
||
|
||
class CharHeroDesignPacker:
|
||
"""主角资产包生成器"""
|
||
|
||
def __init__(self, character_id, assets_root=None):
|
||
self.character_id = character_id
|
||
self.assets_root = Path(assets_root or PROJECT_ROOT / "assets" / "characters" / character_id)
|
||
self.manifest_path = self.assets_root / "manifest.hdlp"
|
||
self.approved_dir = self.assets_root / "approved"
|
||
self.candidates_dir = self.assets_root / "candidates"
|
||
self.rejected_dir = self.assets_root / "rejected"
|
||
self.turnarounds_dir = self.assets_root / "turnarounds"
|
||
self.voice_dir = self.assets_root / "voice"
|
||
|
||
# 确保目录存在
|
||
for d in [self.approved_dir, self.candidates_dir, self.rejected_dir,
|
||
self.turnarounds_dir, self.voice_dir]:
|
||
d.mkdir(parents=True, exist_ok=True)
|
||
|
||
# 读取 manifest
|
||
self.manifest = self._read_manifest()
|
||
|
||
# 读取角色描述
|
||
self.character_desc = self._load_character_description()
|
||
|
||
def _read_manifest(self):
|
||
"""读取 manifest.hdlp"""
|
||
if not self.manifest_path.exists():
|
||
return {"asset_id": self.character_id, "approval_status": "DRAFT"}
|
||
# 简单解析 HLDP 文件
|
||
manifest = {}
|
||
with open(self.manifest_path, "r", encoding="utf-8") as f:
|
||
for line in f:
|
||
line = line.strip()
|
||
if line.startswith("asset_id:") or line.startswith("canonical_id:"):
|
||
manifest["asset_id"] = line.split(":", 1)[1].strip()
|
||
elif line.startswith("approval_status:"):
|
||
manifest["approval_status"] = line.split(":", 1)[1].strip()
|
||
elif line.startswith("candidate_front_half_body:"):
|
||
manifest["candidate_front_half_body"] = line.split(":", 1)[1].strip()
|
||
return manifest
|
||
|
||
def _load_character_description(self):
|
||
"""从 data/characters-v2.hdlp 加载角色描述"""
|
||
char_file = PROJECT_ROOT / "data" / "characters-v2.hdlp"
|
||
if not char_file.exists():
|
||
return None
|
||
|
||
# 简单解析
|
||
description = {}
|
||
with open(char_file, "r", encoding="utf-8") as f:
|
||
content = f.read()
|
||
# 找 CHAR-003 段落
|
||
if "CHAR-003" in content:
|
||
lines = content.split("\n")
|
||
in_char = False
|
||
for line in lines:
|
||
if "CHAR-003" in line:
|
||
in_char = True
|
||
elif in_char:
|
||
if line.strip().startswith("---"):
|
||
break
|
||
if ":" in line:
|
||
key, _, val = line.partition(":")
|
||
description[key.strip()] = val.strip()
|
||
return description
|
||
|
||
def generate_front_half_body(self, output_name="front_half_body.png"):
|
||
"""
|
||
生成正面半身批准图
|
||
使用候选图作为参考，用 Seedance/Kling 图生图生成稳定版本
|
||
"""
|
||
print(f"[1/4] 生成正面半身图: {output_name}")
|
||
|
||
candidate = self.manifest.get("candidate_front_half_body")
|
||
if not candidate or not Path(candidate).exists():
|
||
print(f" ❌ 候选图不存在: {candidate}")
|
||
print(f" 💡 请先准备候选图，放入 candidates/ 目录")
|
||
return None
|
||
|
||
# 构建提示词
|
||
prompt = self._build_prompt("front_half_body")
|
||
|
||
# 调用图像生成 API (图生图)
|
||
print(f" 参考图: {candidate}")
|
||
print(f" 提示词: {prompt[:100]}...")
|
||
|
||
if generate_image is None:
|
||
print(" ⚠️ 当前仓库只有 JS 图片适配器 image-api-adapter.js，未提供 Python generate_image。")
|
||
print(" ⚠️ 暂不能生成新主角资产；仅将候选图复制到 candidates/ 供人工/后续工具处理。")
|
||
fallback_path = self.candidates_dir / output_name
|
||
shutil.copy2(candidate, fallback_path)
|
||
print(f" 🟡 已复制候选图: {fallback_path}")
|
||
return None
|
||
|
||
try:
|
||
result_path = generate_image(
|
||
prompt=prompt,
|
||
reference_image=candidate,
|
||
output_dir=str(self.approved_dir),
|
||
output_name=output_name
|
||
)
|
||
print(f" ✅ 已生成: {result_path}")
|
||
return result_path
|
||
except Exception as e:
|
||
print(f" ❌ 生成失败: {e}")
|
||
return None
|
||
|
||
def generate_side_face(self, output_name="side_face.png"):
|
||
"""生成侧脸批准图"""
|
||
print(f"[2/4] 生成侧脸图: {output_name}")
|
||
prompt = self._build_prompt("side_face")
|
||
candidate = self.approved_dir / "front_half_body.png"
|
||
|
||
if not candidate.exists():
|
||
print(f" ❌ 请先生成正面半身图")
|
||
return None
|
||
|
||
try:
|
||
result_path = generate_image(
|
||
prompt=prompt,
|
||
reference_image=str(candidate),
|
||
output_dir=str(self.approved_dir),
|
||
output_name=output_name
|
||
)
|
||
print(f" ✅ 已生成: {result_path}")
|
||
return result_path
|
||
except Exception as e:
|
||
print(f" ❌ 生成失败: {e}")
|
||
return None
|
||
|
||
def generate_full_body_costume(self, output_name="full_body_costume.png"):
|
||
"""生成全身服装图"""
|
||
print(f"[3/4] 生成全身服装图: {output_name}")
|
||
prompt = self._build_prompt("full_body_costume")
|
||
candidate = self.approved_dir / "front_half_body.png"
|
||
|
||
if not candidate.exists():
|
||
print(f" ❌ 请先生成正面半身图")
|
||
return None
|
||
|
||
try:
|
||
result_path = generate_image(
|
||
prompt=prompt,
|
||
reference_image=str(candidate),
|
||
output_dir=str(self.approved_dir),
|
||
output_name=output_name
|
||
)
|
||
print(f" ✅ 已生成: {result_path}")
|
||
return result_path
|
||
except Exception as e:
|
||
print(f" ❌ 生成失败: {e}")
|
||
return None
|
||
|
||
def generate_expression_sheet(self, output_name="expression_sheet.png"):
|
||
"""生成表情表"""
|
||
print(f"[4/4] 生成表情表: {output_name}")
|
||
prompt = self._build_prompt("expression_sheet")
|
||
candidate = self.approved_dir / "front_half_body.png"
|
||
|
||
if not candidate.exists():
|
||
print(f" ❌ 请先生成正面半身图")
|
||
return None
|
||
|
||
try:
|
||
result_path = generate_image(
|
||
prompt=prompt,
|
||
reference_image=str(candidate),
|
||
output_dir=str(self.approved_dir),
|
||
output_name=output_name
|
||
)
|
||
print(f" ✅ 已生成: {result_path}")
|
||
return result_path
|
||
except Exception as e:
|
||
print(f" ❌ 生成失败: {e}")
|
||
return None
|
||
|
||
def _build_prompt(self, view_type):
|
||
"""构建特定视角的提示词"""
|
||
base_desc = self.character_desc or {}
|
||
|
||
prompts = {
|
||
"front_half_body": f"""
|
||
{base_desc.get('visual_description', '中国古代修仙少年，16岁，主角辨识度强')}
|
||
正面半身像，胸部以上，面部清晰，眼神坚定，
|
||
3D国风漫剧风格，统一渲染风格，
|
||
高清，8K，最佳质量
|
||
""".strip(),
|
||
|
||
"side_face": f"""
|
||
{base_desc.get('visual_description', '中国古代修仙少年')}
|
||
侧脸45度，能看到面部轮廓和发型，
|
||
3D国风漫剧风格，统一渲染风格，
|
||
高清，8K，最佳质量
|
||
""".strip(),
|
||
|
||
"full_body_costume": f"""
|
||
{base_desc.get('visual_description', '中国古代修仙少年')}
|
||
全身像，站立姿势，完整展示服装细节，
|
||
服装细节必须来自角色资产清单，不得自由改色，
|
||
3D国风漫剧风格，统一渲染风格，
|
||
高清，8K，最佳质量
|
||
""".strip(),
|
||
|
||
"expression_sheet": f"""
|
||
{base_desc.get('visual_description', '中国古代修仙少年')}
|
||
表情表，网格布局，包含:
|
||
平静，微笑，大笑，生气，惊讶，悲伤，
|
||
每个表情单独一格，统一光照和背景，
|
||
3D国风漫剧风格，
|
||
高清，8K，最佳质量
|
||
""".strip(),
|
||
}
|
||
|
||
return prompts.get(view_type, prompts["front_half_body"])
|
||
|
||
def generate_all(self):
|
||
"""生成所有资产"""
|
||
print(f"\n🎬 开始生成 {self.character_id} 主角资产包")
|
||
print(f"=" * 60)
|
||
|
||
results = {}
|
||
|
||
# 1. 正面半身
|
||
path = self.generate_front_half_body()
|
||
if path:
|
||
results["front_half_body"] = path
|
||
|
||
# 2. 侧脸
|
||
path = self.generate_side_face()
|
||
if path:
|
||
results["side_face"] = path
|
||
|
||
# 3. 全身服装
|
||
path = self.generate_full_body_costume()
|
||
if path:
|
||
results["full_body_costume"] = path
|
||
|
||
# 4. 表情表
|
||
path = self.generate_expression_sheet()
|
||
if path:
|
||
results["expression_sheet"] = path
|
||
|
||
# 更新 manifest。只有四项资产都真实生成时才可进入 APPROVED。
|
||
self._update_manifest(results)
|
||
|
||
print(f"\n✅ 资产包生成完成!")
|
||
print(f" 已生成: {len(results)}/4 个资产")
|
||
print(f" 位置: {self.approved_dir}")
|
||
|
||
return results
|
||
|
||
def _update_manifest(self, results):
|
||
"""更新 manifest.hdlp"""
|
||
print(f"\n📝 更新 manifest.hdlp...")
|
||
|
||
required_assets = {"front_half_body", "side_face", "full_body_costume", "expression_sheet"}
|
||
is_complete = required_assets.issubset(results.keys())
|
||
approval_status = "APPROVED" if is_complete else "DRAFT_GENERATED"
|
||
type_label = "已批准" if is_complete else "待质检"
|
||
lock_note = (
|
||
"⊢ 资产已批准，可用于成片镜头。"
|
||
if is_complete
|
||
else "⊢ 资产包尚未完整生成/质检，禁止作为成片锁定资产。"
|
||
)
|
||
|
||
manifest_content = f"""# 资产清单 · {self.character_id}
|
||
|
||
> HLDP://video-ai-system/assets/characters/{self.character_id}/manifest
|
||
> 类型: 角色资产 · {type_label}
|
||
> 建立: D143 · 2026-06-23
|
||
> 更新: D144 · 2026-06-24
|
||
> 项目: 付费才能修仙 · EP01
|
||
|
||
---
|
||
|
||
## 状态
|
||
|
||
```
|
||
approval_status: {approval_status}
|
||
asset_type: character
|
||
canonical_id: {self.character_id}
|
||
canonical_name: 苏白
|
||
```
|
||
|
||
---
|
||
|
||
## 批准资产
|
||
|
||
```
|
||
front_half_body: {results.get("front_half_body", "NOT_GENERATED")}
|
||
side_face: {results.get("side_face", "NOT_GENERATED")}
|
||
full_body_costume: {results.get("full_body_costume", "NOT_GENERATED")}
|
||
expression_sheet: {results.get("expression_sheet", "NOT_GENERATED")}
|
||
```
|
||
|
||
---
|
||
|
||
## 视觉锁
|
||
|
||
```
|
||
face_shape: TODO_FROM_APPROVED_ASSET
|
||
hair_style: TODO_FROM_APPROVED_ASSET
|
||
costume: TODO_FROM_APPROVED_ASSET
|
||
age_band: 16岁
|
||
render_style: 3D国风漫剧
|
||
color_palette: TODO_FROM_APPROVED_ASSET
|
||
```
|
||
|
||
---
|
||
|
||
## 锁定
|
||
|
||
{lock_note}
|
||
⊢ 禁止使用 candidates/ 或 rejected/ 中的图片作为最终资产。
|
||
"""
|
||
|
||
with open(self.manifest_path, "w", encoding="utf-8") as f:
|
||
f.write(manifest_content)
|
||
|
||
print(f" ✅ 已更新: {self.manifest_path}")
|
||
|
||
|
||
def main():
|
||
parser = argparse.ArgumentParser(description="CHAR-HERO-DESIGN-PACKER")
|
||
parser.add_argument("--character", type=str, default="CHAR-003-SuBai",
|
||
help="角色ID (默认: CHAR-003-SuBai)")
|
||
parser.add_argument("--generate-all", action="store_true",
|
||
help="生成所有资产")
|
||
parser.add_argument("--view", type=str,
|
||
choices=["front_half_body", "side_face", "full_body_costume", "expression_sheet"],
|
||
help="生成特定视角的资产")
|
||
|
||
args = parser.parse_args()
|
||
|
||
packer = CharHeroDesignPacker(args.character)
|
||
|
||
if args.generate_all:
|
||
packer.generate_all()
|
||
elif args.view:
|
||
method_map = {
|
||
"front_half_body": packer.generate_front_half_body,
|
||
"side_face": packer.generate_side_face,
|
||
"full_body_costume": packer.generate_full_body_costume,
|
||
"expression_sheet": packer.generate_expression_sheet,
|
||
}
|
||
method_map[args.view]()
|
||
else:
|
||
print("请指定 --generate-all 或 --view <view_type>")
|
||
parser.print_help()
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|