feat: add wait_ready checkbox for cluster up button

Signed-off-by: Bobbins228 <mcampbel@redhat.com>

Author: Bobbins228

src/codeflare_sdk/cluster/cluster.py

@@ -42,7 +42,11 @@
     RayCluster,
     RayClusterStatus,
 )
-from .widgets import cluster_up_down_buttons, is_notebook
+from .widgets import (
+    cluster_up_down_buttons,
+    is_notebook,
+    wait_ready_check_box,
+)
 from kubernetes import client, config
 from kubernetes.utils import parse_quantity
 import yaml
@@ -173,16 +177,23 @@ def up(self):
 
     def _cluster_up_down_buttons(self):
         up_button, delete_button = cluster_up_down_buttons(self.config.appwrapper)
+        wait_ready_check = wait_ready_check_box()
         output = widgets.Output()
-        # Display the buttons in an HBox
-        display(widgets.HBox([up_button, delete_button]), output)
 
-        def on_up_button_clicked(b):
+        # Display the buttons in an HBox wrapped in a VBox which includes the wait_ready Checkbox
+        button_display = widgets.HBox([up_button, delete_button])
+        display(widgets.VBox([button_display, wait_ready_check]), output)
+
+        def on_up_button_clicked(b):  # Handle the up button click event
             with output:
                 output.clear_output()
                 self.up()
 
-        def on_down_button_clicked(b):
+                # If the wait_ready Checkbox is clicked(value == True) trigger the wait_ready function
+                if wait_ready_check.value:
+                    self.wait_ready()
+
+        def on_down_button_clicked(b):  # Handle the down button click event
             with output:
                 output.clear_output()
                 self.down()

src/codeflare_sdk/cluster/widgets.py

@@ -43,6 +43,17 @@ def cluster_up_down_buttons(appwrapper: bool) -> widgets.Button:
     return up_button, delete_button
 
 
+def wait_ready_check_box():
+    """
+    The wait_ready_check_box function will return a checkbox widget used for waiting for the resource to be in the state READY.
+    """
+    wait_ready_check_box = widgets.Checkbox(
+        False,
+        description="Wait for Cluster?",
+    )
+    return wait_ready_check_box
+
+
 def is_notebook() -> bool:
     """
     The is_notebook function checks if Jupyter Notebook environment variables exist in the given environment and return True/False based on that.