# УРОК 4 — IP-ADAPTER + FACEID V2 (декабрь 2025)

# 1. Установка (5–7 минут один раз)
!pip install -q diffusers transformers accelerate insightface onnxruntime-gpu
!pip install -q torch==2.3.0+cu121 torchvision --extra-index-url https://download.pytorch.org/whl/cu121

import torch
from diffusers import StableDiffusionXLPipeline
from diffusers.utils import load_image
from insightface.app import FaceAnalysis
import cv2
import numpy as np
from IPython.display import display

print("Готовим FaceID...")

# 2. Загружаем модели
pipe = StableDiffusionXLPipeline.from_pretrained(
    "stabilityai/stable-diffusion-xl-base-1.0",
    torch_dtype=torch.float16,
    variant="fp16"
).to("cuda")

# IP-Adapter + FaceID
pipe.load_ip_adapter("h94/IP-Adapter", subfolder="sdxl_models", weight_name="ip-adapter-plus-face_sdxl_vit-h.safetensors")
pipe.load_ip_adapter("h94/IP-Adapter-FaceID", subfolder="sdxl_models", weight_name="ip-adapter-faceid-plusv2_sdxl.bin")

app = FaceAnalysis(name="buffalo_l", providers=['CUDAExecutionProvider'])
app.prepare(ctx_id=0, det_size=(640, 640))

print("Готово! Загрузи своё селфи")

# 3. Загрузи своё селфи (замени ссылку!)
face_image = load_image("https://i.imgur.com/your-selfie.jpg")  # ←←← СЮДА СВОЁ ФОТО!
face_image = face_image.convert("RGB")

# 4. Извлекаем FaceID
faces = app.get(cv2.cvtColor(np.array(face_image), cv2.COLOR_RGB2BGR))
faceid_embeds = torch.from_numpy(faces[0].normed_embedding).unsqueeze(0)

# 5. Твой промпт
prompt = "портрет той же девушки в стиле Studio Ghibli, аниме, высокое качество, детализация"
negative_prompt = "размыто, уродливо, деформированное лицо"

# 6. Генерация
image = pipe(
    prompt=prompt,
    negative_prompt=negative_prompt,
    ip_adapter_image=face_image,
    faceid_embeds=faceid_embeds,
    num_inference_steps=30,
    guidance_scale=5.0,
    controlnet_conditioning_scale=0.8
).images[0]

display(image)
print("Готово! Меняй промпт и запускай снова")