改变了相册逻辑 删除了无效资源
This commit is contained in:
parent
7f8d9f8c0c
commit
cca622d34f
Binary file not shown.
4
main.py
4
main.py
@ -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:
|
||||||
|
|||||||
@ -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",
|
||||||
|
|||||||
Binary file not shown.
@ -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
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user