Update app.py

app.py

@@ -59,47 +59,42 @@ def estimate_time(steps, resolution):
     return f"⏱️ Estimated time: ~{int(minutes)}m {int(seconds)}s"
 
 # -------------------------------
-# IMAGE GENERATION WITH PROGRESSIVE BLUR REVEAL
+# IMAGE GENERATION WITH PROGRESSIVE BLUR GALLERY
 # -------------------------------
 def generate(prompt, resolution, steps):
     # --- THINKING PHASE ---
-    start_time = time.time()
-    yield None, "🧠 Understanding your prompt..."
     refined_prompt, neg_prompt = refine_prompt_fast(prompt)
-    yield None, "🎨 Generating image (CPU, please wait)..."
+
+    # --- SHOW WHITE IMAGE WHILE GENERATING ---
+    width, height = int(resolution), int(resolution)
+    blank_img = Image.new("RGB", (width, height), (255, 255, 255))
+    yield [blank_img], ""  # always show gallery placeholder
 
     # --- GENERATION PHASE ---
     seed = random.randint(0, 10**9)
     gen = torch.Generator("cpu").manual_seed(seed)
-    
     pipe.scheduler.set_timesteps(int(steps))
     img = pipe(
         prompt=refined_prompt,
         negative_prompt=neg_prompt,
         num_inference_steps=int(steps),
         guidance_scale=1.2,
-        width=int(resolution),
-        height=int(resolution),
+        width=width,
+        height=height,
         generator=gen
     ).images[0]
 
     # --- PROGRESSIVE BLUR REVEAL ---
-    # First show white placeholder
-    width, height = img.size
-    white_img = Image.new("RGB", (width, height), (255, 255, 255))
-    yield [white_img], "🖼 Image generated! Revealing..."
-
-    # Apply progressive blur
-    max_blur = 20  # max blur radius
-    steps_blur = 10  # number of steps
+    max_blur = 20  # max blur radius for 100%
+    steps_blur = 10
     for i in range(steps_blur):
         blur_percent = 100 - i*10
         blurred_img = img.filter(ImageFilter.GaussianBlur(radius=max_blur * blur_percent / 100))
-        yield [blurred_img], f"🖼 Revealing image... {i*10}%"
+        yield [blurred_img], ""  # gallery updated, no status text
         time.sleep(1)  # 1 second per step
 
-    duration = int(time.time() - start_time)
-    yield [img], f"✅ Finished in {duration}s | Seed: {seed}"
+    # --- FINAL IMAGE ---
+    yield [img], ""  # fully revealed image
 
 # -------------------------------
 # GRADIO UI
@@ -119,16 +114,17 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             gen_btn = gr.Button("Generate")
         
         with gr.Column():
-            status = gr.Markdown("🟢 Ready")
             gallery = gr.Gallery(columns=1)
-    
+
+    # Update ETA dynamically
     for ctrl in [steps, resolution]:
         ctrl.change(estimate_time, [steps, resolution], eta)
-    
+
+    # Button click triggers generation
     gen_btn.click(
         generate,
         inputs=[prompt_in, resolution, steps],
-        outputs=[gallery, status]
+        outputs=[gallery, gallery]  # gallery updated for progressive blur
     )
 
 demo.launch()