Update app_quant_latent.py

app_quant_latent.py

@@ -580,7 +580,7 @@ def generate_image(prompt, height, width, steps, seed, guidance_scale=0.0):
 
                     latents = pipe.scheduler.step(noise_pred, t, latents)["prev_sample"]
 
-                # Convert latents to image
+                # convert latent → image
                 try:
                     latent_img = latent_to_image(latents)
                 except Exception:
@@ -588,29 +588,28 @@ def generate_image(prompt, height, width, steps, seed, guidance_scale=0.0):
 
                 latent_gallery.append(latent_img)
 
-                # 🔥 STREAM UPDATE: (final_image=None for now)
-                yield {
-                    "final_image": None,
-                    "latent_gallery": latent_gallery,
-                    "logs_box": "\n".join(LOGS),
-                }
+                # 🔥 STREAM update
+                yield (
+                    None,               # final_image
+                    latent_gallery,     # latent gallery list
+                    "\n".join(LOGS),    # logs
+                )
 
-            # ----------------------
-            # Final decode
-            # ----------------------
+            # ---------------------
+            # FINAL decode
+            # ---------------------
             final_img = pipe.decode_latents(latents)[0]
             final_gallery.append(final_img)
             LOGS.append("✅ Advanced latent pipeline succeeded.")
 
-            # 🔥 Final output
-            yield {
-                "final_image": final_img,
-                "latent_gallery": latent_gallery,
-                "logs_box": "\n".join(LOGS),
-            }
+            yield (
+                final_img,
+                latent_gallery,
+                "\n".join(LOGS),
+            )
 
         # ==========================================================
-        # STANDARD PIPELINE (fallback)
+        # FALLBACK STANDARD PIPELINE
         # ==========================================================
         except Exception as e:
             LOGS.append(f"⚠️ Advanced latent mode failed: {e}")
@@ -626,25 +625,23 @@ def generate_image(prompt, height, width, steps, seed, guidance_scale=0.0):
             )
 
             final_img = output.images[0]
-            final_gallery.append(final_img)
             latent_gallery.append(final_img)
             LOGS.append("✅ Standard pipeline succeeded.")
 
-            yield {
-                "final_image": final_img,
-                "latent_gallery": latent_gallery,
-                "logs_box": "\n".join(LOGS),
-            }
+            yield (
+                final_img,
+                latent_gallery,
+                "\n".join(LOGS),
+            )
 
     except Exception as e:
         LOGS.append(f"❌ Total failure: {e}")
         placeholder_img = placeholder
-
-        yield {
-            "final_image": placeholder_img,
-            "latent_gallery": [placeholder_img],
-            "logs_box": "\n".join(LOGS),
-        }
+        yield (
+            placeholder_img,
+            [placeholder_img],
+            "\n".join(LOGS),
+        )
 
 
 
@@ -963,11 +960,11 @@ with gr.Blocks(title="Z-Image-Turbo") as demo:
     run_btn.click(
     fn=generate_image,
     inputs=[prompt, height, width, steps, seed],
-    outputs={
-        "final_image": final_image,
-        "latent_gallery": latent_gallery,
-        "logs_box": logs_box,
-         }
+    run_btn.click(
+         generate_image,
+         inputs=[prompt, height, width, steps, seed],
+         outputs=[final_image, latent_gallery, logs_box]
     )
 
+
 demo.launch()