feat: enhance UI

.python-version

@@ -0,0 +1 @@
+3.10

app.py

@@ -6,27 +6,32 @@ import spaces
 from transformers import CLIPTokenizer
 
 import torch
-from diffusers import StableDiffusionXLPipeline
 
+DEV_MODE = os.getenv("DEV_MODE", "0") == "1"
+if DEV_MODE:
+    from mock import MockPipe
+    pipe = MockPipe()
+    device = "cpu"
+else:
+    from diffusers import StableDiffusionXLPipeline
+    device = "cuda"
+    model_repo_id = "Ine007/waiNSFWIllustrious_v140"
 
-device = "cuda" if torch.cuda.is_available() else "cpu"
-model_repo_id = "Ine007/waiNSFWIllustrious_v140"
+    torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
 
-torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
+    pipe = StableDiffusionXLPipeline.from_pretrained(
+        model_repo_id,
+        torch_dtype=torch_dtype,
+        use_safetensors=True,
+        add_watermarker=None,
+    ).to(device)
 
-pipe = StableDiffusionXLPipeline.from_pretrained(
-    model_repo_id,
-    torch_dtype=torch_dtype,
-    use_safetensors=True,
-    add_watermarker=None,
-).to(device)
-tokenizer = CLIPTokenizer.from_pretrained("openai/clip-vit-base-patch32")
-
-if torch.cuda.is_available():
-    torch.backends.cuda.matmul.allow_tf32 = True
+    if torch.cuda.is_available():
+        torch.backends.cuda.matmul.allow_tf32 = True
 
+tokenizer = CLIPTokenizer.from_pretrained("openai/clip-vit-base-patch32")
 MAX_SEED = np.iinfo(np.int32).max
-MAX_IMAGE_SIZE = 1024
+MAX_IMAGE_SIZE = 1344
 
 # ---- helpers ----
 def apply_preset(preset):
@@ -37,7 +42,7 @@ def apply_preset(preset):
         "1152×896 (landscape)": (1152, 896),
         "768×1344 (portrait, lighter)": (768, 1344),
     }
