changes

Author: abidlabs

examples/convert-gradio-to-static.py

@@ -0,0 +1,57 @@
+"""
+Example: deploy a Gradio Space during training, then convert it to a static Space once done.
+
+This demonstrates the Gradio -> Static conversion flow:
+1. Start training with a live Gradio dashboard (real-time updates)
+2. After training finishes, convert the Space to static (no server needed, cheaper)
+
+Usage:
+    python examples/convert-gradio-to-static.py
+"""
+
+import math
+import random
+import time
+
+import trackio
+
+PROJECT = f"gradio-to-static-{random.randint(100000, 999999)}"
+SPACE_ID = f"convert-demo-{random.randint(100000, 999999)}"
+EPOCHS = 10
+
+for run in range(2):
+    trackio.init(
+        project=PROJECT,
+        name=f"run-{run}",
+        config={"epochs": EPOCHS, "lr": 0.001 * (run + 1), "batch_size": 32},
+        space_id=SPACE_ID,
+        auto_log_gpu=False,
+    )
+
+    for epoch in range(EPOCHS):
+        progress = epoch / EPOCHS
+        loss = 2.0 * math.exp(-3 * progress) + 0.1 + random.gauss(0, 0.05)
+        acc = 0.95 / (1 + math.exp(-6 * (progress - 0.5))) + random.gauss(0, 0.02)
+
+        trackio.log(
+            {
+                "train/loss": round(max(0.01, loss), 4),
+                "train/accuracy": round(min(0.99, max(0, acc)), 4),
+            },
+            step=epoch,
+        )
+
+        trackio.log_system(
+            {
+                "cpu_percent": round(40 + epoch * 3 + random.uniform(-2, 2), 1),
+                "memory_gb": round(4.0 + epoch * 0.1 + random.uniform(-0.05, 0.05), 2),
+            }
+        )
+
+        time.sleep(0.3)
+
+    trackio.finish()
+
+print("\nTraining complete. Converting Gradio Space to static...")
+space_id = trackio.sync(project=PROJECT, space_id=SPACE_ID, sdk="static")
+print(f"Static dashboard: https://huggingface.co/spaces/{space_id}")

trackio/__init__.py

@@ -259,9 +259,6 @@ def init(
             space_url = deploy.SPACE_HOST_URL.format(
                 user_name=user_name, space_name=space_name
             )
-            print(
-                f"* View dashboard by going to: {deploy._BOLD_ORANGE}{space_url}{deploy._RESET}"
-            )
             if utils.is_in_notebook() and embed:
                 utils.embed_url_in_notebook(space_url)
     context_vars.current_project.set(project)

trackio/bucket_storage.py

@@ -43,7 +43,32 @@ def upload_project_to_bucket(project: str, bucket_id: str) -> None:
     huggingface_hub.batch_bucket_files(bucket_id, add=files_to_add)
 
 
+def _download_db_from_bucket(project: str, bucket_id: str) -> bool:
+    db_filename = SQLiteStorage.get_project_db_filename(project)
+    remote_path = f"trackio/{db_filename}"
+    local_path = SQLiteStorage.get_project_db_path(project)
+    local_path.parent.mkdir(parents=True, exist_ok=True)
+    try:
+        huggingface_hub.download_bucket_files(
+            bucket_id,
+            files=[(remote_path, str(local_path))],
+        )
+        return local_path.exists()
+    except Exception:
+        return False
+
+
 def upload_project_to_bucket_for_static(project: str, bucket_id: str) -> None:
+    db_path = SQLiteStorage.get_project_db_path(project)
+    local_db_empty = not db_path.exists() or db_path.stat().st_size == 0
+    if not local_db_empty:
+        with sqlite3.connect(str(db_path)) as conn:
+            count = conn.execute("SELECT COUNT(*) FROM metrics").fetchone()[0]
+            local_db_empty = count == 0
+
+    if local_db_empty:
+        _download_db_from_bucket(project, bucket_id)
+
     with tempfile.TemporaryDirectory() as tmp_dir:
         output_dir = Path(tmp_dir)
         SQLiteStorage.export_for_static_space(project, output_dir)