cotchange

app.py

@@ -6,7 +6,7 @@ import os
 import re
 import json
 
-BASE        = os.path.dirname(os.path.abspath(__file__))
+BASE        = os.path.dirname(os.path.abspath(__file__)) if "__file__" in dir() else os.getcwd()
 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")
@@ -115,46 +115,47 @@ def build_mobility_summary(agent_sp):
     obs_end   = agent_sp["end_datetime"].max().strftime("%Y-%m-%d")
     days      = (agent_sp["end_datetime"].max() - agent_sp["start_datetime"].min()).days
 
-    lines = [
-        "MOBILITY TRAJECTORY DATA",
-        "===========================",
-        f"Observation Period: {obs_start} to {obs_end} ({days} days)",
-        f"Total Stay Points: {len(agent_sp)}",
-        f"Unique Locations: {agent_sp['name'].nunique()}",
-        "",
-        "LOCATION PATTERNS",
-        "----------------",
-    ]
-    for i, (name, row) in enumerate(top5.iterrows(), 1):
-        lines += [f"{i}. {name}",
-                  f"   Visits: {int(row['visits'])} times",
-                  f"   Average Duration: {int(row['avg_dur'])} minutes", ""]
+    # Top activity types
+    act_counts = agent_sp["act_label"].value_counts().head(3)
+    top_acts = ", ".join(f"{a} ({n})" for a, n in act_counts.items())
 
+    # Time of day
     agent_sp2 = agent_sp.copy()
     agent_sp2["hour"] = agent_sp2["start_datetime"].dt.hour
     def tod(h):
-        if 5 <= h < 12:  return "morning"
-        if 12 <= h < 17: return "afternoon"
-        if 17 <= h < 21: return "evening"
-        return "night"
+        if 5 <= h < 12:  return "Morning"
+        if 12 <= h < 17: return "Afternoon"
+        if 17 <= h < 21: return "Evening"
+        return "Night"
     agent_sp2["tod"] = agent_sp2["hour"].apply(tod)
-    tod_pct = (agent_sp2["tod"].value_counts(normalize=True) * 100).round(0).astype(int)
+    peak_tod = agent_sp2["tod"].value_counts().idxmax()
     agent_sp2["is_weekend"] = agent_sp2["start_datetime"].dt.dayofweek >= 5
     wd_pct = int((~agent_sp2["is_weekend"]).mean() * 100)
 
-    lines += ["TEMPORAL PATTERNS", "----------------", "Activity by Time of Day:"]
-    for k, v in tod_pct.items():
-        lines.append(f"- {k}: {v}%")
-    lines += ["", "Weekday vs Weekend:",
-              f"- weekday: {wd_pct}%", f"- weekend: {100 - wd_pct}%"]
+    lines = [
+        f"Period: {obs_start} ~ {obs_end}  ({days} days)",
+        f"Stay points: {len(agent_sp)}  |  Unique locations: {agent_sp['name'].nunique()}",
+        f"Weekday/Weekend: {wd_pct}% / {100-wd_pct}%  |  Peak time: {peak_tod}",
+        f"Top activities: {top_acts}",
+        "",
+        "Top Locations:",
+    ]
+    for i, (name, row) in enumerate(top5.iterrows(), 1):
+        lines.append(f"  {i}. {name}  —  {int(row['visits'])} visits, avg {int(row['avg_dur'])} min")
+
     return "\n".join(lines)
 
 
-def build_weekly_checkin(agent_sp):
-    lines = ["WEEKLY CHECK-IN SUMMARY", "======================="]
+def build_weekly_checkin(agent_sp, max_days=None):
     agent_sp2 = agent_sp.copy()
     agent_sp2["date"] = agent_sp2["start_datetime"].dt.date
-    for date, grp in agent_sp2.groupby("date"):
+    all_dates = sorted(agent_sp2["date"].unique())
+    dates_to_show = all_dates[:max_days] if max_days else all_dates
+    total_days = len(all_dates)
+
+    lines = ["WEEKLY CHECK-IN SUMMARY", "======================="]
+    for date in dates_to_show:
+        grp = agent_sp2[agent_sp2["date"] == date]
         dow   = grp["start_datetime"].iloc[0].strftime("%A")
         label = "Weekend" if grp["start_datetime"].iloc[0].dayofweek >= 5 else "Weekday"
         lines.append(f"\n--- {dow}, {date} ({label}) ---")
@@ -166,6 +167,8 @@ def build_weekly_checkin(agent_sp):
                 f"({int(row['duration_min'])} mins): "
                 f"{row['name']} - {row['act_label']}"
             )
+    if max_days and total_days > max_days:
+        lines.append(f"\n... ({total_days - max_days} more days)")
     return "\n".join(lines)
 
 
@@ -432,18 +435,41 @@ def on_select(agent_id):
     return map_html, raw_text, demo_text, chain_html
 
 
