Update app.py

app.py

@@ -58,7 +58,7 @@ def clear_fn(*args):
 def run_model(prompt, model_idx):
     return send_it_idx(model_idx)(prompt)
 
-def build_ui():
+with gr.Blocks(title="SD Models") as my_interface:
     primary_prompt = gr.Textbox(label="Prompt", value="")
     run = gr.Button("Run")
     clear_btn = gr.Button("Clear")
@@ -71,24 +71,38 @@ def build_ui():
     end_box = gr.Number(visible=False)
     tog_box = gr.Textbox(value=0, visible=False)
 
-    return primary_prompt, run, clear_btn, sd_outputs, start_box, end_box, tog_box
-
-# Define the function that combines all components
-def main_function(prompt):
-    return {idx: run_model(prompt, idx) for idx in range(1, len(models) + 1)}
-
-primary_prompt, run, clear_btn, sd_outputs, start_box, end_box, tog_box = build_ui()
-
-# Setup the interface
-interface = gr.Interface(
-    fn=main_function,
-    inputs=[primary_prompt],
-    outputs=list(sd_outputs.values())
-)
-
-# Set concurrency limit directly on event listeners
-run.click(fn=main_function, inputs=[primary_prompt], outputs=list(sd_outputs.values()), concurrency_limit=10)
-clear_btn.click(fn=clear_fn, inputs=None, outputs=[primary_prompt] + list(sd_outputs.values()), concurrency_limit=10)
-
-# Launch with max_threads if necessary
-interface.launch(max_threads=10)
+    # Define event handlers within the Blocks context
+    def main_function(prompt):
+        return {idx: run_model(prompt, idx) for idx in range(1, len(models) + 1)}
+
+    run.click(
+        fn=main_function,
+        inputs=[primary_prompt],
+        outputs=list(sd_outputs.values()),
+        concurrency_limit=10
+    )
+
+    clear_btn.click(
+        fn=clear_fn,
+        inputs=None,
+        outputs=[primary_prompt] + list(sd_outputs.values()),
+        concurrency_limit=10
+    )
+
+    start_box.change(
+        fn=all_task_end,
+        inputs=[start_box, end_box],
+        outputs=[start_box, tog_box],
+        every=1
+    )
+
+    primary_prompt.submit(all_task_start, inputs=None, outputs=[start_box, end_box, tog_box])
+    run.click(all_task_start, inputs=None, outputs=[start_box, end_box, tog_box])
+    tog_box.change(
+        fn=clear_it,
+        inputs=tog_box,
+        outputs=tog_box
+    )
+
+    # Launch the interface
+    my_interface.launch(max_threads=10)