Update app.py

app.py

@@ -201,46 +201,70 @@ def _(client_instantiation_form, os):
 
 
 @app.cell
-def _(
+def client_instantiation(
     APIClient,
     Credentials,
+    client,
+    client_key,
+    client_options,
+    client_selector,
     client_setup,
+    get_key_by_value,
+    mo,
     project_id,
     space_id,
+    wrap_with_spaces,
     wx_api_key,
     wx_url,
 ):
+    ### Instantiate the watsonx.ai client
     if client_setup:
-        wx_credentials = Credentials(url=wx_url, 
-                                                api_key=wx_api_key)
+        try:
+            wx_credentials = Credentials(url=wx_url, api_key=wx_api_key)
 
-        project_client = APIClient(credentials=wx_credentials, 
-                                                project_id=project_id) if project_id else None
+            project_client = (
+                APIClient(credentials=wx_credentials, project_id=project_id)
+                if project_id
+                else None
+            )
+
+            deployment_client = (
+                APIClient(credentials=wx_credentials, space_id=space_id)
+                if space_id
+                else None
+            )
 
-        deployment_client = APIClient(credentials=wx_credentials, 
-                                                space_id=space_id) if space_id else None
+            active_client_name = get_key_by_value(client_options, client_key)
+            client_status = mo.md(
+                f"### ✅ Client Instantiation Successful ✅\n\n"
+                f"{client_selector}\n\n"
+                f"**Active Client:**{wrap_with_spaces(active_client_name, prefix_spaces=5)}"
+            )
+            client_callout_kind = "success"
 
+        except Exception as e:
+            error_message = str(e)
+            client_status = mo.md(
+                f"### ❌ Client Instantiation Failed\n**Error:** {error_message}\n\nCheck your region selection and credentials"
+            )
+            client_callout_kind = "danger"
+            wx_credentials = project_client = deployment_client = None
     else:
         wx_credentials = project_client = deployment_client = None
-    return deployment_client, project_client
-
-
-@app.cell
-def _(
-    client,
-    client_key,
-    client_options,
-    client_selector,
-    get_key_by_value,
-    mo,
-    wrap_with_spaces,
-):
-    active_client_name = get_key_by_value(client_options,client_key)
-    client_status = mo.md(f"### Client Instantiation Status will turn Green When Ready\n\n {client_selector}\n\n**Active Client:**{wrap_with_spaces(active_client_name, prefix_spaces=5)}")
-
-    client_callout_kind = "success" if client is not None else "neutral"
-    return client_callout_kind, client_status
+        active_client_name = get_key_by_value(client_options, client_key)
+        client_status = mo.md(
+            f"### Client Instantiation Status will turn Green When Ready\n\n"
+            f"{client_selector}\n\n"
+            f"**Active Client:**{wrap_with_spaces(active_client_name, prefix_spaces=5)}"
+        )
+        client_callout_kind = "neutral"
 
+    return (
+        client_callout_kind,
+        client_status,
+        deployment_client,
+        project_client,
+    )
 
 @app.cell
 def _(client_callout_kind, client_instantiation_form, client_status, mo):