Add Run Forever checkbox below Run button for all tabs

ui/shared/hires_fix_ui.py

@@ -34,8 +34,9 @@ def create_ui():
                 scale=3,
                 allow_custom_value=True
             )
-            with gr.Column(scale=1): 
-                components[f'run_{prefix}'] = gr.Button("Run Hires. Fix", variant="primary")
+                with gr.Column(scale=1): 
+                    components[f'run_{prefix}'] = gr.Button("Run Hires. Fix", variant="primary")
+                    components[f'run_forever_{prefix}'] = gr.Checkbox(label="Run Forever", value=False)
         
         with gr.Row():
             with gr.Column(scale=1):

ui/shared/img2img_ui.py

@@ -30,6 +30,7 @@ def create_ui():
             components[f'base_model_{prefix}'] = gr.Dropdown(label="Base Model", choices=list(MODEL_MAP_CHECKPOINT.keys()), value=list(MODEL_MAP_CHECKPOINT.keys())[0], scale=3, allow_custom_value=True)
             with gr.Column(scale=1): 
                 components[f'run_{prefix}'] = gr.Button("Run", variant="primary")
+                components[f'run_forever_{prefix}'] = gr.Checkbox(label="Run Forever", value=False)
 
         with gr.Row():
             with gr.Column(scale=1):

ui/shared/inpaint_ui.py

@@ -34,8 +34,9 @@ def create_ui():
                 scale=3,
                 allow_custom_value=True
             )
-            with gr.Column(scale=1): 
-                components[f'run_{prefix}'] = gr.Button("Run Inpaint", variant="primary")
+                with gr.Column(scale=1): 
+                    components[f'run_{prefix}'] = gr.Button("Run Inpaint", variant="primary")
+                    components[f'run_forever_{prefix}'] = gr.Checkbox(label="Run Forever", value=False)
         
         components[f'model_and_run_row_{prefix}'] = [arch_row, model_and_run_row]
 

ui/shared/outpaint_ui.py

@@ -34,8 +34,9 @@ def create_ui():
                 scale=3,
                 allow_custom_value=True
             )
-            with gr.Column(scale=1): 
-                components[f'run_{prefix}'] = gr.Button("Run Outpaint", variant="primary")
+                with gr.Column(scale=1): 
+                    components[f'run_{prefix}'] = gr.Button("Run Outpaint", variant="primary")
+                    components[f'run_forever_{prefix}'] = gr.Checkbox(label="Run Forever", value=False)
 
         with gr.Row():
             with gr.Column(scale=1):

ui/shared/txt2img_ui.py

@@ -32,6 +32,7 @@ def create_ui():
             )
             with gr.Column(scale=1):
                 components[f'run_{prefix}'] = gr.Button("Run", variant="primary")
+                components[f'run_forever_{prefix}'] = gr.Checkbox(label="Run Forever", value=False)
         
         # Prompt fields
         components[f'prompt_{prefix}'] = gr.Text(label="Prompt", lines=3)