changes

Author: abidlabs

trackio/__init__.py

@@ -158,8 +158,11 @@ def init(
         settings (`Any`, *optional*):
             Not used. Provided for compatibility with `wandb.init()`.
         embed (`bool`, *optional*, defaults to `True`):
-            If running inside a jupyter/Colab notebook, whether the dashboard should
-            automatically be embedded in the cell when trackio.init() is called.
+            If running inside a Jupyter/Colab notebook, whether the dashboard should
+            automatically be embedded in the cell when trackio.init() is called. For
+            local runs, this launches a local Gradio app and embeds it. For Space runs,
+            this embeds the Space URL. In Colab, the local dashboard will be accessible
+            via a public share URL (default Gradio behavior).
         auto_log_gpu (`bool` or `None`, *optional*, defaults to `None`):
             Controls automatic GPU metrics logging. If `None` (default), GPU logging
             is automatically enabled when `nvidia-ml-py` is installed and an NVIDIA
@@ -194,6 +197,8 @@ def init(
             context_vars.current_server.set(url)
             context_vars.current_space_id.set(space_id)
 
+    _should_embed_local = False
+
     if (
         context_vars.current_project.get() is None
         or context_vars.current_project.get() != project
@@ -207,7 +212,9 @@ def init(
             )
         if space_id is None:
             print(f"* Trackio metrics logged to: {TRACKIO_DIR}")
-            utils.print_dashboard_instructions(project)
+            _should_embed_local = embed and utils.is_in_notebook()
+            if not _should_embed_local:
+                utils.print_dashboard_instructions(project)
         else:
             deploy.create_space_if_not_exists(
                 space_id, space_storage, dataset_id, private
@@ -275,6 +282,10 @@ def init(
 
     context_vars.current_run.set(run)
     globals()["config"] = run.config
+
+    if _should_embed_local:
+        show(project=project, open_browser=False, block_thread=False)
+
     return run