KiaraSasatImage Setup.py

KiaraSasatImage Setup.py

@@ -0,0 +1,439 @@
+#!/usr/bin/env python3
+"""
+========================================================================
+ KiaraSuperSpace v2 — ПОЛНЫЙ SETUP
+========================================================================
+ Скачивает модели + устанавливает кастомные ноды + создаёт start script.
+ 
+ Checkpoint: GonzaLomo v6.0 Photo XL DMD (LCM/Karras/cfg1)
+ LoRA: Super Skin Detailer + Realism Lora v3
+ Stage II: FLUX Klein 4B
+ Custom Nodes: KiaraPanels (ControlPanelSD + ControlPanelFlux + KiaraReferenceLatent)
+ 
+ Запуск:
+   python3 /workspace/setup_KiaraSuperSpace.py
+========================================================================
+"""
+
+import os, subprocess, time, textwrap
+
+HF_TOKEN = "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"
+CIVITAI_TOKEN = "c371cc55e5bc388aba3083f9ad9b2cb7"
+B = "/workspace/ComfyUI/models"
+CN = "/workspace/ComfyUI/custom_nodes"
+
+# =====================================================================
+# МОДЕЛИ
+# =====================================================================
+HF = [
+    ("Comfy-Org/flux2-klein-4B", "split_files/diffusion_models/flux-2-klein-4b.safetensors",
+     f"{B}/diffusion_models", "flux-2-klein-4b.safetensors"),
+    ("Comfy-Org/flux2-klein-4B", "split_files/text_encoders/qwen_3_4b.safetensors",
+     f"{B}/text_encoders", "qwen_3_4b.safetensors"),
+    ("Comfy-Org/flux2-klein-4B", "split_files/vae/flux2-vae.safetensors",
+     f"{B}/vae", "flux2-vae.safetensors"),
+    ("Kim2091/UltraSharp", "4x-UltraSharp.pth",
+     f"{B}/upscale_models", "4x-UltraSharp.pth"),
+]
+
+DIRECT = [
+    ("https://dl.fbaipublicfiles.com/segment_anything/sam_vit_b_01ec64.pth",
+     f"{B}/sams", "sam_vit_b_01ec64.pth", "SAM ViT-B"),
+    ("https://huggingface.co/Bingsu/adetailer/resolve/main/face_yolov8m.pt",
+     f"{B}/ultralytics/bbox", "face_yolov8m.pt", "Face YOLOv8m"),
+    ("https://huggingface.co/Bingsu/adetailer/resolve/main/hand_yolov8s.pt",
+     f"{B}/ultralytics/bbox", "hand_yolov8s.pt", "Hand YOLOv8s"),
+    ("https://huggingface.co/Bingsu/adetailer/resolve/main/person_yolov8m-seg.pt",
+     f"{B}/ultralytics/segm", "person_yolov8m-seg.pt", "Person Segmenter"),
+]
+
+CIVITAI = [
+    (f"https://civitai.com/api/download/models/2368123?type=Model&format=SafeTensor&size=pruned&fp=fp16&token={CIVITAI_TOKEN}",
+     f"{B}/checkpoints", "gonzalomoXLFluxPony_v60PhotoXLDMD.safetensors",
+     "GonzaLomo v6.0 Photo XL DMD (checkpoint)", 3000),
+    (f"https://civitai.com/api/download/models/1089573?token={CIVITAI_TOKEN}",
+     f"{B}/loras", None, "Super Skin Detailer (LoRA)", 50),
+    (f"https://civitai.com/api/download/models/2074888?token={CIVITAI_TOKEN}",
+     f"{B}/loras", None, "Realism Lora By Stable Yogi v3 (LoRA)", 50),
+]
+
+# =====================================================================
+# КАСТОМНАЯ НОДА — KiaraPanels
+# =====================================================================
+KIARA_PANELS_CODE = textwrap.dedent('''\
+"""
+KiaraPanels - Control Panel nodes for Stage I (SD) and Stage II (Flux) workflows.
+Drop this folder into ComfyUI/custom_nodes/ to use.
+"""
+
+# ============================================================
+# Stage I - SD Control Panel
+# ============================================================
+class ControlPanelSD:
+    CATEGORY = "KiaraPanels"
+    FUNCTION = "execute"
+    RETURN_TYPES = ("INT", "INT", "INT", "FLOAT")
+    RETURN_NAMES = ("width", "height", "steps", "cfg")
+
+    SD_PRESETS = {
+        "1:1 Square 512x512": (512, 512),
+        "1:1 Square 768x768": (768, 768),
+        "1:1 Square 1024x1024": (1024, 1024),
+        "2:3 Portrait 512x768": (512, 768),
+        "2:3 Portrait 768x1152": (768, 1152),
+        "3:2 Landscape 768x512": (768, 512),
+        "3:2 Landscape 1152x768": (1152, 768),
+        "9:16 Tall 576x1024": (576, 1024),
+        "16:9 Wide 1024x576": (1024, 576),
+        "9:16 Tall 720x1280": (720, 1280),
+        "16:9 Wide 1280x720": (1280, 720),
+        "4:7 Tall 576x1008": (576, 1008),
+        "9:16 Tall 896x1536": (896, 1536),
+    }
+
+    QUALITY_PRESETS = {
+        "Draft 4 steps": 4,
+        "Fast 8 steps": 8,
+        "Normal 12 steps": 12,
+        "High Quality 20 steps": 20,
+        "Ultra 30 steps": 30,
+        "Maximum 50 steps": 50,
+    }
+
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "resolution": (list(cls.SD_PRESETS.keys()), {"default": "9:16 Tall 896x1536"}),
+                "quality": (list(cls.QUALITY_PRESETS.keys()), {"default": "Fast 8 steps"}),
+                "cfg": ("FLOAT", {"default": 1.0, "min": 1.0, "max": 30.0, "step": 0.5}),
+            }
+        }
+
+    def execute(self, resolution, quality, cfg):
+        w, h = self.SD_PRESETS[resolution]
+        steps = self.QUALITY_PRESETS[quality]
+        return (w, h, steps, cfg)
+
+
+# ============================================================
+# Stage II - Flux Control Panel
+# ============================================================
+class ControlPanelFlux:
+    CATEGORY = "KiaraPanels"
+    FUNCTION = "execute"
+    RETURN_TYPES = ("INT", "INT", "INT")
+    RETURN_NAMES = ("width", "height", "steps")
+
+    FLUX_PRESETS = {
+        "1:1 Square 1024x1024": (1024, 1024),
+        "3:4 Portrait 896x1152": (896, 1152),
+        "4:3 Landscape 1152x896": (1152, 896),
+        "9:16 Tall 768x1344": (768, 1344),
+        "9:16 Tall 896x1536": (896, 1536),
+        "16:9 Wide 1344x768": (1344, 768),
+        "2:3 Portrait 832x1216": (832, 1216),
+        "3:2 Landscape 1216x832": (1216, 832),
+    }
+
+    QUALITY_PRESETS = {
+        "Fast 4 steps": 4,
+        "Normal 8 steps": 8,
+        "High Quality 16 steps": 16,
+        "Ultra 28 steps": 28,
+    }
+
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "resolution": (list(cls.FLUX_PRESETS.keys()), {"default": "9:16 Tall 896x1536"}),
+                "quality": (list(cls.QUALITY_PRESETS.keys()), {"default": "Fast 4 steps"}),
+            }
+        }
+
+    def execute(self, resolution, quality):
+        w, h = self.FLUX_PRESETS[resolution]
+        steps = self.QUALITY_PRESETS[quality]
+        return (w, h, steps)
+
+
+# ============================================================
+# Stage II - Reference Latent for FLUX Klein
+# ============================================================
+class KiaraReferenceLatent:
+    """
+    Injects VAE-encoded reference latent into conditioning for FLUX Klein.
+    Replicates kx7_cb243e9c ("Reference Latent").
+    Uses the native FLUX reference_latents mechanism.
+    """
+    CATEGORY = "KiaraPanels"
+    FUNCTION = "execute"
+    RETURN_TYPES = ("CONDITIONING",)
+
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "conditioning": ("CONDITIONING",),
+            },
+            "optional": {
+                "latent": ("LATENT",),
+            }
+        }
+
+    def execute(self, conditioning, latent=None):
+        if latent is None:
+            return (conditioning,)
+        import copy
+        c = []
+        for t in conditioning:
+            d = copy.deepcopy(t[1])
+            ref = latent["samples"]
+            if "reference_latents" in d:
+                d["reference_latents"] = d["reference_latents"] + [ref]
+            else:
+                d["reference_latents"] = [ref]
+            c.append([t[0], d])
+        return (c,)
+
+
+# ============================================================
+# Registration
+# ============================================================
+NODE_CLASS_MAPPINGS = {
+    "ControlPanelSD": ControlPanelSD,
+    "ControlPanelFlux": ControlPanelFlux,
+    "KiaraReferenceLatent": KiaraReferenceLatent,
+}
+
+NODE_DISPLAY_NAME_MAPPINGS = {
+    "ControlPanelSD": "Control Panel - SD",
+    "ControlPanelFlux": "Control Panel - Flux",
+    "KiaraReferenceLatent": "Reference Latent (FLUX)",
+}
+''')
+
+# =====================================================================
+# УТИЛИТЫ
+# =====================================================================
+def fmt(b):
+    if b < 1024**2: return f"{b/1024:.1f} KB"
+    if b < 1024**3: return f"{b/1024**2:.1f} MB"
+    return f"{b/1024**3:.2f} GB"
+
+def ok(p, mb=1): return os.path.exists(p) and os.path.getsize(p) >= mb*1024*1024
+def md(p): os.makedirs(p, exist_ok=True)
+
+def install_hf():
+    try:
+        import huggingface_hub
+        print(f"  ✓ huggingface_hub {huggingface_hub.__version__}")
+    except ImportError:
+        pip = "/venv/main/bin/pip" if os.path.exists("/venv/main/bin/pip") else "pip3"
+        subprocess.run([pip, "install", "-q", "--break-system-packages", "huggingface_hub[hf_xet]"], check=True)
+        print("  ✓ Установлен")
+
+# =====================================================================
+# ЗАГРУЗЧИКИ
+# =====================================================================
+def dl_hf():
+    from huggingface_hub import hf_hub_download
+    print("\n" + "="*70 + "\n📦 HUGGINGFACE\n" + "="*70)
+    for i, (repo, fn, dd, dn) in enumerate(HF, 1):
+        dest = os.path.join(dd, dn)
+        print(f"\n[{i}/{len(HF)}] {dn}")
+        if ok(dest):
+            print(f"  ✅ ЕСТЬ ({fmt(os.path.getsize(dest))})")
+            continue
+        md(dd)
+        try:
+            t0 = time.time()
+            dl = hf_hub_download(repo_id=repo, filename=fn, local_dir="/tmp/hf_cache",
+                                 local_dir_use_symlinks=False, token=HF_TOKEN)
+            if dl != dest: subprocess.run(["mv", "-f", dl, dest], check=True)
+            print(f"  ✅ {fmt(os.path.getsize(dest))} за {time.time()-t0:.0f}с")
+        except Exception as e:
+            print(f"  ❌ {e}")
+
+def dl_direct():
+    print("\n" + "="*70 + "\n🔍 ДЕТЕКТОРЫ\n" + "="*70)
+    for i, (url, dd, dn, desc) in enumerate(DIRECT, 1):
+        dest = os.path.join(dd, dn)
+        print(f"\n[{i}/{len(DIRECT)}] {desc}")
+        if ok(dest):
+            print(f"  ✅ ЕСТЬ ({fmt(os.path.getsize(dest))})")
+            continue
+        md(dd)
+        h = ["--header", f"Authorization: Bearer {HF_TOKEN}"] if "huggingface" in url else []
+        try:
+            subprocess.run(["wget", "-q", "--show-progress", "-O", dest, *h, url], timeout=300)
+            if ok(dest): print(f"  ✅ {fmt(os.path.getsize(dest))}")
+            else:
+                if os.path.exists(dest): os.remove(dest)
+        except Exception as e:
+            print(f"  ❌ {e}")
+
+def dl_civitai():
+    print("\n" + "="*70 + "\n🎨 CIVITAI\n" + "="*70)
+    for url, dd, dest_name, desc, min_mb in CIVITAI:
+        print(f"\n  {desc}")
+        md(dd)
+        if dest_name:
+            dest = os.path.join(dd, dest_name)
+            if ok(dest, min_mb):
+                print(f"  ✅ ЕСТЬ ({fmt(os.path.getsize(dest))})")
+                continue
+            try:
+                t0 = time.time()
+                subprocess.run(["wget", "-q", "--show-progress", "-O", dest, url], timeout=900)
+                if ok(dest, min_mb):
+                    print(f"  ✅ {dest_name} — {fmt(os.path.getsize(dest))} за {time.time()-t0:.0f}с")
+                else:
+                    if os.path.exists(dest) and os.path.getsize(dest) < 1024:
+                        with open(dest, 'r', errors='ignore') as f:
+                            print(f"  ⚠️ Ошибка: {f.read()[:200]}")
+                        os.remove(dest)
+            except Exception as e:
+                print(f"  ❌ {e}")
+        else:
+            existing = set(os.listdir(dd))
+            try:
+                t0 = time.time()
+                subprocess.run(["wget", "-q", "--show-progress", "--content-disposition", "-P", dd, url], timeout=600)
+                new = set(os.listdir(dd)) - existing
+                for f in new:
+                    print(f"  ✅ {f} — {fmt(os.path.getsize(os.path.join(dd, f)))} за {time.time()-t0:.0f}с")
+                if not new:
+                    print(f"  ⚠️ Файл уже был")
+            except Exception as e:
+                print(f"  ❌ {e}")
+
+# =====================================================================
+# КАСТОМНЫЕ НОДЫ + CLIP COPY + START SCRIPT
+# =====================================================================
+def setup_custom_nodes():
+    print("\n" + "="*70 + "\n🔧 КАСТОМНЫЕ НОДЫ\n" + "="*70)
+    path = os.path.join(CN, "KiaraPanels", "__init__.py")
+    md(os.path.dirname(path))
+    with open(path, 'w') as f:
+        f.write(KIARA_PANELS_CODE)
+    print(f"  ✅ KiaraPanels → {path}")
+    print(f"     ControlPanelSD + ControlPanelFlux + KiaraReferenceLatent")
+
+def setup_clip_copy():
+    print("\n" + "="*70 + "\n📋 CLIP COPY\n" + "="*70)
+    src = f"{B}/text_encoders/qwen_3_4b.safetensors"
+    dst = f"{B}/clip/qwen_3_4b.safetensors"
+    md(os.path.dirname(dst))
+    if os.path.exists(dst) and os.path.getsize(dst) > 1024*1024*1000:
+        print(f"  ✅ ЕСТЬ ({fmt(os.path.getsize(dst))})")
+    elif os.path.exists(src):
+        import shutil
+        shutil.copy2(src, dst)
+        print(f"  ✅ Скопировал → models/clip/ ({fmt(os.path.getsize(dst))})")
+    else:
+        print(f"  ⚠️ Источник не найден, скопируй после загрузки")
+
+def setup_start_script():
+    print("\n" + "="*70 + "\n🚀 START SCRIPT\n" + "="*70)
+    script = textwrap.dedent("""\
+    #!/bin/bash
+    # KiaraSuperSpace — ComfyUI Start
+    export LD_LIBRARY_PATH=/venv/main/lib/python3.12/site-packages/nvidia/cuda_runtime/lib:$LD_LIBRARY_PATH
+    cd /workspace/ComfyUI
+    fuser -k 8188/tcp 2>/dev/null
+    sleep 1
+    /venv/main/bin/python main.py --listen --port 8188 &
+    echo "ComfyUI запущен на порту 8188"
+    """)
+    path = "/workspace/start_comfyui.sh"
+    with open(path, 'w') as f:
+        f.write(script)
+    os.chmod(path, 0o755)
+    print(f"  ✅ {path}")
+
+# =====================================================================
+# ИТОГ
+# =====================================================================
+def final():
+    print("\n" + "="*70 + "\n📋 ИТОГ\n" + "="*70)
+    total = 0
+
+    for _,_,dd,dn in HF:
+        p = os.path.join(dd,dn)
+        if os.path.exists(p):
+            sz = os.path.getsize(p); total += sz
+            print(f"  ✅ {dn:<55s} {fmt(sz):>10s}")
+        else:
+            print(f"  ❌ {dn:<55s} MISSING")
+
+    for _,dd,dn,_ in DIRECT:
+        p = os.path.join(dd,dn)
+        if os.path.exists(p):
+            sz = os.path.getsize(p); total += sz
+            print(f"  ✅ {dn:<55s} {fmt(sz):>10s}")
+        else:
+            print(f"  ❌ {dn:<55s} MISSING")
+
+    print()
+    for folder, label in [(f"{B}/checkpoints","checkpoints"), (f"{B}/loras","loras")]:
+        if os.path.exists(folder):
+            for f in sorted(os.listdir(folder)):
+                if f.endswith('.safetensors'):
+                    sz = os.path.getsize(os.path.join(folder,f)); total += sz
+                    print(f"  ✅ {label}/{f:<48s} {fmt(sz):>10s}")
+
+    print()
+    p = os.path.join(CN, "KiaraPanels", "__init__.py")
+    print(f"  {'✅' if os.path.exists(p) else '❌'} custom_nodes/KiaraPanels")
+
+    p2 = f"{B}/clip/qwen_3_4b.safetensors"
+    print(f"  {'✅' if os.path.exists(p2) and os.path.getsize(p2)>1e9 else '❌'} models/clip/qwen_3_4b.safetensors")
+
+    print(f"\n  💾 Общий размер: {fmt(total)}")
+    print("="*70)
+    print("""
+  📝 ПОСЛЕ УСТАНОВКИ:
+  
+  1. bash /workspace/start_comfyui.sh
+  2. Загрузить KiaraSasatImage.json
+  3. Выбрать из дропдауна:
+     Checkpoint: gonzalomoXLFluxPony_v60PhotoXLDMD
+     LoRA: Realism Lora или Super Skin Detailer
+  4. Reference 1 — Body (фото тела)
+     Reference 2 — Face (фото лица)
+  5. Run!
+  
+  Stage I: LCM, Karras, cfg=1, steps=8, 896x1536
+  Stage II: FLUX Klein, Euler, steps=4, face swap
+""")
+
+# =====================================================================
+def main():
+    print("="*70)
+    print("  🚀 KiaraSuperSpace v2 — ПОЛНЫЙ SETUP")
+    print("="*70)
+    print(f"  {time.strftime('%Y-%m-%d %H:%M:%S')}\n")
+
+    for d in ["checkpoints","loras","diffusion_models","text_encoders",
+              "vae","clip","upscale_models","sams","clip_vision",
+              "ultralytics/bbox","ultralytics/segm"]:
+        md(os.path.join(B, d))
+
+    print("🔧 Подготовка...")
+    install_hf()
+    dl_hf()
+    dl_direct()
+    dl_civitai()
+    setup_custom_nodes()
+    setup_clip_copy()
+    setup_start_script()
+    final()
+
+    import shutil
+    if os.path.exists("/tmp/hf_cache"):
+        shutil.rmtree("/tmp/hf_cache", ignore_errors=True)
+
+if __name__ == "__main__":
+    main()