Update app.py

app.py

@@ -3,7 +3,7 @@ import torch
 import random
 import time
 from diffusers import DiffusionPipeline, LCMScheduler
-from PIL import Image, ImageDraw, ImageFont, ImageFilter
+from PIL import Image, ImageFilter
 
 # -------------------------------
 # MODEL SETUP (CPU SAFE)
@@ -50,24 +50,6 @@ def estimate_time(steps, resolution):
     seconds = est % 60
     return f"⏱️ Estimated time: ~{int(minutes)}m {int(seconds)}s"
 
-# -------------------------------
-# GENERATING PLACEHOLDER IMAGE
-# -------------------------------
-def generating_placeholder(resolution):
-    resolution = int(resolution)
-    img = Image.new("RGB", (resolution, resolution), (30, 30, 30))  # dark background
-    draw = ImageDraw.Draw(img)
-    text = "Generating..."
-    try:
-        font = ImageFont.truetype("arial.ttf", resolution // 12)
-    except:
-        font = ImageFont.load_default()
-    text_width, text_height = draw.textsize(text, font=font)
-    x = (resolution - text_width) / 2
-    y = (resolution - text_height) / 2
-    draw.text((x, y), text, fill=(255, 255, 255), font=font)
-    return img
-
 # -------------------------------
 # IMAGE GENERATION WITH PROGRESSIVE BLUR
 # -------------------------------
@@ -78,9 +60,10 @@ def generate(prompt, resolution, steps):
     # 1️⃣ Refine prompt
     refined_prompt, neg_prompt = refine_prompt(prompt)
 
-    # 2️⃣ Show "Generating..." placeholder
-    placeholder = generating_placeholder(resolution)
-    yield [placeholder]
+    # 2️⃣ White placeholder image
+    placeholder = Image.new("RGB", (resolution, resolution), (255, 255, 255))
+    start_time = time.time()
+    yield [placeholder], "🟡 Generating..."
 
     # 3️⃣ CPU Image generation
     seed = random.randint(0, 10**9)
@@ -103,11 +86,13 @@ def generate(prompt, resolution, steps):
     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]
-        time.sleep(1)  # 1 second per step
+        yield [blurred_img], "🟢 Generating..."
+
+        time.sleep(1)
 
-    # 5️⃣ Fully revealed image
-    yield [img]
+    # 5️⃣ Final image with total time
+    total_time = round(time.time() - start_time, 2)
+    yield [img], f"✅ Done in {total_time}s"
 
 # -------------------------------
 # GRADIO UI
@@ -126,6 +111,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             eta = gr.Markdown("⏱️ Estimated time: ~1m 0s")
             gen_btn = gr.Button("Generate")
 
+            status = gr.Markdown("🟢 Ready")
+
         with gr.Column():
             gallery = gr.Gallery(columns=1)
 
@@ -137,7 +124,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gen_btn.click(
         generate,
         inputs=[prompt_in, resolution, steps],
-        outputs=[gallery]
+        outputs=[gallery, status]
     )
 
 demo.launch()