From 0ffc31ae3eeda7d2318168dd67941438fb648f26 Mon Sep 17 00:00:00 2001 From: jinye_huang Date: Fri, 16 May 2025 17:21:39 +0800 Subject: [PATCH] =?UTF-8?q?=E9=87=8D=E6=96=B0=E7=94=9F=E6=88=90=E5=85=A8?= =?UTF-8?q?=E9=83=A8=E9=85=8D=E5=9B=BE?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- scripts/reimage/regenerate_images.py | 97 ++++++++++++++++++++++++++-- 1 file changed, 93 insertions(+), 4 deletions(-) diff --git a/scripts/reimage/regenerate_images.py b/scripts/reimage/regenerate_images.py index d80d88e..d9ce218 100644 --- a/scripts/reimage/regenerate_images.py +++ b/scripts/reimage/regenerate_images.py @@ -28,8 +28,17 @@ def load_config(config_path="poster_gen_config.json"): logging.error(f"加载配置文件时出错: {e}") return None -def regenerate_posters_and_images(run_dir: str, config: Dict): - """为所有主题重新生成海报和配图""" +def regenerate_posters_and_images(run_dir: str, config: Dict, target_topic: int = None, + ignore_metadata: bool = False, poster_config_path: str = None): + """为所有主题重新生成海报和配图 + + Args: + run_dir: 运行目录路径 + config: 配置字典 + target_topic: 只处理特定主题ID + ignore_metadata: 是否忽略现有的元数据 + poster_config_path: 海报配置文件路径 + """ run_id = os.path.basename(run_dir) logging.info(f"处理运行ID: {run_id}") @@ -51,6 +60,17 @@ def regenerate_posters_and_images(run_dir: str, config: Dict): logging.error("无法加载主题列表") return + # 如果指定了poster_config_path，加载自定义海报配置 + poster_configs = None + if poster_config_path: + try: + with open(poster_config_path, 'r', encoding='utf-8') as f: + poster_configs = json.load(f) + logging.info(f"已加载自定义海报配置，包含 {len(poster_configs)} 个配置项") + except Exception as e: + logging.error(f"加载海报配置文件时出错: {e}") + return + # 加载配置参数 poster_variants = config.get("variants", 1) poster_assets_dir = config.get("poster_assets_base_dir") @@ -90,8 +110,48 @@ def regenerate_posters_and_images(run_dir: str, config: Dict): logging.warning(f"主题缺少索引，跳过: {topic_item}") continue + # 如果指定了target_topic且不匹配当前主题，则跳过 + if target_topic is not None and topic_index != target_topic: + continue + + # 检查是否有自定义海报配置 + custom_config = None + if poster_configs: + for config_item in poster_configs: + if config_item.get('index') == topic_index: + custom_config = config_item + break + + if custom_config: + # 更新主题标题 + if 'main_title' in custom_config: + topic_item['title'] = custom_config['main_title'] + logging.info(f"使用自定义标题: {custom_config['main_title']}") + logging.info(f"处理主题 {topic_index}: {topic_item.get('object', 'N/A')}") + # 如果忽略元数据，先清空所有现有的海报和配图文件 + if ignore_metadata: + for variant_index in range(1, poster_variants + 1): + variant_key = f"{topic_index}_{variant_index}" + topic_dir = os.path.join(run_dir, variant_key) + + if os.path.exists(topic_dir): + poster_dir = os.path.join(topic_dir, poster_subdir) + if os.path.exists(poster_dir): + logging.info(f"清空变体 {variant_key} 的海报目录") + for file_name in os.listdir(poster_dir): + file_path = os.path.join(poster_dir, file_name) + if os.path.isfile(file_path): + os.remove(file_path) + logging.debug(f"已删除文件: {file_path}") + else: + os.makedirs(poster_dir, exist_ok=True) + else: + os.makedirs(topic_dir, exist_ok=True) + os.makedirs(os.path.join(topic_dir, poster_subdir), exist_ok=True) + logging.info(f"创建变体目录: {topic_dir}") + # 1. 为此主题生成所有变体的海报（一次性调用） logging.info(f"为主题 {topic_index} 生成所有变体的海报...") @@ -152,6 +212,13 @@ def regenerate_posters_and_images(run_dir: str, config: Dict): continue try: + # 删除现有的额外配图文件 + for file_name in os.listdir(poster_dir): + if file_name.startswith("additional_") and file_name.endswith(".jpg"): + file_path = os.path.join(poster_dir, file_name) + os.remove(file_path) + logging.debug(f"已删除额外配图: {file_path}") + # 查找元数据文件 metadata_files = [f for f in os.listdir(poster_dir) if f.endswith("_metadata.json")] @@ -189,6 +256,10 @@ def regenerate_posters_and_images(run_dir: str, config: Dict): except Exception as e: logging.exception(f"生成变体 {variant_key} 额外配图时出错: {e}") + + # 如果指定了target_topic但没有找到 + if target_topic is not None and all(topic.get('index') != target_topic for topic in topics_list): + logging.warning(f"未找到指定的主题ID: {target_topic}") def main(): parser = argparse.ArgumentParser(description="为所有主题重新生成海报和配图") @@ -196,6 +267,8 @@ def main(): parser.add_argument("--config", default="poster_gen_config.json", help="配置文件路径") parser.add_argument("--debug", action="store_true", help="启用调试日志") parser.add_argument("--topic", type=int, help="只处理指定主题索引") + parser.add_argument("--ignore-metadata", action="store_true", help="忽略现有的元数据，重新生成所有海报和图片") + parser.add_argument("--poster-config", help="指定海报配置文件路径，用于自定义海报内容") args = parser.parse_args() # 设置日志级别 @@ -216,9 +289,25 @@ def main(): logging.error(f"指定的运行目录不存在: {args.run_dir}") sys.exit(1) + # 修改regenerate_posters_and_images函数，添加新参数 + if args.ignore_metadata: + logging.info("将忽略现有元数据，完全重新生成所有海报和配图") + + # 如果指定了poster-config + if args.poster_config: + logging.info(f"使用自定义海报配置文件: {args.poster_config}") + # 检查文件是否存在 + if not os.path.exists(args.poster_config): + logging.error(f"指定的海报配置文件不存在: {args.poster_config}") + sys.exit(1) + # 开始处理 - logging.info(f"开始为 {args.run_dir} 重新生成海报和配图") - regenerate_posters_and_images(args.run_dir, config) + if args.topic: + logging.info(f"只处理主题 {args.topic} 的海报和配图") + else: + logging.info(f"开始为 {args.run_dir} 重新生成海报和配图") + + regenerate_posters_and_images(args.run_dir, config, args.topic, args.ignore_metadata, args.poster_config) logging.info("处理完成") if __name__ == "__main__":