Add UTM attribution to outbound wavespeed.ai links

app.py

@@ -38,7 +38,8 @@ SPEC = {
     "extra": {
         "enable_base64_output": False,
         "enable_sync_mode": False
-    }
+    },
+    "campaign": "flux2_klein_9b_edit"
 }
 
 SPEC["css"] = Path(__file__).with_name("style.css").read_text(encoding="utf-8")

spaceui.py

@@ -35,11 +35,20 @@ import gradio as gr
 
 import wavespeed as ws
 
-DASH = "https://wavespeed.ai/dashboard"
+SITE = "https://wavespeed.ai"
 
 
-def _model_link(model: str) -> str:
-    return f"https://wavespeed.ai/models/{model}"
+def link(path: str, campaign: str) -> str:
+    """Build an outbound wavespeed.ai URL carrying UTM attribution.
+
+    Without these, traffic from Hugging Face lands in analytics as plain
+    referral with no way to tell which Space produced it.
+    """
+    sep = "&" if "?" in path else "?"
+    return (
+        f"{SITE}{path}{sep}utm_source=huggingface&utm_medium=space"
+        f"&utm_campaign={campaign}"
+    )
 
 
 def _collect(spec, key, values, progress):
@@ -85,6 +94,7 @@ def _collect(spec, key, values, progress):
 def build(spec):
     """Return a configured gr.Blocks for this Space."""
     css = spec["css"]
+    camp = spec["campaign"]
     outputs_are_video = spec["output"] == "video"
 
     with gr.Blocks(
@@ -94,7 +104,8 @@ def build(spec):
         gr.HTML(
             f"""
             <div class="hero-container">
-              <div class="hero-badge">WAVESPEED AI</div>
+              <a class="hero-badge" href="{link('/', camp)}"
+                 target="_blank" rel="noopener">WAVESPEED AI</a>
               <h1 class="hero-title">{spec['title']}</h1>
               <p class="hero-desc">{spec['tagline']}</p>
             </div>
@@ -111,8 +122,8 @@ def build(spec):
                 scale=4,
             )
             gr.HTML(
-                f'<a class="get-key-btn" href="{DASH}" target="_blank" '
-                'rel="noopener">Get a key</a>'
+                f'<a class="get-key-btn" href="{link("/dashboard", camp)}" '
+                'target="_blank" rel="noopener">Get a key</a>'
             )
         gr.Markdown(
             "Your key is sent only to `api.wavespeed.ai` to run this model. "
@@ -145,11 +156,14 @@ def build(spec):
         gr.HTML(
             f"""
             <div class="cta-container">
-              <p class="cta-desc">Model:
-                <a href="{_model_link(spec['model'])}" target="_blank" rel="noopener">
-                  <code>{spec['model']}</code></a>
-                &middot; <a href="https://wavespeed.ai/docs" target="_blank"
-                  rel="noopener">API docs</a>
+              <p class="cta-desc">Runs
+                <a href="{link('/models/' + spec['model'], camp)}"
+                   target="_blank" rel="noopener"><code>{spec['model']}</code></a>
+                on WaveSpeed &middot;
+                <a href="{link('/models', camp)}" target="_blank"
+                   rel="noopener">Browse all models</a> &middot;
+                <a href="{link('/docs', camp)}" target="_blank"
+                   rel="noopener">API docs</a>
               </p>
             </div>
             """

style.css

@@ -16,6 +16,7 @@
 
 .hero-badge {
 	display: inline-block;
+	text-decoration: none;
 	background: linear-gradient(135deg, #4f46e5, #2563eb);
 	color: #fff;
 	font-size: 0.66rem;
@@ -26,6 +27,11 @@
 	margin-bottom: 16px;
 }
 
+.hero-badge:hover {
+	filter: brightness(1.1);
+	text-decoration: none;
+}
+
 .hero-title {
 	font-size: 2.4rem;
 	font-weight: 800;