-def run_reveal(agent_id):
-    import time
+def run_step(agent_id, step):
+    """Reveal one stage per click. step: 0->1->2->done(-1)"""
     agent_id = int(agent_id)
     s1, s2, s3 = get_cot(agent_id)
+    next_step = step + 1
+    if next_step == 1:
+        html = render_chain(s1, "", "", status="running2")
+        label = "▶ Stage 2: Behavioral Analysis"
+        return html, 1, gr.update(value=label)
+    elif next_step == 2:
+        html = render_chain(s1, s2, "", status="running3")
+        label = "▶ Stage 3: Demographic Inference"
+        return html, 2, gr.update(value=label)
+    else:
+        html = render_chain(s1, s2, s3, status="done")
+        return html, -1, gr.update(value="↺ Reset")
+
+
+def handle_btn(agent_id, step):
+    if step == -1:
+        html = render_chain("", "", "", status="idle")
+        return html, 0, gr.update(value="▶ Stage 1: Feature Extraction")
+    return run_step(agent_id, step)
 
-    yield render_chain("", "", "", status="running1")
-    time.sleep(0.8)
-    yield render_chain(s1, "", "", status="running2")
-    time.sleep(0.8)
-    yield render_chain(s1, s2, "", status="running3")
-    time.sleep(0.8)
-    yield render_chain(s1, s2, s3, status="done")
+
+def on_select_reset(agent_id):
+    agent_id_int = int(agent_id)
+    agent_sp   = sp[sp["agent_id"] == agent_id_int].sort_values("start_datetime")
+    agent_demo = demo[demo["agent_id"] == agent_id_int].iloc[0]
+    map_html   = build_map(agent_sp)
+    demo_text  = build_demo_text(agent_demo)
+    summary    = build_mobility_summary(agent_sp)
+    raw_text   = build_weekly_checkin(agent_sp, max_days=2)
+    chain_html = render_chain("", "", "", status="idle")
+    return map_html, summary, raw_text, demo_text, chain_html, 0, gr.update(value="▶ Stage 1: Feature Extraction")
 
 
 # ── UI ────────────────────────────────────────────────────────────────────────
@@ -470,28 +496,43 @@ with gr.Blocks(title="HiCoTraj Demo", theme=gr.themes.Soft()) as app:
             gr.Markdown("### Trajectory Map")
             map_out = gr.HTML()
             gr.Markdown("### NUMOSIM Raw Data")
-            raw_out = gr.Textbox(
-                lines=25, interactive=False,
-                label="Mobility Summary + Weekly Check-in"
-            )
+            with gr.Tabs():
+                with gr.Tab("Summary"):
+                    summary_out = gr.Textbox(lines=10, interactive=False, label="")
+                with gr.Tab("Raw Data"):
+                    raw_out = gr.Textbox(lines=10, interactive=False, label="")
+                    show_all_btn = gr.Button("Show All Days", size="sm", variant="secondary")
 
         with gr.Column(scale=1):
             gr.Markdown("### Hierarchical Chain-of-Thought Reasoning")
-            run_btn   = gr.Button("▶ Run HiCoTraj", variant="primary")
-            chain_out = gr.HTML(value=render_chain("", "", "", status="idle"))
+            step_state = gr.State(value=0)
+            run_btn    = gr.Button("▶ Stage 1: Feature Extraction", variant="primary")
+            chain_out  = gr.HTML(value=render_chain("", "", "", status="idle"))
 
     agent_dd.change(
-        fn=on_select, inputs=agent_dd,
-        outputs=[map_out, raw_out, demo_label, chain_out]
+        fn=on_select_reset, inputs=agent_dd,
+        outputs=[map_out, summary_out, raw_out, demo_label, chain_out, step_state, run_btn]
     )
     app.load(
-        fn=on_select, inputs=agent_dd,
-        outputs=[map_out, raw_out, demo_label, chain_out]
+        fn=on_select_reset, inputs=agent_dd,
+        outputs=[map_out, summary_out, raw_out, demo_label, chain_out, step_state, run_btn]
     )
     run_btn.click(
-        fn=run_reveal, inputs=agent_dd,
-        outputs=chain_out
+        fn=handle_btn, inputs=[agent_dd, step_state],
+        outputs=[chain_out, step_state, run_btn]
+    )
+
+    def toggle_raw(agent_id, current_text):
+        agent_id_int = int(agent_id)
+        agent_sp = sp[sp["agent_id"] == agent_id_int].sort_values("start_datetime")
+        if "more days" in current_text:
+            return build_weekly_checkin(agent_sp), gr.update(value="Show Less")
+        else:
+            return build_weekly_checkin(agent_sp, max_days=2), gr.update(value="Show All Days")
+
+    show_all_btn.click(
+        fn=toggle_raw, inputs=[agent_dd, raw_out],
+        outputs=[raw_out, show_all_btn]
     )
 
-if __name__ == "__main__":
-    app.launch(share=True)
+app.launch(show_error=True)