label fix

app.py

@@ -1,66 +1,84 @@
 import gradio as gr
 import pandas as pd
 import folium
+import numpy as np
 import os
 
-# ── Data paths ───────────────────────────────────────────────────────────────
-BASE = os.path.dirname(os.path.abspath(__file__))
+BASE        = os.path.dirname(os.path.abspath(__file__))
 STAY_POINTS = os.path.join(BASE, "data", "stay_points_sampled.csv")
 POI_PATH    = os.path.join(BASE, "data", "poi_sampled.csv")
 DEMO_PATH   = os.path.join(BASE, "data", "demographics_sampled.csv")
 
-AGE_MAP = {1:"<18", 2:"18-24", 3:"25-34", 4:"35-44", 5:"45-54", 6:"55-64", 7:"65+"}
-SEX_MAP = {1:"Male", 2:"Female"}
-EDU_MAP = {1:"No HS", 2:"HS Grad", 3:"Some College", 4:"Bachelor's", 5:"Graduate"}
-INC_MAP = {1:"<$10k", 2:"$10-15k", 3:"$15-25k", 4:"$25-35k", 5:"$35-50k",
-           6:"$50-75k", 7:"$75-100k", 8:"$100-125k", 9:"$125-150k", 10:">$150k"}
+SEX_MAP  = {1:"Male", 2:"Female", -8:"Unknown", -7:"Prefer not to answer"}
+EDU_MAP  = {1:"Less than HS", 2:"HS Graduate/GED", 3:"Some College/Associate",
+            4:"Bachelor's Degree", 5:"Graduate/Professional Degree",
+            -1:"N/A", -7:"Prefer not to answer", -8:"Unknown"}
+INC_MAP  = {1:"<$10,000", 2:"$10,000–$14,999", 3:"$15,000–$24,999",
+            4:"$25,000–$34,999", 5:"$35,000–$49,999", 6:"$50,000–$74,999",
+            7:"$75,000–$99,999", 8:"$100,000–$124,999", 9:"$125,000–$149,999",
+            10:"$150,000–$199,999", 11:"$200,000+",
+            -7:"Prefer not to answer", -8:"Unknown", -9:"Not ascertained"}
+RACE_MAP = {1:"White", 2:"Black or African American", 3:"Asian",
+            4:"American Indian or Alaska Native",
+            5:"Native Hawaiian or Other Pacific Islander",
+            6:"Multiple races", 97:"Other",
+            -7:"Prefer not to answer", -8:"Unknown"}
+ACT_MAP  = {0:"Transportation", 1:"Home", 2:"Work", 3:"School", 4:"ChildCare",
+            5:"BuyGoods", 6:"Services", 7:"EatOut", 8:"Errands", 9:"Recreation",
+            10:"Exercise", 11:"Visit", 12:"HealthCare", 13:"Religious",
+            14:"SomethingElse", 15:"DropOff"}
 
-# ── Load ─────────────────────────────────────────────────────────────────────
 print("Loading data...")
 sp   = pd.read_csv(STAY_POINTS)
 poi  = pd.read_csv(POI_PATH)
 demo = pd.read_csv(DEMO_PATH)
 
-sp  = sp.merge(poi, on="poi_id", how="left")
+sp = sp.merge(poi, on="poi_id", how="left")
 sp["start_datetime"] = pd.to_datetime(sp["start_datetime"], utc=True)
 sp["end_datetime"]   = pd.to_datetime(sp["end_datetime"],   utc=True)
 sp["duration_min"]   = ((sp["end_datetime"] - sp["start_datetime"]).dt.total_seconds() / 60).round(1)
 
+def parse_act_types(x):
+    try:
+        codes = list(map(int, str(x).strip("[]").split()))
+        return ", ".join(ACT_MAP.get(c, str(c)) for c in codes)
+    except:
+        return str(x)
+
+sp["act_label"] = sp["act_types"].apply(parse_act_types)
+
 sample_agents = sorted(sp["agent_id"].unique().tolist())
 print(f"Ready. {len(sample_agents)} agents loaded.")
 
-# ── Helpers ───────────────────────────────────────────────────────────────────
+
 def build_map(agent_sp):
