# УРОК 3 — CONTROLNET 2025 (работает на бесплатной T4)

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

import torch
from diffusers import StableDiffusionControlNetPipeline, ControlNetModel
from diffusers.utils import load_image
from PIL import Image
import numpy as np

print("Готово! Загружаем ControlNet...")

# 2. Загружаем ControlNet + Stable Diffusion
controlnet = ControlNetModel.from_pretrained("lllyasviel/sd-controlnet-openpose", torch_dtype=torch.float16)
pipe = StableDiffusionControlNetPipeline.from_pretrained(
    "runwayml/stable-diffusion-v1-5",
    controlnet=controlnet,
    torch_dtype=torch.float16,
    safety_checker=None
).to("cuda")

pipe.enable_xformers_memory_efficient_attention()
pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)

# 3. Загружаем своё фото (или любое другое)
# Замени ссылку на свою фотографию!
image = load_image("https://i.imgur.com/8z5jK8P.png")  # пример с человеком
image = image.resize((768, 768))

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

# 5. Генерация по позе из фото
result = pipe(
    prompt,
    image=image,
    num_inference_steps=30,
    guidance_scale=7.5,
    controlnet_conditioning_scale=1.0
).images[0]

display(result)
print("Готово! Попробуй своё фото!")