From 1b47a89c179406851a470ecbfaa1dd8842d81310 Mon Sep 17 00:00:00 2001 From: j Date: Fri, 16 Jan 2026 11:12:40 +0000 Subject: [PATCH] tweak characters --- generate.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/generate.py b/generate.py index dc0fce2..2f35366 100644 --- a/generate.py +++ b/generate.py @@ -372,8 +372,11 @@ def process_frame(item, prompt, character=None, position=0): return img def replace_character(item, character, position=0): - prompt = "Replace the foreground character in image 1 with the character in image 2, keep the posture, clothing, background, light, admosthere from image 1, but take the face and personality from image 2. Make sure the size of the character is adjusted since the new character is a child. The quality of the image should be the same between foreground and background, adjust the quality of the character to match the background" - if character in ("P1", "P2", "P3"): + prompt = "Replace the foreground character in image 1 with the character in image 2, keep the posture, clothing, background, light, admosthere from image 1, but take the facial features and personality from image 2. Make sure the size of the character is adjusted since the new character is a child and make sure the size of the head matches the body. The quality of the image should be the same between foreground and background, adjust the quality of the character to match the background. Use the style of image 1 for the character: if image 1 is a photo make the character a real person, if image 1 is a drawing make the character a drawn character, if image 1 is a comic use a comic character and so on" + prompt = "Replace the foreground character in image 1 with the character in image 2, keep the posture, clothing, background, light, admosthere from image 1, but take the facial features and personality from image 2. Make sure the size of the character is adjusted since the new character is a child and make sure the size of the head matches the body. The quality of the image should be the same between foreground and background, adjust the quality of the character to match the background. Use the style of image 1 for the character: if image 1 is a photo make the character a real person, if image 1 is a drawing make the character a drawn character, if image 1 is a comic use a comic character" + if character == "P5": + prompt = prompt.replace('child', 'teenager') + if character in ("P1", "P2", "P3", "P4", "P5"): character = public_document_url(Document.objects.get(data__title="Character " + character)) return process_frame(item, prompt, character, position)