Full loading UI on the result during generation; examples load silently

app.py

@@ -162,39 +162,43 @@ def generate(prompt, images=None, aspect=AUTO, resolution="4 MP", seed=0, random
 
 
 def studio(state, seed, randomize, effort):
-    """The composer's Generate: optional PE (streamed), then the image."""
+    """The composer's Generate, part 1: optional PE (streamed) and encoding. Returns the render plan."""
     state = state or {}
     prompt = (state.get("prompt") or "").strip()
     aspect = state.get("aspect") or AUTO
     resolution = state.get("resolution") or "4 MP"
     if not prompt:
         raise gr.Error("Describe the image first.")
-    images = paths = decode_pictures(state.get("images"))
+    paths = decode_pictures(state.get("images"))
     if randomize:
         seed = random.randint(0, MAX_SEED)
     final, thinking = prompt, ""
     if state.get("enhance") and paths:
         gr.Info("Prompt enhancement is text-to-image only for now; edits use your prompt as written.")
     elif state.get("enhance"):
-        yield gr.update(), "✨ Enhancing the prompt with MiniMax-H3-Image-PE ...", gr.update(), "", seed
+        yield "✨ Enhancing the prompt with MiniMax-H3-Image-PE ...", gr.update(), "", seed, None
         t0 = time.time()
         caption = ratio = None
         for thinking, caption, ratio in enhance_stream(prompt, effort, seed):
             if caption is None:
-                yield gr.update(), f"✨ Enhancing the prompt · thinking ({len(thinking.split())} words) ...", gr.update(), thinking, seed
+                yield f"✨ Enhancing the prompt · thinking ({len(thinking.split())} words) ...", gr.update(), thinking, seed, None
         if caption:
             final = caption
             if aspect == AUTO and ratio:
                 aspect = ratio
-        yield gr.update(), f"✨ Prompt enhanced in {time.time() - t0:.0f}s · encoding ...", final, thinking, seed
+        yield f"✨ Prompt enhanced in {time.time() - t0:.0f}s · encoding ...", final, thinking, seed, None
     width, height = resolve_size(paths, aspect, resolution)
-    yield gr.update(), f"Encoding the prompt · {width}×{height} ...", final, thinking, seed
-    embeds_path, seed, width, height = prepare(final, images, aspect, resolution, seed, False)
-    yield gr.update(), f"Generating {width}×{height} ...", final, thinking, seed
-    t0 = time.time()
-    path = _generate(embeds_path, images, width, height, seed)
-    yield path, f"{width}×{height} · seed {seed} · {time.time() - t0:.0f}s", final, thinking, seed
+    yield f"Encoding the prompt · {width}×{height} ...", final, thinking, seed, None
+    embeds_path, seed, width, height = prepare(final, paths, aspect, resolution, seed, False)
+    plan = {"embeds": embeds_path, "images": paths, "width": width, "height": height, "seed": seed}
+    yield f"Generating {width}×{height} ...", final, thinking, seed, plan
+
 
+def render(plan):
+    """Part 2: the image, as its own event so the result panel shows the normal loading UI."""
+    t0 = time.time()
+    path = _generate(plan["embeds"], plan["images"], plan["width"], plan["height"], plan["seed"])
+    return path, f"{plan['width']}×{plan['height']} · seed {plan['seed']} · {time.time() - t0:.0f}s"
 
 IDEAS = [
     "a two-color risograph print of a fox reading a newspaper on a park bench, headline \"FOX NEWS\"",
@@ -485,14 +489,19 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=CSS, title="MiniMax H3 Image") as d
                 with gr.Accordion("Enhancer thinking", open=False):
                     thinking = gr.Textbox(show_label=False, lines=10)
 
+    plan = gr.State()
     composer.submit(
-        studio, [composer, seed, randomize, effort], [output, stats, prompt_used, thinking, seed],
+        lambda: (None, None), None, [output, plan], show_progress="hidden", api_visibility="private",
+    ).then(
+        studio, [composer, seed, randomize, effort], [stats, prompt_used, thinking, seed, plan],
         show_progress="minimal", api_visibility="private",
+    ).success(
+        render, [plan], [output, stats], show_progress="full", show_progress_on=[output], api_visibility="private",
     )
 
     composer.apply(
         load_example, [composer], [output, stats, prompt_used, thinking],
-        show_progress="full", api_visibility="private",
+        show_progress="hidden", api_visibility="private",
     )
 
     # Plain API endpoint (also what the composer runs, minus the streaming UI updates).