legend

app.py

@@ -432,6 +432,32 @@ def build_map(agent_sp):
             )
         ).add_to(m)
 
+    legend_html = """
+    <div style="
+        position:fixed; bottom:18px; left:18px; z-index:9999;
+        background:rgba(255,255,255,0.92); border-radius:8px;
+        padding:8px 12px; font-size:11px; font-family:sans-serif;
+        box-shadow:0 1px 5px rgba(0,0,0,0.2); line-height:1.8;
+    ">
+        <div style="font-weight:600;margin-bottom:4px;">Stay Point Legend</div>
+        <div style="display:flex;align-items:center;gap:6px;">
+            <svg width="60" height="10">
+              <defs><linearGradient id="lg" x1="0" x2="1" y1="0" y2="0">
+                <stop offset="0%" stop-color="#ffcc00"/>
+                <stop offset="100%" stop-color="#8b0000"/>
+              </linearGradient></defs>
+              <rect width="60" height="10" rx="4" fill="url(#lg)"/>
+            </svg>
+            <span>Earlier &rarr; Later</span>
+        </div>
+        <div style="display:flex;align-items:center;gap:6px;margin-top:2px;">
+            <svg width="14" height="14"><circle cx="7" cy="7" r="5" fill="#cc4444" opacity="0.5"/></svg>
+            <span>Movement path</span>
+        </div>
+        <div style="color:#999;font-size:10px;margin-top:2px;">Click dot for details</div>
+    </div>
+    """
+    m.get_root().html.add_child(folium.Element(legend_html))
     m.get_root().width = "100%"
     m.get_root().height = "420px"
     return m._repr_html_()
@@ -506,10 +532,9 @@ with gr.Blocks(title="HiCoTraj Demo", theme=gr.themes.Soft()) as app:
     gr.Markdown("## HiCoTraj — Trajectory Visualization & Hierarchical CoT Demo")
     gr.Markdown("*Zero-Shot Demographic Reasoning via Hierarchical Chain-of-Thought Prompting from Trajectory*  ·  ACM SIGSPATIAL GeoGenAgent 2025")
     gr.Markdown("""
-**Dataset:** NUMOSIM — a synthetic mobility dataset with realistic activity patterns across 6,000 agents.
-> Stanford C, Adari S, Liao X, et al. *NUMoSim: A Synthetic Mobility Dataset with Anomaly Detection Benchmarks.* ACM SIGSPATIAL Workshop on Geospatial Anomaly Detection, 2024.
+**Dataset:** NUMOSIM[1]
+> [1]Stanford C, Adari S, Liao X, et al. *NUMoSim: A Synthetic Mobility Dataset with Anomaly Detection Benchmarks.* ACM SIGSPATIAL Workshop on Geospatial Anomaly Detection, 2024.
 
-**Map legend:** 🔴 Early visits (light red) → 🔴 Recent visits (dark red)  ·  Each dot = one stay point  ·  Click dot for details
 """)
 
     with gr.Row():