Update app.py

app.py

@@ -4,14 +4,15 @@ import spaces
 from diffusers import StableDiffusionXLPipeline, DPMSolverMultistepScheduler
 from huggingface_hub import hf_hub_download
 import random
+import re
 
-# ── Model ─────────────────────────────────────────────────────────────────────
-MODEL_REPO = "cyberdelia/CyberRealisticPony"
-MODEL_FILE = "CyberRealisticPony_V16.0_FP16.safetensors"
-PONY_POS   = "score_9, score_8_up, score_7_up, "
-PONY_NEG   = "score_6, score_5, score_4, "
+# ── Model — CyberIllustrious V8 ───────────────────────────────────────────────
+MODEL_REPO = "cyberdelia/latest_sdxl_models"
+MODEL_FILE = "CyberIllustrious_V8.0alt.safetensors"
+IL_POS     = "masterpiece, best quality, very aesthetic, absurdres, "
+IL_NEG     = "worst quality, low quality, bad quality, ugly, "
 
-print("Downloading model...")
+print("Downloading CyberIllustrious V8...")
 local_path = hf_hub_download(repo_id=MODEL_REPO, filename=MODEL_FILE)
 print("Loading pipeline...")
 pipe = StableDiffusionXLPipeline.from_single_file(local_path, torch_dtype=torch.float16)