-    return mapping.get(preset, (768, 768))
+    return mapping.get(preset, (1024, 768))
 
 @spaces.GPU
 def infer(
@@ -120,11 +125,11 @@ examples = [
 ]
 
 css = """
-#col-container { margin: 0 auto; max-width: 1100px; width: 100%; padding: 0 12px; }
+#col-container { margin: 0 auto; max-width: 1250px; width: 100%; padding: 0 12px; }
 #left-col { position: sticky; top: 12px; align-self: start; }
 
 /* responsive image (FIT view) */
-#result_fit img { max-height: 85vh; width: auto; height: auto; }
+#result_fit img { max-height: 95vh; width: auto; height: auto; }
 
 /* 'Hide controls' toggle only visible on small screens */
 #hide_controls_row { display: none; }
@@ -138,21 +143,20 @@ with gr.Blocks(css=css) as demo:
     history_state = gr.State([])
 
     with gr.Column(elem_id="col-container"):
-        gr.Markdown("# SDXL Text-to-Image (waiNSFWIllustrious_v140)")
+        
 
         with gr.Row():
             # LEFT: controls
             with gr.Column(scale=1, elem_id="left-col"):
+                gr.Markdown("# SDXL Text-to-Image (waiNSFWIllustrious_v140)")
                 with gr.Row(elem_id="hide_controls_row"):
                     hide_controls_cb = gr.Checkbox(label="Hide controls (mobile)", value=False)
                 with gr.Group(visible=True) as controls_group:
                     prompt = gr.Text(
                         label="Prompt",
-                        show_label=False,
                         lines=2,
                         max_lines=6,
                         placeholder="Enter your prompt",
-                        container=False,
                         scale=1,
                         min_width=0,
                         autofocus=True,
@@ -161,17 +165,34 @@ with gr.Blocks(css=css) as demo:
                         label="Negative prompt",
                         max_lines=1,
                         placeholder="Enter a negative prompt",
-                        value="blurry, low quality, watermark",
+                        value="blurry, low quality, watermark, monochrome, text",
+                    )
+                    # generations = gr.Slider(
+                    # label="Generations",
+                    # minimum=1,
+                    # maximum=4,
+                    # step=1,
+                    # value=1,
+                    # info="Generates images one-by-one to avoid OOM."
+                    # )
+                    generations = gr.Slider(
+                        label="Generations",
+                        maximum=5,
+                        minimum=1,
+                        step=1,
+                        value=1,
+                        info="Control how many images are generated sequentially.",
+                    )
+                    no_rescale_cb = gr.Checkbox(
+                        label="Do not rescale to fit screen",
+                        value=False,
+                        info="Uncheck = fit preview to screen (default).",
+                        visible=True
                     )
                     run_button = gr.Button("Run", variant="primary")
 
             # RIGHT: image + toggle
             with gr.Column(scale=2):
-                no_rescale_cb = gr.Checkbox(
-                    label="Do not rescale to fit screen",
-                    value=False,
-                    info="Uncheck = fit preview to screen (default).",
-                )
                 # two image views: FIT (responsive) and RAW (no scaling)
                 result_fit = gr.Image(label="Result", show_label=False, elem_id="result_fit", visible=True)
                 result_raw = gr.Image(label="Result (original size)", show_label=False, visible=False)
@@ -181,27 +202,20 @@ with gr.Blocks(css=css) as demo:
             with gr.Row():
                 size_preset = gr.Dropdown(
                     ["768×768 (square)", "1024×1024", "832×1216 (portrait)", "1152×896 (landscape)", "768×1344 (portrait, lighter)"],
-                    value="768×768 (square)",
+                    value="1024×1024",
                     label="Size preset",
                 )
-                generations = gr.Slider(
-                    label="Generations",
-                    minimum=1,
-                    maximum=4,
-                    step=1,
-                    value=1,
-                    info="Generates images one-by-one to avoid OOM."
-                )
+
 
             seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=0)
             randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
 
             with gr.Row():
-                width = gr.Slider(label="Width", minimum=256, maximum=MAX_IMAGE_SIZE, step=32, value=768)
-                height = gr.Slider(label="Height", minimum=256, maximum=MAX_IMAGE_SIZE, step=32, value=768)
+                width = gr.Slider(label="Width", minimum=256, maximum=MAX_IMAGE_SIZE, step=32, value=1024)
+                height = gr.Slider(label="Height", minimum=256, maximum=MAX_IMAGE_SIZE, step=32, value=1024)
 
             with gr.Row():
-                guidance_scale = gr.Slider(label="Guidance scale", minimum=0.0, maximum=12.0, step=0.1, value=5.5)
+                guidance_scale = gr.Slider(label="Guidance scale", minimum=0.0, maximum=12.0, step=0.1, value=6)
                 num_inference_steps = gr.Slider(label="Number of inference steps", minimum=5, maximum=75, step=1, value=25)
 
             gr.Examples(examples=examples, inputs=[prompt])

makefile

@@ -0,0 +1,4 @@
+export DEV_MODE=1
+
+dev-gr:
+	uv run gradio app.py

mock.py

@@ -0,0 +1,16 @@
+from PIL import Image, ImageDraw, ImageFont
+
+class MockOut:
+    def __init__(self, img): self.images = [img]
+    
+class MockPipe:
+    def __call__(self, prompt, negative_prompt=None, guidance_scale=6.0,
+                 num_inference_steps=25, width=1024, height=768, generator=None):
+        
+        img = Image.new("RGB", (int(width), int(height)), (230, 230, 230))
+        draw = ImageDraw.Draw(img)
+        text = (prompt or "").strip()[:160] or "MOCK IMAGE"
+        
+        draw.rectangle([(10, 10), (int(width)-10, 90)], outline=(0,0,0), width=2)
+        draw.text((20, 20), f"[MOCK]\n{text}", fill=(0,0,0))
+        return MockOut(img)

pyproject.toml

@@ -0,0 +1,15 @@
+[project]
+name = "wainsfwillustrious-v140"
+version = "0.1.0"
+description = "Add your description here"
+readme = "README.md"
+requires-python = ">=3.10"
+dependencies = [
+    "diffusers>=0.34.0",
+    "gradio>=5.42.0",
+    "pillow>=11.3.0",
+    "safetensors>=0.6.2",
+    "spaces>=0.40.0",
+    "torch>=2.8.0",
+    "transformers>=4.55.0",
+]