Rename app.py.txt to app.py

app.py

@@ -0,0 +1,167 @@
+import os
+import json
+import time
+import random
+import uuid
+import requests
+import gradio as gr
+from io import BytesIO
+from PIL import Image
+from huggingface_hub import snapshot_download
+import shutil
+
+# === AUTO DOWNLOAD EVERYTHING FROM OFFICIAL REPO ===
+REPO_ID = "Tongyi-MAI/Z-Image-Turbo"
+LOCAL_DIR = "z-image-models"
+
+print("Downloading ALL files from Tongyi-MAI/Z-Image-Turbo... (this takes ~3-5 mins first time)")
+
+snapshot_download(
+    repo_id=REPO_ID,
+    local_dir=LOCAL_DIR,
+    local_dir_use_symlinks=False,
+    allow_patterns=["*.safetensors", "*.json", "*.yaml", "*.yml", "*.bin"],
+    ignore_patterns=["*.git*", "*.md"]
+)
+
+# === Organize into ComfyUI folders ===
+os.makedirs("models/checkpoints", exist_ok=True)
+os.makedirs("models/vae", exist_ok=True)
+os.makedirs("models/config", exist_ok=True)
+
+for file in os.listdir(LOCAL_DIR):
+    src = os.path.join(LOCAL_DIR, file)
+    if file.endswith(".safetensors") or file.endswith(".ckpt"):
+        if "vae" in file.lower():
+            shutil.copy(src, f"models/vae/{file}")
+        else:
+            shutil.copy(src, f"models/checkpoints/{file}")
+    elif file.endswith((".json", ".yaml", ".yml")):
+        shutil.copy(src, f"models/config/{file}")
+
+print("All Z-Image-Turbo models downloaded and placed correctly!")
+
+# === Start ComfyUI ===
+import subprocess
+import threading
+
+def start_comfyui():
+    if not os.path.exists("ComfyUI"):
+        print("Cloning ComfyUI...")
+        subprocess.run(["git", "clone", "https://github.com/comfyanonymous/ComfyUI.git"])
+    
+    os.chdir("ComfyUI")
+    subprocess.Popen([
+        "python", "main.py",
+        "--listen", "0.0.0.0",
+        "--port", "8188",
+        "--force-fp16",
+        "--disable-auto-launch"
+    ], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+    print("ComfyUI starting...")
+    time.sleep(25)
+
+threading.Thread(target=start_comfyui, daemon=True).start()
+
+COMFYUI_URL = "http://127.0.0.1:8188"
+WORKFLOW_PATH = "workflow.json"
+
+# === API Helpers ===
+def queue_prompt(workflow):
+    client_id = str(uuid.uuid4())
+    r = requests.post(f"{COMFYUI_URL}/prompt", json={"prompt": workflow, "client_id": client_id})
+    return r.json().get("prompt_id")
+
+def get_image_from_history(prompt_id):
+    while True:
+        try:
+            history = requests.get(f"{COMFYUI_URL}/history/{prompt_id}").json()
+            if prompt_id in history:
+                outputs = history[prompt_id]["outputs"]
+                for node_id in ["7", "14", "15"]:  # SaveImage / PreviewImage
+                    if node_id in outputs and "images" in outputs[node_id]:
+                        img_data = outputs[node_id]["images"][0]
+                        url = f"{COMFYUI_URL}/view?filename={img_data['filename']}&type={img_data['type']}&subfolder={img_data.get('subfolder','')}"
+                        return Image.open(BytesIO(requests.get(url).content))
+        except:
+            pass
+        time.sleep(0.6)
+
+# === Generation (100% random every time) ===
+def generate(
+    prompt, negative_prompt="",
+    seed=-1, steps=20, cfg=7.0,
+    width=1024, height=1024,
+    upscale=1.0, batch_size=1
+):
+    with open(WORKFLOW_PATH, "r", encoding="utf-8") as f:
+        workflow = json.load(f)
+
+    base_seed = random.randint(1, 999999999) if seed < 0 else int(seed)
+    cache_buster = random.random()
+
+    images = []
+    status_msg = ""
+
+    for i in range(batch_size):
+        current_seed = base_seed + i * 1000000
+
+        for node in workflow["nodes"]:
+            # Update prompts
+            if node["type"] == "CLIPTextEncode":
+                if "beautiful landscape" in node["widgets_values"][0]:
+                    node["widgets_values"][0] = prompt
+                if node["widgets_values"][0] == "":
+                    node["widgets_values"][0] = negative_prompt
+
+            # Update sampler
+            if node["type"] == "KSampler":
+                node["widgets_values"][0] = current_seed
+                node["widgets_values"][1] = steps
+                node["widgets_values"][2] = cfg
+
+            # Size
+            if node["type"] == "EmptyLatentImage":
+                node["widgets_values"][0] = width
+                node["widgets_values"][1] = height
+                node["widgets_values"][2] = 1
+
+            # Upscale
+            if node["type"] == "LatentUpscaleBy":
+                node["widgets_values"][0] = upscale
+
+            # Cache buster
+            node["_b"] = cache_buster + i
+
+        pid = queue_prompt(workflow)
+        if not pid:
+            yield "ComfyUI not ready yet...", images
+            return
+
+        img = get_image_from_history(pid)
+        if img:
+            images.append(img)
+
+        status_msg = f"Generated {i+1}/{batch_size} • Seed: {current_seed}"
+        yield status_msg, images
+
+    yield f"Complete! • Seeds: {base_seed} → {current_seed}", images
+
+# === Gradio UI ===
+with gr.Blocks(title="Z-Image Turbo • Official Models • Always Random") as demo:
+    gr.Markdown("# Z-Image Turbo — Official Full Download\n"
+                "Every file from https://huggingface.co/Tongyi-MAI/Z-Image-Turbo\n"
+                "100% random • Hi-Res • Batch • No duplicates ever")
+
+    with gr.Row():
+        with gr.Column(scale=2):
+            prompt = gr.Textbox(
+                label="Prompt", lines=4,
+                value="masterpiece, best quality, ultra-detailed, beautiful mountain lake at golden hour, cinematic lighting"
+            )
+            negative = gr.Textbox(label="Negative Prompt", value="blurry, ugly, deformed, low quality")
+
+        with gr.Column():
+            seed = gr.Number(label="Seed (-1 = random)", value=-1)
+            steps = gr.Slider(10, 40, value=20, step=1, label="Steps")
+            cfg = gr.Slider(3, 15, value=7.0, step=0.1, label

app.py.txt

@@ -1,192 +0,0 @@
-import os
-import json
-import time
-import random
-import uuid
-import requests
-import gradio as gr
-from io import BytesIO
-from PIL import Image
-from huggingface_hub import snapshot_download
-import shutil
-
-# === Auto-download ALL models from repo ===
-REPO_ID = "Tongyi-MAI/Z-Image-Turbo"
-MODELS_DIR = "models"
-os.makedirs(MODELS_DIR, exist_ok=True)
-
-# Download entire repo snapshot
-print("Downloading all files from Z-Image-Turbo repo...")
-snapshot_download(
-    repo_id=REPO_ID,
-    local_dir=MODELS_DIR,
-    local_dir_use_symlinks=False,
-    ignore_patterns=["*.git*", "*.md", "*.txt"]  # Skip non-model files
-)
-
-# Organize into ComfyUI folders
-checkpoints_dir = os.path.join(MODELS_DIR, "checkpoints")
-vae_dir = os.path.join(MODELS_DIR, "vae")
-os.makedirs(checkpoints_dir, exist_ok=True)
-os.makedirs(vae_dir, exist_ok=True)
-
-# Move .safetensors to checkpoints (main models)
-for file in os.listdir(MODELS_DIR):
-    if file.endswith((".safetensors", ".ckpt")):
-        shutil.move(os.path.join(MODELS_DIR, file), os.path.join(checkpoints_dir, file))
-
-# Move VAE files if present (e.g., vae.safetensors)
-for file in os.listdir(MODELS_DIR):
-    if "vae" in file.lower() and file.endswith((".safetensors", ".ckpt")):
-        shutil.move(os.path.join(MODELS_DIR, file), os.path.join(vae_dir, file))
-
-# Configs/JSON to custom folder for workflow use
-config_dir = os.path.join(MODELS_DIR, "config")
-os.makedirs(config_dir, exist_ok=True)
-for file in os.listdir(MODELS_DIR):
-    if file.endswith((".json", ".yaml", ".yml")):
-        shutil.move(os.path.join(MODELS_DIR, file), os.path.join(config_dir, file))
-
-print(f"All models downloaded to {MODELS_DIR}")
-
-# === ComfyUI Setup ===
-COMFYUI_URL = "http://127.0.0.1:8188"
-WORKFLOW_PATH = "workflow.json"
-
-# Start ComfyUI in background
-import threading
-import subprocess
-
-def start_comfyui():
-    if not os.path.exists("ComfyUI"):
-        subprocess.run(["git", "clone", "https://github.com/comfyanonymous/ComfyUI.git"])
-    os.chdir("ComfyUI")
-    subprocess.Popen([
-        "python", "main.py",
-        "--listen", "0.0.0.0", "--port", "8188",
-        "--disable-auto-launch", "--force-fp16"
-    ], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
-    print("Starting ComfyUI...")
-    time.sleep(20)
-
-threading.Thread(target=start_comfyui, daemon=True).start()
-time.sleep(5)
-
-# === ComfyUI API Helpers ===
-def queue_prompt(workflow):
-    client_id = str(uuid.uuid4())
-    response = requests.post(f"{COMFYUI_URL}/prompt", json={"prompt": workflow, "client_id": client_id})
-    return response.json().get("prompt_id")
-
-def get_images(prompt_id):
-    while True:
-        history = requests.get(f"{COMFYUI_URL}/history/{prompt_id}").json()
-        if prompt_id in history:
-            node = history[prompt_id]["outputs"]
-            # Find SaveImage or PreviewImage node (usually ID 7 or 14)
-            for node_id in ["7", "14", "15"]:  # SaveImage nodes
-                if node_id in node and "images" in node[node_id]:
-                    img_data = node[node_id]["images"][0]
-                    img_url = f"{COMFYUI_URL}/view?filename={img_data['filename']}&type={img_data['type']}"
-                    img = Image.open(BytesIO(requests.get(img_url).content))
-                    return [img]
-        time.sleep(0.5)
-
-# === Main Generation Function ===
-def generate_image(
-    prompt: str,
-    negative_prompt: str = "",
-    seed: int = -1,
-    steps: int = 20,
-    cfg: float = 7.0,
-    width: int = 1024,
-    height: int = 1024,
-    upscale_factor: float = 1.0,
-    batch_size: int = 1
-):
-    # Load workflow
-    with open(WORKFLOW_PATH, "r", encoding="utf-8") as f:
-        workflow = json.load(f)
-
-    # === CRITICAL: Force new seed + cache busting ===
-    actual_seed = random.randint(1, 999999999) if seed <= 0 else int(seed)
-    cache_buster = random.random()  # Forces ComfyUI to re-run everything
-
-    # Update nodes
-    for node in workflow["nodes"]:
-        if node["type"] == "CLIPTextEncode" and "beautiful landscape" in node["widgets_values"][0]:
-            node["widgets_values"][0] = prompt
-        if node["type"] == "CLIPTextEncode" and node["widgets_values"][0] == "":
-            node["widgets_values"][0] = negative_prompt
-
-        if node["type"] == "KSampler":
-            node["widgets_values"][0] = actual_seed
-            node["widgets_values"][1] = steps
-            node["widgets_values"][2] = cfg
-
-        if node["type"] == "EmptyLatentImage":
-            node["widgets_values"][0] = width
-            node["widgets_values"][1] = height
-            node["widgets_values"][2] = batch_size
-
-        # Hi-res upscale factor
-        if node["type"] == "LatentUpscaleBy":
-            node["widgets_values"][0] = upscale_factor
-
-        # Add dummy cache buster (harmless but forces re-execution)
-        if "widgets_values" in node and isinstance(node["widgets_values"], list):
-            node["_cache_buster"] = cache_buster
-
-    images = []
-    for i in range(batch_size):
-        # Unique seed per batch item
-        batch_seed = actual_seed + i * 1000000
-        for node in workflow["nodes"]:
-            if node["type"] == "KSampler":
-                node["widgets_values"][0] = batch_seed
-
-        prompt_id = queue_prompt(workflow)
-        if not prompt_id:
-            yield "Error: ComfyUI not responding", images
-            return
-
-        imgs = get_images(prompt_id)
-        images.extend(imgs)
-
-        yield f"Generated {i+1}/{batch_size} (Seed: {batch_seed})", images
-
-    yield f"Done! Final seed: {actual_seed}", images
-
-# === Gradio Interface ===
-with gr.Blocks(title="Z-Image Turbo — Full Repo Models") as demo:
-    gr.Markdown("# Z-Image Turbo — All Models Loaded\nFast SDXL with complete repo support • Always Random • Hi-Res")
-
-    with gr.Row():
-        with gr.Column(scale=2):
-            prompt = gr.Textbox(
-                label="Prompt",
-                lines=3,
-                value="a beautiful mountain landscape with a crystal lake at sunset, photorealistic, ultra detailed"
-            )
-            negative = gr.Textbox(label="Negative Prompt", lines=2, value="blurry, ugly, low quality")
-
-        with gr.Column():
-            seed = gr.Number(label="Seed (-1 = random)", value=-1)
-            steps = gr.Slider(10, 40, value=20, step=1, label="Steps")
-            cfg = gr.Slider(3, 15, value=7.0, step=0.1, label="CFG Scale")
-            width = gr.Slider(512, 1536, value=1024, step=64, label="Width")
-            height = gr.Slider(512, 1536, value=1024, step=64, label="Height")
-            upscale = gr.Slider(1.0, 3.0, value=1.5, step=0.5, label="Hi-Res Upscale (1.0 = off)")
-            batch = gr.Slider(1, 8, value=1, step=1, label="Batch Size")
-
-    generate_btn = gr.Button("Generate Images", variant="primary", size="lg")
-    status = gr.Textbox(label="Status", value="Ready")
-    gallery = gr.Gallery(label="Results", columns=3, height="auto")
-
-    generate_btn.click(
-        fn=generate_image,
-        inputs=[prompt, negative, seed, steps, cfg, width, height, upscale, batch],
-        outputs=[status, gallery]
-    )
-
-demo.launch(server_name="0.0.0.0", server_port=7860)