Update app.py

app.py

@@ -766,7 +766,12 @@ with gr.Blocks(css=BASE_CSS, title="ForgeCaptions") as demo:
                     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")
+                gpu_budget = gr.Slider(20, 120, value=55, step=5, label="Max seconds per GPU call")
+                no_time_limit = gr.Checkbox(
+                    value=True,
+                    label="No time limit (may time out on Zero GPU)"
+                )
+
 
                 # Logo controls
                 logo_auto = gr.Checkbox(value=settings.get("logo_auto", True),
@@ -966,6 +971,7 @@ with gr.Blocks(css=BASE_CSS, title="ForgeCaptions") as demo:
         t, p, m = _tpms()
         files = files or []
 
+        #Manual (Step)
         if mode == "Manual (step)" and files:
             chunks = _split_chunks(files, int(csize))
             batch = chunks[0]
@@ -979,6 +985,7 @@ with gr.Blocks(css=BASE_CSS, title="ForgeCaptions") as demo:
             prog = f"Batch progress: {done}/{total} processed in this step • Remaining overall: {len(remaining)}"
             return new_rows, gal, tbl, stamp, remaining, panel_vis, gr.update(value=msg), gr.update(value=prog)
 
+        #Auto/all-at-once
         new_rows, gal, tbl, stamp, leftover, done, total = run_batch(
             files, rows or [], instr, t, p, m, int(ms), float(budget_s)
         )
@@ -989,19 +996,21 @@ with gr.Blocks(css=BASE_CSS, title="ForgeCaptions") as demo:
 
     run_button.click(
         _run_click,
-        inputs=[input_files, rows_state, instruction_preview, max_side, chunk_mode, chunk_size, gpu_budget],
+        inputs=[input_files, rows_state, instruction_preview, max_side, chunk_mode, chunk_size, gpu_budget, no_time_limit],
         outputs=[rows_state, gallery, table, autosave_md, remaining_state, step_panel, step_msg, progress_md]
     )
 
     def _step_next(remain, rows, instr, ms, csize, budget_s):
         t, p, m = _tpms()
         remain = remain or []
+        budget = None if no_limit else float(budget_s)
+        
         if not remain:
             return rows, gr.update(value="No files remaining."), gr.update(visible=False), [], [], [], "Saved.", gr.update(value="")
         batch = remain[:int(csize)]
         leftover = remain[int(csize):]
         new_rows, gal, tbl, stamp, leftover_from_batch, done, total = run_batch(
-            batch, rows or [], instr, t, p, m, int(ms), float(budget_s)
+            batch, rows or [], instr, t, p, m, int(ms), budget
         )
         leftover = (leftover_from_batch or []) + leftover
         panel_vis = gr.update(visible=bool(leftover))
@@ -1011,7 +1020,7 @@ with gr.Blocks(css=BASE_CSS, title="ForgeCaptions") as demo:
 
     step_next.click(
         _step_next,
-        inputs=[remaining_state, rows_state, instruction_preview, max_side, chunk_size, gpu_budget],
+        inputs=[remaining_state, rows_state, instruction_preview, max_side, chunk_size, gpu_budget, no_time_limit],
         outputs=[rows_state, step_msg, step_panel, remaining_state, gallery, table, autosave_md, progress_md]
     )