handle visibility of run and stop buttons simultaneously

app.py

@@ -140,11 +140,8 @@ with gr.Blocks(title="RFD3 Test") as demo:
             return textbox_update, gen_directory, gen_results
         
     generation_event = run_btn.click(
-        lambda: gr.update(visible=False),
-        outputs=run_btn
-    ).then(
-        lambda: gr.update(visible=True),
-        outputs=stop_btn
+        lambda: (gr.update(visible=False), gr.update(visible=True)),
+        outputs=[run_btn, stop_btn]
     ).then(
         generate, inputs=[config_upload, scaffold_upload, num_batches, num_designs_per_batch, extra_args, max_duration], outputs=[runtextbox, gen_directory, gen_results]
     ).then(
@@ -156,11 +153,8 @@ with gr.Blocks(title="RFD3 Test") as demo:
         inputs=gen_directory,
         outputs=output_file
     ).then(
-        lambda: gr.update(visible=True),
-        outputs=run_btn
-    ).then(
-        lambda: gr.update(visible=False),
-        outputs=stop_btn
+        lambda: (gr.update(visible=True), gr.update(visible=False)),
+        outputs=[run_btn, stop_btn]
     )
     
     stop_btn.click(
@@ -168,11 +162,8 @@ with gr.Blocks(title="RFD3 Test") as demo:
         outputs=runtextbox,
         cancels=[generation_event]
     ).then(
-        lambda: gr.update(visible=True),
-        outputs=run_btn
-    ).then(
-        lambda: gr.update(visible=False),
-        outputs=stop_btn
+        lambda: (gr.update(visible=True), gr.update(visible=False)),
+        outputs=[run_btn, stop_btn]
     )