feat: add ui cluster creation and deletion buttons

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

Author: Bobbins228

src/codeflare_sdk/cluster/cluster.py

@@ -42,6 +42,7 @@
     RayCluster,
     RayClusterStatus,
 )
+from .widgets import cluster_up_down_buttons, is_notebook
 from kubernetes import client, config
 from kubernetes.utils import parse_quantity
 import yaml
@@ -51,6 +52,9 @@
 from kubernetes import config
 from kubernetes.client.rest import ApiException
 
+import ipywidgets as widgets
+from IPython.display import display
+
 
 class Cluster:
     """
@@ -71,6 +75,8 @@ 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 is_notebook():
+            self._cluster_up_down_buttons()
 
     @property
     def _client_headers(self):
@@ -156,11 +162,34 @@ 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):
+        up_button, delete_button = cluster_up_down_buttons(self.config.appwrapper)
+        output = widgets.Output()
+        # Display the buttons in an HBox
+        display(widgets.HBox([up_button, delete_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 +227,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)
 

src/codeflare_sdk/cluster/widgets.py

@@ -0,0 +1,56 @@
+# Copyright 2022 IBM, Red Hat
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""
+The widgets sub-module contains the ui widgets created using the ipywidgets package.
+"""
+import ipywidgets as widgets
+import os
+
+
+def cluster_up_down_buttons(appwrapper: bool) -> widgets.Button:
+    """
+    The cluster_up_down_buttons function returns two button widgets for a create and delete button.
+    The function accepts a bool(appwrapper) for distinguishing between resource type for the tool tip.
+    """
+    resource = "Ray Cluster"
+    if appwrapper:
+        resource = "AppWrapper"
+
+    up_button = widgets.Button(
+        description="Cluster Up",
+        tooltip=f"Create the {resource}",
+        icon="play",
+    )
+
+    delete_button = widgets.Button(
+        description="Cluster Down",
+        tooltip=f"Delete the {resource}",
+        icon="trash",
+    )
+
+    return up_button, delete_button
+
+
+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.
+    """
+    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
+        return True
+    else:
+        return False