import torch import torchvision.transforms.functional as TF import numpy as np import random import os import sys import tempfile from PIL import Image from diffusers.utils import load_image from diffusers import EulerDiscreteScheduler, T2IAdapter from huggingface_hub import hf_hub_download import spaces import gradio as gr from pipeline_t2i_adapter import PhotoMakerStableDiffusionXLAdapterPipeline from face_utils import FaceAnalysis2, analyze_faces from style_template import styles from aspect_ratio_template import aspect_ratios # global variable base_model_path = 'SG161222/RealVisXL_V4.0' face_detector = FaceAnalysis2(providers=['CPUExecutionProvider', 'CUDAExecutionProvider'], allowed_modules=['detection', 'recognition']) face_detector.prepare(ctx_id=0, det_size=(640, 640)) try: if torch.cuda.is_available(): device = "cuda" elif sys.platform == "darwin" and torch.backends.mps.is_available(): device = "mps" else: device = "cpu" except: device = "cpu" HF_TEST_USER_NAME = os.environ["HF_TEST_USER_NAME"] HF_PASS = os.environ["HF_PASS"] MAX_SEED = np.iinfo(np.int32).max STYLE_NAMES = list(styles.keys()) DEFAULT_STYLE_NAME = "Photographic (Default)" ASPECT_RATIO_LABELS = list(aspect_ratios) DEFAULT_ASPECT_RATIO = ASPECT_RATIO_LABELS[0] enable_doodle_arg = False photomaker_ckpt = hf_hub_download(repo_id="TencentARC/PhotoMaker-V2", filename="photomaker-v2.bin", repo_type="model") if torch.cuda.is_available() and torch.cuda.is_bf16_supported(): torch_dtype = torch.bfloat16 else: torch_dtype = torch.float16 if device == "mps": torch_dtype = torch.float16 # load adapter adapter = T2IAdapter.from_pretrained( "TencentARC/t2i-adapter-sketch-sdxl-1.0", torch_dtype=torch_dtype, variant="fp16" ).to(device) pipe = PhotoMakerStableDiffusionXLAdapterPipeline.from_pretrained( base_model_path, adapter=adapter, torch_dtype=torch_dtype, use_safetensors=True, variant="fp16", ).to(device) pipe.load_photomaker_adapter( os.path.dirname(photomaker_ckpt), subfolder="", weight_name=os.path.basename(photomaker_ckpt), trigger_word="img", pm_version="v2", ) pipe.id_encoder.to(device) pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config) # pipe.set_adapters(["photomaker"], adapter_weights=[1.0]) pipe.fuse_lora() pipe.to(device) @spaces.GPU() def generate_image( upload_images, prompt, negative_prompt, aspect_ratio_name, style_name, num_steps, style_strength_ratio, num_outputs, guidance_scale, seed, use_doodle, sketch_image, adapter_conditioning_scale, adapter_conditioning_factor, progress=gr.Progress(track_tqdm=True) ): if use_doodle: sketch_image = sketch_image["composite"] r, g, b, a = sketch_image.split() sketch_image = a.convert("RGB") sketch_image = TF.to_tensor(sketch_image) > 0.5 # Inversion sketch_image = TF.to_pil_image(sketch_image.to(torch.float32)) adapter_conditioning_scale = adapter_conditioning_scale adapter_conditioning_factor = adapter_conditioning_factor else: adapter_conditioning_scale = 0. adapter_conditioning_factor = 0. sketch_image = None # check the trigger word image_token_id = pipe.tokenizer.convert_tokens_to_ids(pipe.trigger_word) input_ids = pipe.tokenizer.encode(prompt) if image_token_id not in input_ids: raise gr.Error(f"Cannot find the trigger word '{pipe.trigger_word}' in text prompt! Please refer to step 2️⃣") if input_ids.count(image_token_id) > 1: raise gr.Error(f"Cannot use multiple trigger words '{pipe.trigger_word}' in text prompt!") # determine output dimensions by the aspect ratio output_w, output_h = aspect_ratios[aspect_ratio_name] print(f"[Debug] Generate image using aspect ratio [{aspect_ratio_name}] => {output_w} x {output_h}") # apply the style template prompt, negative_prompt = apply_style(style_name, prompt, negative_prompt) if upload_images is None: raise gr.Error(f"Cannot find any input face image! Please refer to step 1️⃣") input_id_images = [] for img in upload_images: input_id_images.append(load_image(img)) id_embed_list = [] for img in input_id_images: img = np.array(img) img = img[:, :, ::-1] faces = analyze_faces(face_detector, img) if len(faces) > 0: id_embed_list.append(torch.from_numpy((faces[0]['embedding']))) if len(id_embed_list) == 0: raise gr.Error(f"No face detected, please update the input face image(s)") id_embeds = torch.stack(id_embed_list) generator = torch.Generator(device=device).manual_seed(seed) print("Start inference...") print(f"[Debug] Seed: {seed}") print(f"[Debug] Prompt: {prompt}, \n[Debug] Neg Prompt: {negative_prompt}") start_merge_step = int(float(style_strength_ratio) / 100 * num_steps) if start_merge_step > 30: start_merge_step = 30 print(start_merge_step) images = pipe( prompt=prompt, width=output_w, height=output_h, input_id_images=input_id_images, negative_prompt=negative_prompt, num_images_per_prompt=num_outputs, num_inference_steps=num_steps, start_merge_step=start_merge_step, generator=generator, guidance_scale=guidance_scale, id_embeds=id_embeds, image=sketch_image, adapter_conditioning_scale=adapter_conditioning_scale, adapter_conditioning_factor=adapter_conditioning_factor, ).images # Save images to temp files for download temp_paths = [] for i, img in enumerate(images): tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".png") img.save(tmp.name) temp_paths.append(tmp.name) return images, gr.update(visible=True), temp_paths[0], temp_paths[1] # return images, gr.update(visible=True) def swap_to_gallery(images): return gr.update(value=images, visible=True), gr.update(visible=True), gr.update(visible=False) def upload_example_to_gallery(images, prompt, style, negative_prompt): return gr.update(value=images, visible=True), gr.update(visible=True), gr.update(visible=False) def remove_back_to_files(): return gr.update(visible=False), gr.update(visible=False), gr.update(visible=True) def change_doodle_space(use_doodle): if use_doodle: return gr.update(visible=True) else: return gr.update(visible=False) def remove_tips(): return gr.update(visible=False) def randomize_seed_fn(seed: int, randomize_seed: bool) -> int: if randomize_seed: seed = random.randint(0, MAX_SEED) return seed def apply_style(style_name: str, positive: str, negative: str = "") -> tuple[str, str]: p, n = styles.get(style_name, styles[DEFAULT_STYLE_NAME]) return p.replace("{prompt}", positive), n + ' ' + negative def get_image_path_list(folder_name): image_basename_list = os.listdir(folder_name) image_path_list = sorted([os.path.join(folder_name, basename) for basename in image_basename_list]) return image_path_list def get_example(): case = [ [ get_image_path_list('./examples/scarletthead_woman'), "instagram photo, portrait photo of a woman img, colorful, perfect face, natural skin, hard shadows, film grain", "(No style)", "(asymmetry, worst quality, low quality, illustration, 3d, 2d, painting, cartoons, sketch), open mouth", ], [ get_image_path_list('./examples/newton_man'), "sci-fi, closeup portrait photo of a man img wearing the sunglasses in Iron man suit, face, slim body, high quality, film grain", "(No style)", "(asymmetry, worst quality, low quality, illustration, 3d, 2d, painting, cartoons, sketch), open mouth", ], ] return case def set_gender_prompt(gender): return f"A photo of a {gender} img" man_prompts = { "Garden":"Photo of a man img, realistic eyes, natural skin, no hands. The model wears a light cream linen shirt with beige trousers, standing in a bloomed garden, with a romantic vibe.", "Balcony":"Photo of a man img, realistic eyes, natural skin, no hands. The model wears a tailored evening suit, standing on a balcony with an elegant vibe.", "Park":"Photo of a man img, realistic eyes, natural skin, no hands. The model wears a fitted athletic T-shirt and joggers, posing in a park with an energetic vibe.", "Cafe":"Photo of a man img, realistic eyes, natural skin, no hands. The model wears a turtleneck sweater with tailored trousers, sitting by a cafe window.", "Beach":"Photo of a man img, realistic eyes, natural skin, no hands. The model wears stylish swim trunks, standing on a sandy beach at sunset.", "Sidewalk":"Photo of a man img, realistic eyes, natural skin, no hands. The model wears a tailored wool overcoat and leather ankle boots, standing on a sidewalk with a cozy autumn vibe.", "Street":"Photo of a man img, realistic eyes, natural skin, no hands. The model wears a long coat with the collar popped up, standing on a street with a cozy, masculine mood.", "Moders Office":"Photo of a man img, realistic eyes, natural skin, no hands. The model wears a dark tailored suit with a slim tie, standing confidently in a modern office.", "Posing by a desk":"Photo of a man img, realistic eyes, natural skin, no hands. The model wears a fitted shirt with relaxed wide-leg pants, posing confidently by a desk.", "Resort Pool":"Photo of a man img, realistic eyes, natural skin, no hands. The model wears swim trunks, standing by a luxurious pool with natural sunlight, giving a resort vibe.", "Cozy":"Photo of a man img, realistic eyes, natural skin, no hands. The model wears a simple oversized sweater and comfortable pants, lounging on a couch, relaxed atmosphere.", "Palm Breeze":"Photo of a man img, realistic eyes, natural skin, no hands. The model wears solid-color board shorts, standing confidently on a beach lined with palm trees and crystal-clear water.", "Beach Shore":"Photo of a man img, realistic eyes, natural skin, no hands. The model wears bright board shorts, standing on the shore with gentle waves and a beachy atmosphere.", "Vacation":"Photo of a man img, realistic eyes, natural skin, no hands. The model wears swim trunks, vacation-ready vibe, standing by a crystal-clear infinity pool under bright sunlight.", "Field":"Photo of a man img, realistic eyes, natural skin, no hands. The model wears a rugged denim jacket over a simple white tee and cargo pants, standing in a sunlit meadow with wildflowers, giving a rustic, adventurous vibe.", "Ocean Edge":"Photo of a man img, realistic eyes, natural skin, no hands. The model wears a jacket and tailored pants, standing on a cliff overlooking the ocean with a dramatic vibe.", "Club Mood":"Photo of a man img, realistic eyes, natural skin, no hands. The model wears a black leather bomber jacket and dark denim, standing against a dimly lit club wall with a relaxed yet bold atmosphere." } woman_prompts = { "Garden":"Photo of a woman img, realistic eyes, natural skin, no hands. The model wears a floral maxi dress, standing in a bloomed garden, with a romantic vibe.", "Balcony":"Photo of a woman img, realistic eyes, natural skin, no hands. The model wears an evening gown, standing on a balcony with an elegant vibe.", "Park":"Photo of a woman img, realistic eyes, natural skin, no hands. The model wears a sporty crop top and leggings, posing in a park with an energetic vibe.", "Cafe": "Photo of a woman img, realistic eyes, natural skin, no hands. The model wears a high-neck blouse with a skirt, sitting by a cafe window.", "Beach":"Photo of a woman img, realistic eyes, natural skin, no hands. The model wears a stylish bikini, standing on a sandy beach at sunset.", "Sidewalk":"Photo of a woman img, realistic eyes, natural skin, no hands. The model wears a trench coat and ankle boots, standing on sidewalk, with a cozy autumn vibe", "Street":"Photo of a woman img, realistic eyes, natural skin, no hands. The model wears a scarf and long coat, standing on a street, creating a cozy mood.", "Moders Office":"Photo of a woman img, realistic eyes, natural skin, no hands. The model wears a tailored blazer with trousers, standing in a modern office.", "Posing by a desk":"Photo of a woman img, realistic eyes, natural skin, no hands. The model wears a shirt with wide-leg pants, posing by a desk.", "Resort Pool":"Photo of a woman img, realistic eyes, natural skin, no hands. The model wears a minimalist bikini, standing by a luxurious pool with natural sunlight, giving a resort vibe.", "Cozy":"Photo of a woman img, realistic eyes, natural skin, no hands. The model wears an oversized sweater lounging on a couch with a cozy blanket, relaxed atmosphere.", "Palm Breeze":"Photo of a woman img, realistic eyes, natural skin, no hands. The model wears a classic triangle bikini, standing on a beach with palm trees and clear water.", "Beach Shore":"Photo of a woman img, realistic eyes, natural skin, no hands. The model wears a colored bikini with a sunhat, standing on the shore, beachy vibe.", "Vacation":"Photo of a woman img, realistic eyes, natural skin, no hands. The model wears a sleek one-piece bikini, vacation-ready vibe.", "Field":"Photo of a woman img, realistic eyes, natural skin, no hands. The model wears a bohemian-style maxi dress, standing in a vibrant field of wildflowers.", "Ocean Edge":"Photo of a woman img, realistic eyes, natural skin, no hands. The model wears a dress, standing on a cliff overlooking the ocean.", "Club Mood":"Photo of a woman img, realistic eyes, natural skin, no hands. The model wears a leather dress, standing against a dimly lit club wall" } def get_prompts_by_gender(gender): return man_prompts if gender == "man" else woman_prompts def update_theme_buttons_only(gender): prompts = get_prompts_by_gender(gender) updates = [] for i, theme in enumerate(prompts.keys()): if i < len(theme_buttons): updates.append(gr.update(value=theme, visible=True)) for j in range(len(prompts), len(theme_buttons)): updates.append(gr.update(visible=False)) return updates def update_prompt_on_gender_change(gender, theme_name, current_prompt): button_updates = update_theme_buttons_only(gender) if theme_name and theme_name in get_prompts_by_gender(gender): # regenerate prompt for same theme but new gender new_prompt = get_prompts_by_gender(gender)[theme_name] else: new_prompt = current_prompt # ✅ return prompt first, then button updates (match outputs length) return [new_prompt] + button_updates def get_prompt_from_theme(gender, theme): return get_prompts_by_gender(gender)[theme], gr.update(visible=False) def update_suggestion_buttons(gender): prompts = get_prompts_by_gender(gender) return gr.update(visible=True), prompts def make_click_fn(theme_name): return lambda gender: get_prompt_from_theme(gender, theme_name) theme_buttons=[] for i, btn in enumerate(theme_buttons): theme = list(man_prompts.keys())[i] btn.click( fn=make_click_fn(theme), inputs=gender_selector, outputs=[prompt, prompt_suggestions] ) ### Description and style logo = r""" #
