Update visualizer_app.py

visualizer_app.py

@@ -63,7 +63,9 @@ with app.setup:
         if creds_var_name in globals():
             creds_dict = globals()[creds_var_name]
             if isinstance(creds_dict, dict) and key in creds_dict:
-                return creds_dict[key]
+                # Return empty string if the value is None
+                value = creds_dict[key]
+                return "" if value is None else value
         return default
 
 @app.cell
@@ -109,9 +111,8 @@ def _(client_setup, wx_api_key):
 @app.cell
 def _():
     baked_in_creds = {
-            "purpose": "",
-            "api_key": "",
-            "project_id": "",
+            "api_key": os.environ.get("WX_APIKEY"),
+            "project_id": os.environ.get("WX_PROJECT_ID"),
             "space_id": "",
         }
     return baked_in_creds
@@ -127,39 +128,38 @@ def client_instantiation(
 ):
     ### 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
+            )
+    
+            if project_id:
+                project_client = APIClient(credentials=wx_credentials, project_id=project_id)
+            else:
+                project_client = None
+    
+            if space_id:
+                deployment_client = APIClient(credentials=wx_credentials, space_id=space_id)
+            else:
+                deployment_client = None
 
-        if project_id:
-            project_client = APIClient(credentials=wx_credentials, project_id=project_id)
-        else:
+            client_status = mo.md("### ✅ Client Instantiation Successful ✅")
+            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"
             project_client = None
-
-        if space_id:
-            deployment_client = APIClient(credentials=wx_credentials, space_id=space_id)
-        else:
-            deployment_client = None
-
-            
-        if project_client is not None:
-            task_credentials_details = setup_task_credentials(project_client)
-        else:
-            task_credentials_details = setup_task_credentials(deployment_client)
-            
+            deployment_client = None            
     else:
         wx_credentials = None
         project_client = None
         deployment_client = None
-        task_credentials_details = None
-
-    client_status = mo.md("### Client Instantiation Status will turn Green When Ready")
-
-    if project_client is not None or deployment_client is not None:
-        client_callout_kind = "success"
-    else:
+        client_status = mo.md("### This box will turn Green if client instantiation is successful")
         client_callout_kind = "neutral"
+
     return (
         client_callout_kind,
         client_status,
@@ -350,7 +350,7 @@ def file_and_model_select(
     get_embedding_model_list,
     run_upload_button,
 ):
-    select_stack = mo.hstack([get_embedding_model_list(), mo.vstack([file_loader, run_upload_button], align="center")], justify="space-around", align="center", widths=[0.3,0.3])
+    select_stack = mo.hstack([get_embedding_model_list(), mo.vstack([mo.md("Drag & Drop or Double Click to select PDFs, then press **Load Files**"),file_loader, run_upload_button], align="center")], justify="space-around", align="center", widths=[0.3,0.3])
     return (select_stack,)
 
 
@@ -384,13 +384,13 @@ def client_instantiation_form():
         > If you provide both you can switch the active one in the dropdown.
     ''')
         .batch(
-            wx_region = mo.ui.dropdown(regions, label="Select your watsonx.ai region:", value="US", searchable=True),
+            wx_region = mo.ui.dropdown(regions, label="Select your watsonx.ai region:", value="EU", searchable=True),
             wx_api_key = mo.ui.text(placeholder="Add your IBM Cloud api-key...", label="IBM Cloud Api-key:", 
                                     kind="password", value=get_cred_value('api_key', creds_var_name='baked_in_creds')),
             project_id = mo.ui.text(placeholder="Add your watsonx.ai project_id...", label="Project_ID:", 
-                                    kind="text", value=get_cred_value('project_id', creds_var_name='baked_in_creds')),
+                                    kind="password", value=get_cred_value('project_id', creds_var_name='baked_in_creds')),
             space_id = mo.ui.text(placeholder="Add your watsonx.ai space_id...", label="Space_ID:", 
-                                  kind="text", value=get_cred_value('space_id', creds_var_name='baked_in_creds'))
+                                  kind="password", value=get_cred_value('space_id', creds_var_name='baked_in_creds'))
         ,)
         .form(show_clear_button=True, bordered=False)
     )
@@ -720,8 +720,8 @@ def sentence_splitter_setup():
     
     ''')
         .batch(
-            chunk_size = mo.ui.slider(start=100, stop=5000, step=1, label="**Chunk Size:**", value=350, show_value=True, full_width=True),
-            chunk_overlap = mo.ui.slider(start=1, stop=1000, step=1, label="**Chunk Overlap** *(Must always be smaller than Chunk Size)* **:**", value=50, show_value=True, full_width=True),
+            chunk_size = mo.ui.slider(start=100, stop=5000, step=1, label="**Chunk Size:**", value=275, show_value=True, full_width=True),
+            chunk_overlap = mo.ui.slider(start=0, stop=1000, step=1, label="**Chunk Overlap** *(Must always be smaller than Chunk Size)* **:**", value=0, show_value=True, full_width=True),
             separator = mo.ui.text(placeholder="Define a separator", label="**Separator:**", kind="text", value=" "),
             paragraph_separator = mo.ui.text(placeholder="Define a paragraph separator", 
                                              label="**Paragraph Separator:**", kind="text", 
@@ -731,7 +731,7 @@ def sentence_splitter_setup():
                                                   value="[^,.;?!]+[,.;?!]?"),
             include_metadata= mo.ui.checkbox(value=True, label="**Include Metadata**")
         )
-        .form(show_clear_button=True, bordered=False)
+        .form(show_clear_button=True, bordered=False, submit_button_label="Chunk Documents")
     )
     return (sentence_splitter_config,)
 
@@ -1651,7 +1651,7 @@ def prepare_doc_select(sentence_splitter_config):
             full_width=True, 
             show_value=True, 
             label="**Select a Range of Chunks to Visualize:**"
-        ).form(submit_button_disabled=check_state(sentence_splitter_config.value))
+        ).form(submit_button_disabled=check_state(sentence_splitter_config.value), submit_button_label="Change Document View Range")
 
         return llama_docs
     return (prepare_document_selection,)
@@ -1839,7 +1839,7 @@ def create_baseline_chart(
 def test_query(get_chunk_state):
     placeholder = """How can i use watsonx.data to perform vector search?"""
 
-    query = mo.ui.text_area(label="**Write text to check:**", full_width=True, rows=8, value=placeholder).form(show_clear_button=True, submit_button_disabled=check_state(get_chunk_state()))
+    query = mo.ui.text_area(label="**Write text to check:**", full_width=True, rows=8, value=placeholder).form(show_clear_button=True, submit_button_disabled=check_state(get_chunk_state()),  submit_button_label="Query and View Visualization")
     return (query,)