Lazy load the UI when trackio is imported (#431)

Co-authored-by: gradio-pr-bot <gradio-pr-bot@users.noreply.github.com>

Author: abidlabs

.changeset/polite-pugs-kick.md

@@ -0,0 +1,5 @@
+---
+"trackio": patch
+---
+
+feat:Lazy load the UI when trackio is imported

trackio/__init__.py

@@ -25,7 +25,6 @@
 from trackio.sqlite_storage import SQLiteStorage
 from trackio.table import Table
 from trackio.typehints import UploadEntry
-from trackio.ui.main import CSS, HEAD, demo
 from trackio.utils import TRACKIO_DIR, TRACKIO_LOGO_DIR
 
 logging.getLogger("httpx").setLevel(logging.WARNING)
@@ -69,6 +68,14 @@
 config = {}
 
 
+def _get_demo():
+    # Lazy import to avoid initializing Gradio Blocks (and FastAPI) at import time,
+    # which causes import lock errors for libraries that just `import trackio`.
+    from trackio.ui.main import CSS, HEAD, demo
+
+    return demo, CSS, HEAD
+
+
 def init(
     project: str,
     name: str | None = None,
@@ -159,6 +166,7 @@ def init(
         raise LocalTokenNotFoundError(
             f"You must be logged in to Hugging Face locally when `space_id` is provided to deploy to a Space. {e}"
         ) from e
+    demo, CSS, HEAD = _get_demo()
     url = context_vars.current_server.get()
     share_url = context_vars.current_share_server.get()
 
@@ -518,6 +526,8 @@ def show(
             `share_url`: The public share URL of the dashboard.
             `full_url`: The full URL of the dashboard including the write token (will use the public share URL if launched publicly, otherwise the local URL).
     """
+    demo, CSS, HEAD = _get_demo()
+
     if color_palette is not None:
         os.environ["TRACKIO_COLOR_PALETTE"] = ",".join(color_palette)