-    import numpy as np
-    agent_sp = agent_sp.reset_index(drop=True)
+    agent_sp = agent_sp.reset_index(drop=True).copy()
     agent_sp["latitude"]  += np.random.uniform(-0.0003, 0.0003, len(agent_sp))
     agent_sp["longitude"] += np.random.uniform(-0.0003, 0.0003, len(agent_sp))
-    
+
     lat = agent_sp["latitude"].mean()
     lon = agent_sp["longitude"].mean()
     m = folium.Map(location=[lat, lon], zoom_start=12, tiles="CartoDB positron")
 
-    n = len(agent_sp)
     coords = list(zip(agent_sp["latitude"], agent_sp["longitude"]))
-    
     if len(coords) > 1:
         folium.PolyLine(coords, color="#aaaaaa", weight=1.5, opacity=0.4).add_to(m)
 
+    n = len(agent_sp)
     for i, row in agent_sp.iterrows():
-        # green → red gradient
         ratio = i / max(n - 1, 1)
         r = int(255 * ratio)
         g = int(255 * (1 - ratio))
         color = f"#{r:02x}{g:02x}33"
-
         folium.CircleMarker(
             location=[row["latitude"], row["longitude"]],
             radius=7, color=color, fill=True, fill_color=color, fill_opacity=0.9,
             popup=folium.Popup(
                 f"<b>#{i+1} {row['name']}</b><br>"
                 f"{row['start_datetime'].strftime('%a %m/%d %H:%M')}<br>"
-                f"{int(row['duration_min'])} min",
-                max_width=200
+                f"{int(row['duration_min'])} min<br>"
+                f"{row['act_label']}",
+                max_width=220
             )
         ).add_to(m)
 
@@ -68,23 +86,26 @@ def build_map(agent_sp):
     m.get_root().height = "500px"
     return m._repr_html_()
 
+
 def build_poi_sequence(agent_sp):
     lines = []
     for _, row in agent_sp.iterrows():
         lines.append(
             f"{row['start_datetime'].strftime('%a %m/%d')}  "
             f"{row['start_datetime'].strftime('%H:%M')}–{row['end_datetime'].strftime('%H:%M')}  "
-            f"({int(row['duration_min'])} min)  |  {row['name']}  |  {row['act_types']}"
+            f"({int(row['duration_min'])} min)  |  {row['name']}  |  {row['act_label']}"
         )
     return "\n".join(lines)
 
 
 def build_demo_text(row):
+    age  = int(row["age"]) if row["age"] > 0 else "Unknown"
     return (
-        f"Age: {AGE_MAP.get(row['age'], row['age'])}  |  "
-        f"Sex: {SEX_MAP.get(row['sex'], row['sex'])}  |  "
-        f"Education: {EDU_MAP.get(row['education'], row['education'])}  |  "
-        f"Income: {INC_MAP.get(row['hh_income'], row['hh_income'])}"
+        f"Age: {age}  |  "
+        f"Sex: {SEX_MAP.get(int(row['sex']), row['sex'])}  |  "
+        f"Race: {RACE_MAP.get(int(row['race']), row['race'])}  |  "
+        f"Education: {EDU_MAP.get(int(row['education']), row['education'])}  |  "
+        f"Income: {INC_MAP.get(int(row['hh_income']), row['hh_income'])}"
     )
 
 
@@ -95,7 +116,6 @@ def on_select(agent_id):
     return build_map(agent_sp), build_poi_sequence(agent_sp), build_demo_text(agent_demo)
 
 
-# ── UI ────────────────────────────────────────────────────────────────────────
 with gr.Blocks(title="HiCoTraj Demo", theme=gr.themes.Soft()) as app:
     gr.Markdown("## HiCoTraj: Trajectory Visualization")
 
@@ -104,7 +124,7 @@ with gr.Blocks(title="HiCoTraj Demo", theme=gr.themes.Soft()) as app:
                                  label="Select Agent", value=str(sample_agents[0]))
         demo_label = gr.Textbox(label="Ground Truth Demographics", interactive=False)
 
-    map_out = gr.HTML(label="Trajectory Map", value="<div style='height:500px'></div>")
+    map_out = gr.HTML(label="Trajectory Map")
     poi_out = gr.Textbox(label="POI Sequence", lines=20, interactive=False)
 
     agent_dd.change(fn=on_select, inputs=agent_dd, outputs=[map_out, poi_out, demo_label])