@@ -21,85 +22,196 @@ pipe.scheduler = DPMSolverMultistepScheduler.from_config(
 pipe.enable_attention_slicing()
 print("Ready.")
 
-# ── Style presets (Fooocus-style) ─────────────────────────────────────────────
-STYLES = {
-    "None": {
-        "pos": "",
-        "neg": "",
+# ── Smart prompt expansion (Fooocus-style) ─────────────────────────────────────
+# Detects scene type and expands with the right vocabulary.
+# Specific unique details (numbers, colours, states) get attention weights.
+
+SCENE_TAGS = {
+    "portrait": {
+        "keywords": ["woman", "man", "girl", "boy", "person", "face", "portrait",
+                     "lady", "guy", "model", "character"],
+        "pos": "sharp focus on face, detailed eyes, professional photography, "
+               "85mm lens, f/1.8 bokeh, skin texture, natural lighting, ",
+        "neg": "wide angle distortion, bad eyes, asymmetric face, blurry face, ",
+    },
+    "architecture": {
+        "keywords": ["building", "house", "tower", "city", "street", "bridge",
+                     "castle", "cathedral", "skyscraper", "facade", "structure"],
+        "pos": "architectural photography, sharp geometry, detailed textures, "
+               "golden hour lighting, wide angle lens, high resolution, ",
+        "neg": "distorted perspective, blurry, watermarks, ",
+    },
+    "landscape": {
+        "keywords": ["mountain", "forest", "ocean", "beach", "valley", "sky",
+                     "field", "river", "lake", "landscape", "nature", "countryside"],
+        "pos": "landscape photography, epic vista, golden hour, volumetric light, "
+               "sharp foreground, atmospheric perspective, 16mm lens, ",
+        "neg": "oversaturated, blurry horizon, flat lighting, ",
     },
-    "📸 Photorealistic": {
-        "pos": "RAW photo, photorealistic, hyperrealistic, 8k uhd, dslr, high quality, film grain, Fujifilm XT3, sharp focus, detailed skin texture, natural lighting, ",
-        "neg": "painting, illustration, cartoon, anime, render, cgi, drawing, ",
+    "interior": {
+        "keywords": ["room", "interior", "bedroom", "kitchen", "office",
+                     "living room", "hallway", "bathroom", "studio", "cafe"],
+        "pos": "interior photography, ambient lighting, detailed surfaces, "
+               "realistic materials, depth of field, architectural digest style, ",
+        "neg": "fisheye distortion, dark, muddy colours, ",
+    },
+    "cinematic": {
+        "keywords": ["cinematic", "movie", "scene", "dramatic", "epic",
+                     "action", "night", "rain", "fog", "smoke"],
+        "pos": "cinematic shot, anamorphic lens, film grain, color graded, "
+               "dramatic lighting, shallow depth of field, movie still, ",
+        "neg": "flat lighting, amateur, snapshot, overexposed, ",
+    },
+}
+
+# Things that are specific and easy to lose — boost their weight
+SPECIFIC_PATTERNS = [
+    r'\bone\b', r'\btwo\b', r'\bthree\b', r'\bsingle\b',
+    r'\bopen\b', r'\bclosed\b', r'\bbroken\b', r'\bempty\b',
+    r'\bright\b', r'\bleft\b', r'\btop\b', r'\bbottom\b',
+    r'rainbow', r'red ', r'blue ', r'green ', r'yellow ',
+    r'purple ', r'pink ', r'black ', r'white ', r'golden ',
+]
+
+def detect_scene(prompt_lower):
+    scores = {}
+    for scene, data in SCENE_TAGS.items():
+        score = sum(1 for kw in data["keywords"] if kw in prompt_lower)
+        if score > 0:
+            scores[scene] = score
+    if not scores:
+        return None
+    return max(scores, key=scores.get)
+
+def boost_specific_details(prompt):
+    """
+    Wraps specific/unique details in attention weights so the model
+    doesn't gloss over them. e.g. 'one window open' -> '(one window open:1.4)'
+    """
+    boosted = prompt
+
+    # Find phrases containing specific words and wrap them
+    specific_words = [
+        r'\b(one|single)\s+\w+(\s+\w+)?',      # "one window", "single door open"
+        r'\b(open|closed|broken|cracked)\s+\w+', # "open window", "broken glass"
+        r'\b\w+\s+(rainbow|aurora|lightning)\b',  # "rainbow over", "lightning bolt"
+        r'\b(rainbow|aurora|lightning)\b',
+        r'\b(left|right)\s+\w+',                 # "left side", "right hand"
+    ]
+
+    for pattern in specific_words:
+        def wrap(m):
+            return f"({m.group(0)}:1.4)"
+        boosted = re.sub(pattern, wrap, boosted, flags=re.IGNORECASE)
+
+    return boosted
+
+def expand_prompt(raw_prompt, style_choice):
+    """
+    Takes a short natural language prompt and expands it Fooocus-style.
+    Returns (expanded_positive, extra_negative)
+    """
+    prompt_lower = raw_prompt.lower()
+
+    # Detect scene
+    scene = detect_scene(prompt_lower)
+
+    # Start with quality prefix (added later outside this fn)
+    extra_pos = ""
+    extra_neg = ""
+
+    # Add scene vocabulary
+    if scene and style_choice == "Auto":
+        extra_pos += SCENE_TAGS[scene]["pos"]
+        extra_neg += SCENE_TAGS[scene]["neg"]
+
+    # Boost specific details in the original prompt
+    weighted_prompt = boost_specific_details(raw_prompt.strip())
+
+    # Add general realism boosters if no style override
+    if style_choice == "Auto":
+        extra_pos += "highly detailed, sharp focus, realistic, high resolution, "
+
+    return weighted_prompt, extra_pos, extra_neg
+
+
+# ── Style presets ─────────────────────────────────────────────────────────────
+STYLES = {
+    "Auto":        {"pos": "", "neg": ""},
+    "📸 Photo":    {
+        "pos": "RAW photo, photorealistic, DSLR, 8k uhd, film grain, Fujifilm XT3, ",
+        "neg": "painting, illustration, cartoon, anime, cgi, ",
     },
     "🎬 Cinematic": {
-        "pos": "cinematic shot, movie scene, dramatic lighting, shallow depth of field, anamorphic lens, film grain, color graded, professional cinematography, epic composition, ",
-        "neg": "static, flat lighting, amateur, snapshot, ",
+        "pos": "cinematic, movie still, anamorphic, film grain, color graded, dramatic lighting, ",
+        "neg": "flat, amateur, snapshot, overexposed, ",
     },
     "🖼️ Portrait": {
-        "pos": "professional portrait, studio lighting, bokeh background, sharp eyes, detailed face, perfect skin, 85mm lens, f/1.8, headshot, ",
-        "neg": "wide angle, fish eye, distorted face, bad eyes, cropped head, ",
+        "pos": "portrait, studio lighting, 85mm lens, bokeh, sharp eyes, detailed skin, ",
+        "neg": "wide angle, distorted, bad eyes, cropped, ",
     },
     "🌆 Neon City": {
-        "pos": "cyberpunk city, neon lights, rain reflections on pavement, night scene, atmospheric fog, vibrant colors, futuristic architecture, blade runner aesthetic, ",
-        "neg": "daytime, rural, nature, warm tones, bright sunshine, ",
+        "pos": "cyberpunk, neon lights, rain reflections, night scene, blade runner aesthetic, ",
+        "neg": "daytime, rural, warm tones, ",
     },
-    "🎨 Oil Painting": {
-        "pos": "oil painting, impressionist, visible brushstrokes, canvas texture, artistic masterpiece, museum quality, rich colors, painted by a master, ",
-        "neg": "photo, photograph, digital art, flat colors, ",
+    "✨ Fantasy":   {
+        "pos": "fantasy art, epic, magical atmosphere, volumetric lighting, concept art, artstation, ",
+        "neg": "modern, mundane, flat, ",
     },
-    "✨ Fantasy Epic": {
-        "pos": "fantasy art, epic scene, magical atmosphere, dramatic clouds, volumetric lighting, highly detailed, concept art, artstation, 4k, ",
-        "neg": "modern, contemporary, mundane, boring, flat, ",
+    "🎨 Painterly": {
+        "pos": "oil painting, impressionist, visible brushstrokes, canvas texture, museum quality, ",
+        "neg": "photo, digital art, flat colours, ",
     },
 }
 
-# ── LoRA presets ──────────────────────────────────────────────────────────────
+# ── LoRAs ─────────────────────────────────────────────────────────────────────
 LORAS = {
     "None": None,
-    "✋ Perfect Hands": {
+    "✋ Better Hands": {
         "repo": "WolfAether21/PONY-DIFFUSION-SDXL-LORA",
         "file": "Perfect Hands v2.safetensors",
-        "default_strength": 0.7,
+        "strength": 0.7,
     },
-    "🔍 Detail Enhancer": {
+    "🔍 More Detail": {
         "repo": "WolfAether21/PONY-DIFFUSION-SDXL-LORA",
         "file": "SDXL Detail.safetensors",
-        "default_strength": 0.6,
+        "strength": 0.6,
     },
 }
 
 # ── Generation ────────────────────────────────────────────────────────────────
 @spaces.GPU(duration=180)
-def generate(prompt, negative_prompt, style, lora_name, lora_strength,
-             width, height, steps, guidance, seed, randomize):
+def generate(raw_prompt, negative_prompt, style, lora_name, lora_strength,
+             width, height, steps, guidance, seed, randomize, show_expanded):
 
-    if not prompt.strip():
+    if not raw_prompt.strip():
         raise gr.Error("Please enter a prompt.")
 
     if randomize:
         seed = random.randint(0, 2**32 - 1)
     seed = int(seed)
 
-    # Build final prompt with style
-    style_data = STYLES.get(style, STYLES["None"])
-    final_pos = PONY_POS + style_data["pos"] + prompt.strip()
-    final_neg = PONY_NEG + style_data["neg"] + negative_prompt.strip()
+    # ── Smart expansion ──
+    weighted_prompt, scene_pos, scene_neg = expand_prompt(raw_prompt, style)
+    style_data = STYLES.get(style, STYLES["Auto"])
+
+    final_pos = IL_POS + style_data["pos"] + scene_pos + weighted_prompt
+    final_neg = IL_NEG + style_data["neg"] + scene_neg + negative_prompt.strip()
 
-    # Move to GPU
+    # ── Move to GPU ──
     pipe.to("cuda")
 
-    # Load LoRA if selected
+    # ── Load LoRA ──
     lora_loaded = False
     lora_data = LORAS.get(lora_name)
     if lora_data:
         try:
-            lora_path = hf_hub_download(repo_id=lora_data["repo"], filename=lora_data["file"])
-            pipe.load_lora_weights(lora_path)
+            lp = hf_hub_download(repo_id=lora_data["repo"], filename=lora_data["file"])
+            pipe.load_lora_weights(lp)
             pipe.fuse_lora(lora_scale=float(lora_strength))
             lora_loaded = True
-            print(f"LoRA loaded: {lora_name} @ {lora_strength}")
         except Exception as e:
-            print(f"LoRA load failed, continuing without: {e}")
+            print(f"LoRA failed, skipping: {e}")
 
     generator = torch.Generator(device="cpu").manual_seed(seed)
 
@@ -111,367 +223,356 @@ def generate(prompt, negative_prompt, style, lora_name, lora_strength,
         num_inference_steps=int(steps),
         guidance_scale=float(guidance),
         generator=generator,
-        clip_skip=2,
+        clip_skip=1,
     )
 
-    # Unload LoRA and free GPU
     if lora_loaded:
         pipe.unfuse_lora()
         pipe.unload_lora_weights()
 
     pipe.to("cpu")
 
-    return result.images[0], seed
+    # Show expanded prompt if toggled on
+    expanded_text = f"**Sent to model:**\n{final_pos}" if show_expanded else ""
+
+    return result.images[0], seed, expanded_text
+
 
 # ── CSS ───────────────────────────────────────────────────────────────────────
 css = """
 * { box-sizing: border-box; margin: 0; padding: 0; }
 
 body, .gradio-container {
-    background: #080810 !important;
+    background: #07070e !important;
     font-family: 'Inter', system-ui, -apple-system, sans-serif !important;
-    max-width: 480px !important;
+    max-width: 500px !important;
     margin: 0 auto !important;
-    padding: 10px !important;
+    padding: 8px !important;
 }
 
-/* ── Top bar ── */
+/* ── Topbar ── */
 .topbar {
     display: flex;
     align-items: center;
     justify-content: space-between;
-    padding: 12px 4px;
-    margin-bottom: 10px;
+    padding: 10px 2px 14px;
 }
 .topbar-title {
-    color: #fff;
-    font-size: 1.0em;
+    color: #e8e0ff;
+    font-size: 0.95em;
     font-weight: 800;
     letter-spacing: -0.3px;
 }
-.topbar-badge {
-    background: #1aff7a;
-    color: #000;
+.gpu-pill {
+    background: #1aff7a18;
+    border: 1px solid #1aff7a44;
+    color: #1aff7a;
     font-size: 0.6em;
-    font-weight: 900;
-    padding: 3px 10px;
+    font-weight: 800;
+    padding: 4px 12px;
     border-radius: 20px;
-    letter-spacing: 1px;
+    letter-spacing: 1.5px;
     text-transform: uppercase;
 }
 
-/* ── Output image ── */
-.img-output {
-    background: #0e0e18;
-    border: 1px solid #1c1c2e;
-    border-radius: 18px;
+/* ── Image output ── */
+.img-out {
+    background: #0d0d1a;
+    border: 1px solid #16162a;
+    border-radius: 20px;
     overflow: hidden;
-    margin-bottom: 10px;
-    aspect-ratio: 3/4;
+    margin-bottom: 8px;
+    min-height: 380px;
+    position: relative;
     display: flex;
     align-items: center;
     justify-content: center;
-    position: relative;
 }
-.img-output img {
+.img-out img {
     width: 100% !important;
-    height: 100% !important;
-    object-fit: contain;
-    border-radius: 18px;
+    border-radius: 20px;
+    display: block;
 }
-/* Empty state placeholder */
-.img-output .empty {
-    color: #2a2a3e;
-    font-size: 3em;
+
+/* ── Seed pill under image ── */
+.seed-pill {
+    text-align: center;
+    margin-bottom: 12px;
+}
+.seed-pill input[type=number] {
+    background: transparent !important;
+    border: none !important;
+    color: #2e2848 !important;
+    font-size: 0.7em !important;
+    text-align: center !important;
+    padding: 0 !important;
+    width: 100% !important;
+    pointer-events: none;
 }
 
-/* ── Prompt area ── */
-.prompt-wrap {
-    background: #0e0e18;
-    border: 1px solid #1c1c2e;
+/* ── Card ── */
+.card {
+    background: #0d0d1a;
+    border: 1px solid #16162a;
     border-radius: 14px;
-    padding: 12px;
+    padding: 14px;
     margin-bottom: 8px;
-    transition: border-color 0.2s;
-}
-.prompt-wrap:focus-within {
-    border-color: #5533aa;
 }
-.prompt-label {
-    color: #44334a;
-    font-size: 0.65em;
-    font-weight: 700;
+.card-label {
+    color: #3d3060;
+    font-size: 0.62em;
+    font-weight: 800;
     text-transform: uppercase;
-    letter-spacing: 1.5px;
-    margin-bottom: 6px;
+    letter-spacing: 2px;
+    margin-bottom: 8px;
 }
 
+/* ── Prompt textarea ── */
 textarea {
     background: transparent !important;
     border: none !important;
-    border-radius: 0 !important;
-    color: #c8b8e8 !important;
-    font-size: 14px !important;
-    line-height: 1.5 !important;
+    color: #c8b8f0 !important;
+    font-size: 15px !important;
+    line-height: 1.6 !important;
     padding: 0 !important;
     resize: none !important;
     box-shadow: none !important;
     width: 100% !important;
+    outline: none !important;
 }
+textarea::placeholder { color: #252038 !important; }
 textarea:focus {
     outline: none !important;
     box-shadow: none !important;
-    border: none !important;
 }
-textarea::placeholder { color: #2e2640 !important; }
 
 /* ── Style pills ── */
-.style-pills {
-    margin-bottom: 8px;
-}
-.style-pills .gr-radio-group {
+.style-wrap .gr-radio {
     display: flex !important;
     flex-wrap: wrap !important;
     gap: 6px !important;
 }
-.style-pills label {
-    background: #0e0e18 !important;
-    border: 1px solid #1c1c2e !important;
-    border-radius: 20px !important;
-    color: #7766aa !important;
-    cursor: pointer !important;
-    font-size: 0.78em !important;
+.style-wrap label {
+    background: #0d0d1a !important;
+    border: 1px solid #1a1a2e !important;
+    border-radius: 30px !important;
+    color: #4a3a6a !important;
+    font-size: 0.75em !important;
     font-weight: 600 !important;
     padding: 6px 14px !important;
-    transition: all 0.15s !important;
+    cursor: pointer !important;
+    transition: all 0.15s ease !important;
     white-space: nowrap !important;
 }
-.style-pills label:has(input:checked) {
-    background: #1e0e3e !important;
-    border-color: #6633cc !important;
-    color: #cc99ff !important;
+.style-wrap label:has(input:checked) {
+    background: #18083a !important;
+    border-color: #7744ee !important;
+    color: #bb99ff !important;
+    box-shadow: 0 0 10px #7744ee33 !important;
 }
-.style-pills input[type=radio] { display: none !important; }
+.style-wrap input[type=radio] { display: none !important; }
 
 /* ── Accordion ── */
 .gradio-accordion {
-    background: #0e0e18 !important;
-    border: 1px solid #1c1c2e !important;
+    background: #0d0d1a !important;
+    border: 1px solid #16162a !important;
     border-radius: 14px !important;
     margin-bottom: 8px !important;
     overflow: hidden !important;
 }
-.gradio-accordion > .label-wrap {
-    padding: 12px 16px !important;
-    color: #7766aa !important;
-    font-size: 0.78em !important;
+.gradio-accordion .label-wrap button {
+    color: #4a3a6a !important;
+    font-size: 0.72em !important;
     font-weight: 700 !important;
     text-transform: uppercase !important;
-    letter-spacing: 1px !important;
-}
-.gradio-accordion > .label-wrap:hover {
-    color: #aa88ff !important;
+    letter-spacing: 1.5px !important;
+    padding: 12px 16px !important;
 }
 
 /* ── Sliders ── */
 .gradio-slider {
-    padding: 4px 0 !important;
     background: transparent !important;
     border: none !important;
+    padding: 4px 0 10px !important;
 }
 input[type=range] {
-    accent-color: #6633cc !important;
+    accent-color: #6633bb !important;
     width: 100% !important;
-    height: 3px !important;
 }
-.gradio-slider span {
-    color: #7766aa !important;
-    font-size: 0.75em !important;
+.gradio-slider .wrap {
+    color: #6644aa !important;
+    font-size: 0.72em !important;
     font-weight: 600 !important;
 }
 
-/* ── Number input ── */
+/* ── Number inputs ── */
 input[type=number] {
     background: #0a0a14 !important;
-    border: 1px solid #1c1c2e !important;
+    border: 1px solid #18182a !important;
     border-radius: 10px !important;
-    color: #aa88ff !important;
+    color: #9977cc !important;
     font-size: 13px !important;
     padding: 8px 10px !important;
-    width: 100% !important;
 }
 
 /* ── Checkbox ── */
-input[type=checkbox] {
-    accent-color: #6633cc !important;
-    width: 16px !important;
-    height: 16px !important;
-}
+input[type=checkbox] { accent-color: #6633bb !important; }
 .gradio-checkbox label span {
-    color: #7766aa !important;
-    font-size: 0.78em !important;
+    color: #4a3a6a !important;
+    font-size: 0.75em !important;
     font-weight: 600 !important;
 }
 
-/* ── Dropdown (LoRA) ── */
+/* ── Dropdown ── */
 .gradio-dropdown {
     background: #0a0a14 !important;
-    border: 1px solid #1c1c2e !important;
+    border: 1px solid #18182a !important;
     border-radius: 10px !important;
 }
-.gradio-dropdown select, .gradio-dropdown input {
-    background: transparent !important;
-    color: #aa88ff !important;
-    font-size: 13px !important;
-}
 
-/* ── Seed row ── */
-.seed-row {
-    display: flex;
-    align-items: center;
-    gap: 10px;
-    margin-top: 4px;
+/* ── Expanded prompt box ── */
+.expanded-box {
+    background: #080814;
+    border: 1px solid #111122;
+    border-radius: 10px;
+    padding: 10px 12px;
+    color: #332255;
+    font-size: 0.7em;
+    line-height: 1.6;
+    font-family: monospace;
+    word-break: break-word;
+    min-height: 32px;
 }
-.seed-out input {
-    background: #0a0a14 !important;
-    border: 1px solid #111120 !important;
-    border-radius: 8px !important;
-    color: #332255 !important;
-    font-size: 0.75em !important;
-    text-align: center !important;
-    padding: 6px !important;
+
+/* ── Labels ── */
+label > span:first-child {
+    color: #3a2d55 !important;
+    font-size: 0.7em !important;
+    font-weight: 700 !important;
+    text-transform: uppercase !important;
+    letter-spacing: 1px !important;
 }
 
 /* ── Generate button ── */
 .gen-btn button {
-    background: linear-gradient(135deg, #5522aa 0%, #3a1580 100%) !important;
-    border: 1px solid #7744cc !important;
+    background: linear-gradient(135deg, #4a1aaa 0%, #2d0e77 100%) !important;
+    border: 1px solid #6633cc !important;
     border-radius: 14px !important;
     color: #fff !important;
-    font-size: 0.95em !important;
-    font-weight: 800 !important;
-    padding: 16px !important;
+    font-size: 0.88em !important;
+    font-weight: 900 !important;
+    padding: 17px !important;
     width: 100% !important;
-    letter-spacing: 1.5px !important;
+    letter-spacing: 2px !important;
     text-transform: uppercase !important;
-    box-shadow: 0 4px 20px #5522aa44 !important;
-    transition: all 0.15s !important;
-    margin-top: 4px !important;
+    box-shadow: 0 4px 24px #4a1aaa55 !important;
+    transition: all 0.15s ease !important;
+    margin-top: 6px !important;
 }
 .gen-btn button:hover {
-    box-shadow: 0 6px 28px #5522aa88 !important;
+    box-shadow: 0 6px 32px #4a1aaa99 !important;
     transform: translateY(-1px) !important;
 }
 .gen-btn button:active {
     transform: scale(0.98) !important;
-    box-shadow: 0 2px 10px #5522aa33 !important;
+    box-shadow: 0 2px 12px #4a1aaa33 !important;
 }
 
-/* ── Labels ── */
-label > span:first-child {
-    color: #55446a !important;
-    font-size: 0.72em !important;
-    font-weight: 700 !important;
-    text-transform: uppercase !important;
-    letter-spacing: 1px !important;
-}
-
-footer, .built-with, .svelte-1ipelgc { display: none !important; }
+footer, .built-with { display: none !important; }
 """
 
 # ── UI ────────────────────────────────────────────────────────────────────────
 with gr.Blocks(css=css, title="ImageGen") as demo:
 
-    # Top bar
     gr.HTML("""
     <div class="topbar">
-        <span class="topbar-title">🐴 CyberRealistic Pony</span>
-        <span class="topbar-badge">⚡ GPU</span>
+        <span class="topbar-title">CyberIllustrious</span>
+        <span class="gpu-pill">⚡ ZeroGPU</span>
     </div>
     """)
 
     # Output
     output_image = gr.Image(
-        show_label=False,
-        type="pil",
-        height=440,
-        elem_classes="img-output",
+        show_label=False, type="pil",
+        height=460, elem_classes="img-out",
     )
-
-    # Seed used (small, below image)
     used_seed = gr.Number(
-        label="Last seed",
-        interactive=False,
-        elem_classes="seed-out",
+        label="seed", interactive=False,
+        elem_classes="seed-pill",
     )
 
-    # Prompt
-    gr.HTML('<div class="prompt-label">Prompt</div>')
+    # Prompt card
+    gr.HTML('<div class="card"><div class="card-label">Prompt</div>')
     prompt = gr.Textbox(
         show_label=False,
-        placeholder="describe your image...",
+        placeholder="describe anything — short or long, it gets expanded automatically...",
         lines=3,
-        elem_classes="prompt-wrap",
     )
+    gr.HTML('</div>')
 
-    # Negative
-    gr.HTML('<div class="prompt-label" style="margin-top:8px">Negative</div>')
-    negative_prompt = gr.Textbox(
-        show_label=False,
-        value=(
-            "(worst quality:1.2), (low quality:1.2), (normal quality:1.2), "
-            "lowres, bad anatomy, bad hands, signature, watermarks, "
-            "ugly, imperfect eyes, skewed eyes, unnatural face, "
-            "unnatural body, error, extra limb, missing limbs"
-        ),
-        lines=2,
-        elem_classes="prompt-wrap",
-    )
-
-    # Style presets
-    gr.HTML('<div class="prompt-label" style="margin-top:8px">Style</div>')
+    # Style pills
+    gr.HTML('<div class="card-label" style="padding:4px 2px 8px;color:#3d3060;font-size:0.62em;font-weight:800;text-transform:uppercase;letter-spacing:2px;">Style</div>')
     style = gr.Radio(
         choices=list(STYLES.keys()),
-        value="None",
+        value="Auto",
         show_label=False,
-        elem_classes="style-pills",
+        elem_classes="style-wrap",
+    )
+
+    # Generate
+    generate_btn = gr.Button(
+        "Generate ✦", variant="primary",
+        size="lg", elem_classes="gen-btn",
     )
 
-    # Advanced settings accordion
-    with gr.Accordion("⚙️  Advanced Settings", open=False):
-        gr.HTML('<div style="height:8px"></div>')
+    # Advanced accordion
+    with gr.Accordion("⚙️  Settings", open=False):
+        gr.HTML('<div style="height:6px"></div>')
+
+        negative_prompt = gr.Textbox(
+            label="Negative Prompt",
+            value=(
+                "worst quality, low quality, bad anatomy, bad hands, "
+                "signature, watermarks, ugly, blurry, deformed"
+            ),
+            lines=2,
+        )
 
         with gr.Row():
-            width  = gr.Slider(512, 896,  value=832,  step=64, label="Width")
-            height = gr.Slider(512, 1152, value=1024, step=64, label="Height")
+            width  = gr.Slider(512, 1024, value=832,  step=64, label="Width")
+            height = gr.Slider(512, 1216, value=1216, step=64, label="Height")
 
         steps    = gr.Slider(20, 60,   value=30,  step=1,   label="Steps")
-        guidance = gr.Slider(1.0, 12.0, value=5.0, step=0.5, label="CFG Scale")
+        guidance = gr.Slider(1.0, 10.0, value=5.0, step=0.5, label="CFG Scale")
 
         with gr.Row():
-            seed           = gr.Number(label="Seed", value=42, precision=0,
-                                       minimum=0, maximum=2**32-1, scale=3)
-            randomize_seed = gr.Checkbox(label="Random seed", value=True, scale=1)
+            seed = gr.Number(
+                label="Seed", value=42, precision=0,
+                minimum=0, maximum=2**32-1, scale=3,
+            )
+            randomize = gr.Checkbox(label="Random seed", value=True, scale=1)
+
+        show_expanded = gr.Checkbox(
+            label="Show expanded prompt (debug)",
+            value=False,
+        )
 
     # LoRA accordion
     with gr.Accordion("🎨  LoRA", open=False):
-        gr.HTML('<div style="height:8px"></div>')
-        gr.HTML('<div style="color:#554466;font-size:0.72em;margin-bottom:10px;">Add a LoRA to tweak style or fix common issues like hands.</div>')
-        lora_name     = gr.Dropdown(
-            choices=list(LORAS.keys()),
-            value="None",
-            label="LoRA",
-        )
+        gr.HTML('<div style="height:6px"></div>')
+        lora_name     = gr.Dropdown(choices=list(LORAS.keys()), value="None", label="LoRA preset")
         lora_strength = gr.Slider(0.1, 1.0, value=0.7, step=0.05, label="LoRA Strength")
 
-    # Generate
-    generate_btn = gr.Button("Generate ✦", variant="primary", size="lg", elem_classes="gen-btn")
+    # Expanded prompt debug output
+    expanded_out = gr.Markdown(elem_classes="expanded-box")
 
     generate_btn.click(
         fn=generate,
-        inputs=[prompt, negative_prompt, style, lora_name, lora_strength,
-                width, height, steps, guidance, seed, randomize_seed],
-        outputs=[output_image, used_seed],
+        inputs=[
+            prompt, negative_prompt, style, lora_name, lora_strength,
+            width, height, steps, guidance, seed, randomize, show_expanded,
+        ],
+        outputs=[output_image, used_seed, expanded_out],
     )
 
 demo.launch()