115 lines
3.7 KiB
Python
115 lines
3.7 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
"""
|
||
image_api_adapter.py · Python端图片API适配器
|
||
D144 · 铸渊 ICE-GL-ZY001
|
||
|
||
桥接到 Node.js image-api-bridge.js，为 char-hero-design-packer.py 提供
|
||
generate_image() 函数。
|
||
|
||
本地密钥入口: VIDEO_AI_SECRETS_FILE 或 LOCAL-SECRETS-PATH.hdlp 登记路径
|
||
不提交仓库，不硬编码密钥。
|
||
"""
|
||
|
||
import json
|
||
import os
|
||
import subprocess
|
||
from pathlib import Path
|
||
|
||
ENGINES_DIR = Path(__file__).parent
|
||
BRIDGE_SCRIPT = ENGINES_DIR / "image-api-bridge.js"
|
||
NODE_BIN = "/Users/bingshuolingdianyuanhe/.workbuddy/binaries/node/versions/22.22.2/bin/node"
|
||
|
||
|
||
def _call_bridge(action: str, **kwargs) -> dict:
|
||
"""
|
||
调用 Node.js 桥接脚本
|
||
:param action: 操作名 (generate_character_ref | generate_image)
|
||
:param kwargs: 传给 JS 的参数
|
||
:return: {ok: bool, data: {...} | error: str}
|
||
"""
|
||
if not BRIDGE_SCRIPT.exists():
|
||
return {"ok": False, "error": f"桥接脚本不存在: {BRIDGE_SCRIPT}"}
|
||
|
||
payload = {"action": action, **kwargs}
|
||
|
||
try:
|
||
result = subprocess.run(
|
||
[NODE_BIN, str(BRIDGE_SCRIPT)],
|
||
input=json.dumps(payload, ensure_ascii=False),
|
||
capture_output=True,
|
||
text=True,
|
||
timeout=120,
|
||
cwd=str(ENGINES_DIR),
|
||
)
|
||
except FileNotFoundError:
|
||
return {"ok": False, "error": f"Node 未找到: {NODE_BIN}"}
|
||
except subprocess.TimeoutExpired:
|
||
return {"ok": False, "error": "桥接调用超时（120秒）"}
|
||
|
||
if result.returncode != 0:
|
||
stderr = result.stderr.strip()[:500] if result.stderr else "无输出"
|
||
return {"ok": False, "error": f"桥接退出码 {result.returncode}: {stderr}"}
|
||
|
||
try:
|
||
return json.loads(result.stdout.strip())
|
||
except json.JSONDecodeError:
|
||
return {"ok": False, "error": f"桥接输出非JSON: {result.stdout[:200]}"}
|
||
|
||
|
||
def generate_image(prompt: str, reference_image: str = None,
|
||
output_dir: str = None, output_name: str = None) -> str:
|
||
"""
|
||
生成图片（为 char-hero-design-packer.py 提供接口）
|
||
|
||
:param prompt: 提示词
|
||
:param reference_image: 参考图路径（当前桥接为文生图，参考图暂不支持）
|
||
:param output_dir: 输出目录
|
||
:param output_name: 输出文件名
|
||
:return: 生成的图片路径，失败返回 None
|
||
"""
|
||
kwargs = {"prompt": prompt, "size": "2048x2048"}
|
||
|
||
if output_dir and output_name:
|
||
kwargs["outputPath"] = os.path.join(output_dir, output_name)
|
||
elif output_name:
|
||
kwargs["filename"] = output_name
|
||
|
||
result = _call_bridge("generate_image", **kwargs)
|
||
|
||
if not result.get("ok"):
|
||
print(f" ⚠️ 图片生成失败: {result.get('error', '未知错误')}")
|
||
return None
|
||
|
||
data = result.get("data", {})
|
||
image_path = data.get("imagePath", "")
|
||
if image_path and os.path.isfile(image_path):
|
||
return image_path
|
||
return None
|
||
|
||
|
||
def generate_character_ref(char_id: str, l0_descriptor: str = None,
|
||
l1_clothing: str = None) -> str:
|
||
"""
|
||
生成角色参考图
|
||
|
||
:param char_id: 角色编号，如 CHAR-003
|
||
:param l0_descriptor: L0底层编码描述
|
||
:param l1_clothing: L1服装配置
|
||
:return: 生成的图片路径，失败返回 None
|
||
"""
|
||
kwargs = {"charId": char_id}
|
||
if l0_descriptor:
|
||
kwargs["l0"] = l0_descriptor
|
||
if l1_clothing:
|
||
kwargs["l1"] = l1_clothing
|
||
|
||
result = _call_bridge("generate_character_ref", **kwargs)
|
||
|
||
if not result.get("ok"):
|
||
print(f" ⚠️ 角色参考图生成失败: {result.get('error', '未知错误')}")
|
||
return None
|
||
|
||
data = result.get("data", {})
|
||
return data.get("imagePath", "")
|