Update visualizer_app.py

visualizer_app.py

@@ -3,7 +3,6 @@
 # custom_css = ["./custom_header_font.css"]
 # ///
 
-
 import marimo
 
 __generated_with = "0.13.0"
@@ -89,7 +88,6 @@ def client_variables(client_instantiation_form):
     else:
         os.environ["WATSONX_APIKEY"] = ""
         project_id = space_id = wx_api_key = wx_url = None
-        
     return client_setup, project_id, space_id, wx_api_key, wx_url
 
 
@@ -115,21 +113,13 @@ def _():
         markdown_spacing,
     )
 
-
 @app.cell
 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,
 ):
@@ -137,48 +127,31 @@ def client_instantiation(
     if client_setup:
         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
             )
-
             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"
-
+            instantiation_success = True
+            instantiation_error = None
         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"
+            instantiation_success = False
+            instantiation_error = str(e)
             wx_credentials = project_client = deployment_client = None
     else:
         wx_credentials = project_client = deployment_client = None
-        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"
-
+        instantiation_success = None
+        instantiation_error = None
+    
     return (
-        client_callout_kind,
-        client_status,
         deployment_client,
+        instantiation_error,
+        instantiation_success,
         project_client,
     )
 
@@ -429,6 +402,42 @@ def instantiation_status(
     client_stack = mo.hstack([client_instantiation_form, client_callout], align="center", justify="space-around", gap=10)
     return (client_stack,)
 
+def _(
+    client,
+    client_key,
+    client_options,
+    client_selector,
+    client_setup,
+    get_key_by_value,
+    instantiation_error,
+    instantiation_success,
+    mo,
+    wrap_with_spaces,
+):
+    active_client_name = get_key_by_value(client_options, client_key)
+    
+    if client_setup:
+        if instantiation_success:
+            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"
+        else:
+            client_status = mo.md(
+                f"### ❌ Client Instantiation Failed\n**Error:** {instantiation_error}\n\nCheck your region selection and credentials"
+            )
+            client_callout_kind = "danger"
+    else:
+        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 active_client_name, client_callout_kind, client_status
 
 @app.cell
 def client_selector(deployment_client, project_client):
@@ -446,8 +455,7 @@ def client_selector(deployment_client, project_client):
 
     default_client = next(iter(client_options))
     client_selector = mo.ui.dropdown(client_options, value=default_client, label="**Select your active client:**")
-    
-    return (client_selector,)
+    return client_options, client_selector
 
 
 @app.cell
@@ -457,8 +465,7 @@ def active_client(client_selector):
         client = None
     else:
         client = client_key
-    return (client,)
-
+    return client, client_key
 
 @app.cell
 def emb_model_selection(client, set_embedding_model_list):