hi

app.py

@@ -41,7 +41,7 @@ def generate_vstack_from_image(image: Image.Image, input_focal_position: int, nu
     if not first_frame.exists():
         raise gr.Error("frame_0.png not found in output_dir")
 
-    # show first frame + reset slider to 0
+    # Show first frame + reset slider
     return str(first_frame), gr.update(value=0)
 
 def show_frame(idx: int):
@@ -51,15 +51,12 @@ def show_frame(idx: int):
     return str(path)
 
 def advance_frame(idx: int, autoplay: bool):
-    # If autoplay is off, don't change anything.
     if not autoplay:
-        # keep slider value, keep image (returning None keeps the current image)
         return gr.update(value=int(idx)), gr.update()
 
     next_idx = (int(idx) + 1) % NUM_FRAMES
     next_path = OUTPUT_DIR / f"frame_{next_idx}.png"
     if not next_path.exists():
-        # If frames aren't ready yet, hold.
         return gr.update(value=int(idx)), gr.update()
 
     return gr.update(value=next_idx), str(next_path)
@@ -68,7 +65,7 @@ with gr.Blocks(css="footer {visibility: hidden}") as demo:
     gr.Markdown(
         """
         # 🖼️ ➜ 🎬 Generate Focal Stacks from a Single Image
-        Generate a focal stack and scrub through frames using the slider. Toggle autoplay to loop.
+        Generate a focal stack and scrub through frames. Toggle autoplay on the output panel.
         """
     )
 
@@ -93,8 +90,6 @@ with gr.Blocks(css="footer {visibility: hidden}") as demo:
                 info="More steps = better quality but slower",
             )
 
-            autoplay = gr.Checkbox(value=True, label="Autoplay")
-
             generate_btn = gr.Button("Generate stack", variant="primary")
 
         with gr.Column():
@@ -105,10 +100,11 @@ with gr.Blocks(css="footer {visibility: hidden}") as demo:
                 maximum=NUM_FRAMES - 1,
                 step=1,
                 value=0,
-                label="Focal plane",
+                label="Output Focus Position",
             )
 
-    # Generate outputs + reset slider
+            autoplay = gr.Checkbox(value=True, label="Autoplay")
+
     generate_btn.click(
         fn=generate_vstack_from_image,
         inputs=[image_in, input_focal_position, num_inference_steps],
@@ -116,14 +112,12 @@ with gr.Blocks(css="footer {visibility: hidden}") as demo:
         api_name="predict",
     )
 
-    # Manual scrubbing
     frame_slider.change(
         fn=show_frame,
         inputs=frame_slider,
         outputs=frame_view,
     )
 
-    # Autoplay loop (can be toggled off via checkbox)
     timer = gr.Timer(LOOP_MS)
     timer.tick(
         fn=advance_frame,