feat: add ui cluster creation and deletion buttons

Signed-off-by: Bobbins228 <[email protected]>

Author: Bobbins228

src/codeflare_sdk/cluster/cluster.py

@@ -51,6 +51,9 @@
 from kubernetes import config
 from kubernetes.client.rest import ApiException
 
+import ipywidgets as widgets
+from IPython.display import display
+
 
 class Cluster:
     """
@@ -71,6 +74,11 @@ def __init__(self, config: ClusterConfiguration):
         self.app_wrapper_yaml = self.create_app_wrapper()
         self._job_submission_client = None
         self.app_wrapper_name = self.config.name
+        if (
+            "PYDEVD_IPYTHON_COMPATIBLE_DEBUGGING" in os.environ
+            or "JPY_SESSION_NAME" in os.environ
+        ):  # If running Jupyter NBs in VsCode or RHOAI/ODH display UI buttons
+            self._cluster_up_down_buttons()
 
     @property
     def _client_headers(self):
@@ -156,11 +164,42 @@ def up(self):
                         plural="appwrappers",
                         body=aw,
                     )
+                print(f"AppWrapper: '{self.config.name}' has successfully been created")
             else:
                 self._component_resources_up(namespace, api_instance)
+                print(
+                    f"Ray Cluster: '{self.config.name}' has successfully been created"
+                )
         except Exception as e:  # pragma: no cover
             return _kube_api_error_handling(e)
 
+    def _cluster_up_down_buttons(self):
+        delete_button = widgets.Button(
+            description="Cluster Down",
+            icon="trash",
+        )
+        up_button = widgets.Button(
+            description="Cluster Up",
+            icon="play",
+        )
+
+        output = widgets.Output()
+        # Display the buttons in an HBox
+        display(widgets.HBox([delete_button, up_button]), output)
+
+        def on_up_button_clicked(b):
+            with output:
+                output.clear_output()
+                self.up()
+
+        def on_down_button_clicked(b):
+            with output:
+                output.clear_output()
+                self.down()
+
+        up_button.on_click(on_up_button_clicked)
+        delete_button.on_click(on_down_button_clicked)
+
     def _throw_for_no_raycluster(self):
         api_instance = client.CustomObjectsApi(api_config_handler())
         try:
@@ -198,8 +237,12 @@ def down(self):
                     plural="appwrappers",
                     name=self.app_wrapper_name,
                 )
+                print(f"AppWrapper: '{self.config.name}' has successfully been deleted")
             else:
                 self._component_resources_down(namespace, api_instance)
+                print(
+                    f"Ray Cluster: '{self.config.name}' has successfully been deleted"
+                )
         except Exception as e:  # pragma: no cover
             return _kube_api_error_handling(e)