Update main_app.py

main_app.py

@@ -1,14 +1,31 @@
 import marimo
 
-__generated_with = "0.13.0"
+__generated_with = "0.14.12"
 app = marimo.App(width="medium")
 
 
-@app.cell
-def _():
+@app.setup:
+    import marimo as mo
+    import ast, base64, glob, io, json, mimetypes, os, re, tempfile, time, zipfile
+    from typing import Any, Dict, List, Optional, Pattern, Union, Callable, Set, Tuple
+    from datetime import datetime
+    from pathlib import Path
+    import importlib.util
+    import urllib3, uuid, sys, ssl
+    import json
+    import ast
+
+    from dotenv import load_dotenv
+    from ibm_watsonx_ai import APIClient, Credentials
+    from ibmcloudant.cloudant_v1 import CloudantV1, Document
+    from ibm_cloud_sdk_core.authenticators import IAMAuthenticator
     import marimo as mo
-    import os
-    return mo, os
+    import pandas as pd
+    import requests
+    import certifi
+
+    load_dotenv()
+
 
 
 @app.function
@@ -38,29 +55,6 @@ def _(mo):
 
 @app.cell
 def _(os):
-    ### Imports
-    from typing import Any, Dict, List, Optional, Pattern, Set, Union, Tuple
-    from ibm_watsonx_ai import APIClient, Credentials
-    from pathlib import Path
-    import importlib.util
-    import pandas as pd
-    import mimetypes
-    import requests
-    import zipfile
-    import polars
-    import urllib3
-    import tempfile
-    import importlib.util
-    import base64
-    import certifi
-    import uuid
-    import time
-    import json
-    import sys
-    import ssl
-    import ast
-    import io
-    import re
 
     # Set explicit temporary directory
     os.environ['TMPDIR'] = '/tmp/notebook_functions'
@@ -71,14 +65,6 @@ def _(os):
     # Make sure Python's tempfile module also uses this directory
     tempfile.tempdir = '/tmp/notebook_functions'
 
-    def get_iam_token(api_key):
-        return requests.post(
-            'https://iam.cloud.ibm.com/identity/token',
-            headers={'Content-Type': 'application/x-www-form-urlencoded'},
-            data={'grant_type': 'urn:ibm:params:oauth:grant-type:apikey', 'apikey': api_key},
-            verify=certifi.where()
-        ).json()['access_token']
-
     def setup_task_credentials(client):
         # Get existing task credentials
         existing_credentials = client.task_credentials.get_details()
@@ -92,37 +78,9 @@ def _(os):
         # Store new credentials
         return client.task_credentials.store()
 
-    def get_cred_value(key, creds_var_name="baked_in_creds", default=""): ### Helper for working with preset credentials
-        """
-        Helper function to safely get a value from a credentials dictionary.
-
-        Args:
-            key: The key to look up in the credentials dictionary.
-            creds_var_name: The variable name of the credentials dictionary.
-            default: The default value to return if the key is not found.
-
-        Returns:
-            The value from the credentials dictionary if it exists and contains the key,
-            otherwise returns the default value.
-        """
-        # Check if the credentials variable exists in globals
-        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 default
     return (
-        APIClient,
-        Credentials,
-        ast,
-        get_iam_token,
-        importlib,
-        json,
-        pd,
         setup_task_credentials,
-        sys,
         tempfile,
-        uuid,
     )
 
 
@@ -157,16 +115,6 @@ def _(client_instantiation_form, os):
         wx_url = None
     return client_setup, project_id, space_id, wx_api_key, wx_url
 
-
-@app.cell
-def _(client_setup, get_iam_token, wx_api_key):
-    if client_setup and wx_api_key is not None:
-        token = get_iam_token(wx_api_key)
-    else:
-        token = None
-    return
-
-
 @app.cell
 def _(mo):
     ### Credentials for the watsonx.ai SDK client