Update app.py

app.py

@@ -646,6 +646,19 @@ setTimeout(() => {{
                             settings.get("name", ""),
                         ),
                     )
+                dataset_name = gr.Textbox(
+                    label="Dataset name (export title prefix)",
+                    value=settings.get("dataset_name", "forgecaptions"),
+                )
+                max_side = gr.Slider(256, 1024, settings.get("max_side", 896), step=32, label="Max side (resize)")
+                excel_thumb_px = gr.Slider(64, 256, value=settings.get("excel_thumb_px", 128), step=8, label="Excel thumbnail size (px)")
+            
+                # Chunking controls (restored)
+                chunk_mode = gr.Radio(
+                    choices=["Auto", "Manual (all at once)", "Manual (step)"],
+                    value="Manual (step)",
+                    label="Batch mode",
+                )
                 chunk_size = gr.Slider(1, 50, value=10, step=1, label="Chunk size")
                 gpu_budget = gr.Slider(20, 110, value=55, step=5, label="Max seconds per GPU call")
 
@@ -660,18 +673,25 @@ setTimeout(() => {{
             "trigger": trigv, "begin": begv, "end": endv,
             "excel_thumb_px": int(excel_px),
             "max_side": int(ms),
-            "dataset_name": ds_name or "forgecaptions",
         })
         save_settings(cfg)
         return instr
-
-    for comp in [style_checks, extra_opts, name_input, trig, add_start, add_end, excel_thumb_px, max_side, dataset_name]:
-        comp.change(_refresh_instruction,
-                    inputs=[style_checks, extra_opts, name_input, trig, add_start, add_end, excel_thumb_px, max_side, dataset_name],
-                    outputs=[instruction_preview])
-
-    demo.load(lambda s,e,n: final_instruction(s or ["Character training (long)"], e or [], n),
-              inputs=[style_checks, extra_opts, name_input], outputs=[instruction_preview])
+    
+    for comp in (
+        style_checks, extra_opts, name_input, trig, add_start, add_end, excel_thumb_px, max_side
+    ):
+        comp.change(
+            _refresh_instruction,
+            inputs=[style_checks, extra_opts, name_input, trig, add_start, add_end, excel_thumb_px, max_side],
+            outputs=[instruction_preview],
+        )
+    
+    # Also set the initial instruction on load (optional but nice)
+    demo.load(
+        lambda s, e, n: final_instruction(s or ["Character training (long)"], e or [], n),
+        inputs=[style_checks, extra_opts, name_input],
+        outputs=[instruction_preview],
+    )
 
     # ---- Shape Aliases accordion (with examples & buttons)
     with gr.Accordion("Shape Aliases", open=False):