改变了相册逻辑 删除了无效资源

This commit is contained in:
jinye_huang 2025-04-24 19:30:27 +08:00
parent 7f8d9f8c0c
commit cca622d34f
6 changed files with 3 additions and 12 deletions

View File

@ -148,7 +148,6 @@ def generate_content_and_posters_step(config, run_id, topics_list, output_handle
poster_variants = config.get("variants", 1) # 通常与 content variants 相同 poster_variants = config.get("variants", 1) # 通常与 content variants 相同
poster_assets_dir = config.get("poster_assets_base_dir") poster_assets_dir = config.get("poster_assets_base_dir")
img_base_dir = config.get("image_base_dir") img_base_dir = config.get("image_base_dir")
mod_img_subdir = config.get("modify_image_subdir", "modify")
res_dir_config = config.get("resource_dir", []) res_dir_config = config.get("resource_dir", [])
poster_size = tuple(config.get("poster_target_size", [900, 1200])) poster_size = tuple(config.get("poster_target_size", [900, 1200]))
txt_possibility = config.get("text_possibility", 0.3) txt_possibility = config.get("text_possibility", 0.3)
@ -157,7 +156,6 @@ def generate_content_and_posters_step(config, run_id, topics_list, output_handle
collage_subdir = config.get("output_collage_subdir", "collage_img") collage_subdir = config.get("output_collage_subdir", "collage_img")
poster_subdir = config.get("output_poster_subdir", "poster") poster_subdir = config.get("output_poster_subdir", "poster")
poster_filename = config.get("output_poster_filename", "poster.jpg") poster_filename = config.get("output_poster_filename", "poster.jpg")
cam_img_subdir = config.get("camera_image_subdir", "相机")
poster_content_system_prompt = config.get("poster_content_system_prompt", None) poster_content_system_prompt = config.get("poster_content_system_prompt", None)
# 检查关键路径是否存在 # 检查关键路径是否存在
@ -179,7 +177,6 @@ def generate_content_and_posters_step(config, run_id, topics_list, output_handle
variants=poster_variants, variants=poster_variants,
poster_assets_base_dir=poster_assets_dir, poster_assets_base_dir=poster_assets_dir,
image_base_dir=img_base_dir, image_base_dir=img_base_dir,
modify_image_subdir=mod_img_subdir,
resource_dir_config=res_dir_config, resource_dir_config=res_dir_config,
poster_target_size=poster_size, poster_target_size=poster_size,
text_possibility=txt_possibility, text_possibility=txt_possibility,
@ -188,7 +185,6 @@ def generate_content_and_posters_step(config, run_id, topics_list, output_handle
output_collage_subdir=collage_subdir, output_collage_subdir=collage_subdir,
output_poster_subdir=poster_subdir, output_poster_subdir=poster_subdir,
output_poster_filename=poster_filename, output_poster_filename=poster_filename,
camera_image_subdir=cam_img_subdir,
system_prompt=poster_content_system_prompt system_prompt=poster_content_system_prompt
) )
if posters_attempted: if posters_attempted:

View File

@ -42,9 +42,6 @@
"output_dir": "./result", "output_dir": "./result",
"image_base_dir": "/root/autodl-tmp/TravelContentCreator/hotel_img", "image_base_dir": "/root/autodl-tmp/TravelContentCreator/hotel_img",
"poster_assets_base_dir": "/root/autodl-tmp/poster_baseboard_0403", "poster_assets_base_dir": "/root/autodl-tmp/poster_baseboard_0403",
"camera_image_subdir": "",
"modify_image_subdir": "",
"request_timeout": 210, "request_timeout": 210,
"max_retries": 3, "max_retries": 3,
"description_filename": "description.txt", "description_filename": "description.txt",

View File

@ -444,7 +444,6 @@ def generate_posters_for_topic(topic_item: dict,
variants: int, variants: int,
poster_assets_base_dir: str, poster_assets_base_dir: str,
image_base_dir: str, image_base_dir: str,
modify_image_subdir: str,
img_frame_possibility: float, img_frame_possibility: float,
text_bg_possibility: float, text_bg_possibility: float,
resource_dir_config: list, resource_dir_config: list,
@ -453,7 +452,6 @@ def generate_posters_for_topic(topic_item: dict,
output_collage_subdir: str, output_collage_subdir: str,
output_poster_subdir: str, output_poster_subdir: str,
output_poster_filename: str, output_poster_filename: str,
camera_image_subdir: str,
system_prompt: str system_prompt: str
): ):
"""Generates all posters for a single topic item, handling image data via OutputHandler. """Generates all posters for a single topic item, handling image data via OutputHandler.
@ -466,14 +464,14 @@ def generate_posters_for_topic(topic_item: dict,
variants: Number of variants. variants: Number of variants.
poster_assets_base_dir: Path to poster assets (fonts, frames etc.). poster_assets_base_dir: Path to poster assets (fonts, frames etc.).
image_base_dir: Base path for source images. image_base_dir: Base path for source images.
modify_image_subdir: Subdirectory for modified images. img_frame_possibility: Probability of adding image frame.
text_bg_possibility: Probability of adding text background.
resource_dir_config: Configuration for resource directories (used for Description). resource_dir_config: Configuration for resource directories (used for Description).
poster_target_size: Target size tuple (width, height) for the poster. poster_target_size: Target size tuple (width, height) for the poster.
text_possibility: Probability of adding secondary text. text_possibility: Probability of adding secondary text.
output_collage_subdir: Subdirectory name for saving collages. output_collage_subdir: Subdirectory name for saving collages.
output_poster_subdir: Subdirectory name for saving posters. output_poster_subdir: Subdirectory name for saving posters.
output_poster_filename: Filename for the final poster. output_poster_filename: Filename for the final poster.
camera_image_subdir: Subdirectory for camera images (currently unused in logic?).
system_prompt: System prompt for content generation. system_prompt: System prompt for content generation.
output_handler: An instance of OutputHandler to process results. output_handler: An instance of OutputHandler to process results.
@ -543,7 +541,7 @@ def generate_posters_for_topic(topic_item: dict,
return False return False
# Construct and check INPUT image paths # Construct and check INPUT image paths
input_img_dir_path = os.path.join(image_base_dir, modify_image_subdir, object_name) input_img_dir_path = os.path.join(image_base_dir, object_name)
if not os.path.exists(input_img_dir_path) or not os.path.isdir(input_img_dir_path): if not os.path.exists(input_img_dir_path) or not os.path.isdir(input_img_dir_path):
logging.warning(f"Warning: Modify Image directory not found or not a directory: '{input_img_dir_path}'. Skipping posters for this topic.") logging.warning(f"Warning: Modify Image directory not found or not a directory: '{input_img_dir_path}'. Skipping posters for this topic.")
return False return False