Prepend character descriptions to global prompt
This commit is contained in:
@@ -375,6 +375,21 @@ def _character_prompt_replacements(characters: list[dict]) -> dict[str, str]:
|
||||
return replacements
|
||||
|
||||
|
||||
def _prepend_character_descriptions(global_prompt: str, characters: list[dict] | None = None) -> str:
|
||||
descriptions = [
|
||||
(character.get("description", "") or "").strip()
|
||||
for character in (characters or [])
|
||||
if (character.get("description", "") or "").strip()
|
||||
]
|
||||
if not descriptions:
|
||||
return global_prompt or ""
|
||||
|
||||
prefix = ". ".join(descriptions)
|
||||
if not (global_prompt or "").strip():
|
||||
return prefix
|
||||
return f"{prefix}. {global_prompt.strip()}"
|
||||
|
||||
|
||||
def _preprocess_prompts_with_characters(global_prompt, local_prompts, characters: list[dict] | None = None):
|
||||
"""Invisibly swaps out @characterN/@charN/@alias tags with their character descriptions."""
|
||||
if "@" not in (global_prompt or "") and "@" not in (local_prompts or ""):
|
||||
@@ -2084,6 +2099,7 @@ class LTXDirector(io.ComfyNode):
|
||||
image_cache=image_cache,
|
||||
input_dir=input_dir,
|
||||
)
|
||||
global_prompt = _prepend_character_descriptions(global_prompt, characters)
|
||||
char_slot_images = [list(character.get("images") or []) for character in characters]
|
||||
char_images = [img for slot_images in char_slot_images for img in slot_images]
|
||||
|
||||
|
||||
Reference in New Issue
Block a user