Update app_lora1.py

app_lora1.py

@@ -274,56 +274,16 @@ def generate_image(
 # =========================================================
 # GRADIO UI
 # =========================================================
-with gr.Blocks(title="Z-Image-Turbo Multi Image Demo") as demo:
-    gr.Markdown("# 🎨 Z-Image-Turbo — Multi Image")
-
-    with gr.Row():
-        with gr.Column(scale=1):
-            prompt = gr.Textbox(label="Prompt", lines=4)
-
-            with gr.Row():
-                height = gr.Slider(512, 2048, 1024, step=64, label="Height")
-                width = gr.Slider(512, 2048, 1024, step=64, label="Width")
-
-            num_images = gr.Slider(1, 3, 2, step=1, label="Number of Images")
-
-            num_inference_steps = gr.Slider(
-                1, 20, 9, step=1, label="Inference Steps",
-                info="9 steps = 8 DiT forward passes",
-            )
-
-            with gr.Row():
-                seed = gr.Number(label="Seed", value=42, precision=0)
-                randomize_seed = gr.Checkbox(label="Randomize Seed", value=False)
-
-            # Select pipeline script
-            pipeline_picker = gr.Dropdown(
-                choices=SCRIPT_NAMES,
-                value=SCRIPT_NAMES[0] if SCRIPT_NAMES else None,
-                label="Active Pipeline Script",
-            )
-
-            generate_btn = gr.Button("🚀 Generate", variant="primary")
-
-        with gr.Column(scale=1):
-            output_images = gr.Gallery(
-                label="Generated Images",
-                height=512,
-                object_fit="contain"
-            )
-            used_seed = gr.Number(label="Seed Used", interactive=False)
-            debug_log = gr.Textbox(
-                label="Debug Log Output",
-                lines=25,
-                interactive=False
-            )
+with gr.Blocks(title="Z-Image Turbo – ZeroGPU") as demo:
+    gr.Markdown("## ⚡ Z-Image Turbo (Script-Driven · ZeroGPU Safe)")
 
-    register_btn = gr.Button("Register Scripts")
-    status = gr.Textbox(label="Status", interactive=False)
+    # ------------------ Scripts selection ------------------
     script_selector = gr.CheckboxGroup(
         choices=SCRIPT_NAMES,
         label="Select pipeline scripts",
     )
+    register_btn = gr.Button("Register Scripts")
+    status = gr.Textbox(label="Status", interactive=False)
 
     register_btn.click(
         register_scripts,
@@ -331,10 +291,40 @@ with gr.Blocks(title="Z-Image-Turbo Multi Image Demo") as demo:
         outputs=[status],
     )
 
-    generate_btn.click(
-        fn=generate_image,
-        inputs=[prompt, height, width, num_inference_steps, seed, randomize_seed, num_images, pipeline_picker],
-        outputs=[output_images, used_seed, debug_log],
+    # ------------------ Pipeline dropdown ------------------
+    pipeline_picker = gr.Dropdown(
+        choices=[],
+        label="Active Pipeline",
+    )
+
+    register_btn.click(
+        lambda s: gr.update(choices=s, value=s[0] if s else None),
+        inputs=[script_selector],
+        outputs=[pipeline_picker],
+    )
+
+    gr.Markdown("---")
+
+    # ------------------ Prompt + sliders ------------------
+    prompt = gr.Textbox(label="Prompt", lines=3)
+    height = gr.Slider(256, 1024, 512, step=64, label="Height")
+    width = gr.Slider(256, 1024, 512, step=64, label="Width")
+    steps = gr.Slider(1, 8, 4, step=1, label="Inference Steps")
+    images = gr.Slider(1, 3, 1, step=1, label="Images")
+    seed = gr.Number(value=0, label="Seed")
+    random_seed = gr.Checkbox(value=True, label="Randomize Seed")
+
+    run_btn = gr.Button("Generate")
+
+    # ------------------ Outputs ------------------
+    gallery = gr.Gallery(columns=3, height=512, object_fit="contain")
+    used_seed = gr.Number(label="Used Seed")
+    logs = gr.Textbox(lines=25, label="Logs")
+
+    run_btn.click(
+        generate_image,
+        inputs=[prompt, height, width, steps, seed, random_seed, images, pipeline_picker],
+        outputs=[gallery, used_seed, logs],
     )
 
 demo.queue()