Prevent multi-clicking

app.py

@@ -114,7 +114,7 @@ if __name__ == "__main__":
         if state["ipe"]._action_is_discrete:
             symbolic_formula += "\n" + r"$$ a = \underset{i}{\mathrm{argmax}} \ a_i.$$"
 
-        return video_path, symbolic_formula
+        return video_path, symbolic_formula, gr.Button("Compute the symbolic policy!", interactive=True)
 
 
     css = """
@@ -143,8 +143,8 @@ if __name__ == "__main__":
                 sym_video = gr.Video(label="Symbolic policy video", interactive=False, autoplay=True)
         sym_formula = gr.Markdown(elem_id="formula")
         choice.input(load_video_and_dataset, inputs=[choice], outputs=[expert_video, button, state], concurrency_limit=8)
-        button.click(extract_interpretable_policy, inputs=[kan_widths, epochs, state], outputs=[kan_architecture, state]).then(
-            symbolic_policy, inputs=[state], outputs=[sym_video, sym_formula]
+        button.click(lambda: gr.Button("Computation in progress...", interactive=False), None, [button]).then(extract_interpretable_policy, inputs=[kan_widths, epochs, state], outputs=[kan_architecture, state]).then(
+            symbolic_policy, inputs=[state], outputs=[sym_video, sym_formula, button]
         )
     app.queue(default_concurrency_limit=3, max_size=10)
     app.launch()