diff --git a/app/oobabooga - webui.py b/app/oobabooga - webui.py index c673b46..4008869 100644 --- a/app/oobabooga - webui.py +++ b/app/oobabooga - webui.py @@ -735,7 +735,6 @@ def generate_character_avatar( ) print(sd_prompt) sd_filter(nsfw_filter) - process_uploaded_image return image_generate(character_name, sd_prompt, input_none(negative_prompt), @@ -766,9 +765,18 @@ def image_generate(character_name, prompt, negative_prompt): character_name = character_name.replace(" ", "_") os.makedirs(f"characters/{character_name}", exist_ok=True) - card_path = f"characters/{character_name}/{character_name}.png" + image_path = f"characters/{character_name}/{character_name}.png" + + # Save the generated image + generated_image.save(image_path) + + # Load the image back into a NumPy array + reloaded_image = Image.open(image_path) + reloaded_image_np = np.array(reloaded_image) + + # Call process_uploaded_image + process_uploaded_image(reloaded_image_np) - generated_image.save(card_path) print("Generated character avatar") return generated_image diff --git a/characters/Kaori/Kaori.card.png b/characters/Kaori/Kaori.card.png new file mode 100644 index 0000000..a6c933c Binary files /dev/null and b/characters/Kaori/Kaori.card.png differ diff --git a/characters/Kaori/Kaori.png b/characters/Kaori/Kaori.png new file mode 100644 index 0000000..542f582 Binary files /dev/null and b/characters/Kaori/Kaori.png differ diff --git a/characters/uploaded_character/uploaded_character.png b/characters/uploaded_character/uploaded_character.png new file mode 100644 index 0000000..542f582 Binary files /dev/null and b/characters/uploaded_character/uploaded_character.png differ