agent update

app.py

@@ -526,29 +526,55 @@ def on_select_reset(agent_id):
     return map_html, summary, raw_text, demo_text, chain_html, 0, gr.update(value="▶ Stage 1: Feature Extraction")
 
 
+
+SHOWCASE_AGENTS = sample_agents[:8]
+
+
+def build_agent_cards(selected_id):
+    selected_id = int(selected_id)
+    parts = []
+    parts.append("<div style='display:grid;grid-template-columns:repeat(4,1fr);gap:10px;padding:4px 0;'>")
+    for aid in SHOWCASE_AGENTS:
+        row = demo[demo["agent_id"] == aid].iloc[0]
+        age = int(row["age"]) if row["age"] > 0 else "?"
+        sex = SEX_MAP.get(int(row["sex"]), "?")
+        edu = EDU_MAP.get(int(row["education"]), "?")
+        inc = INC_MAP.get(int(row["hh_income"]), "?")
+        is_sel = (aid == selected_id)
+        sel_style = "border:2px solid #c0392b;background:#fff8f8;box-shadow:0 2px 8px rgba(192,57,43,0.15);"
+        nor_style = "border:1.5px solid #ddd;background:#fafafa;box-shadow:0 1px 3px rgba(0,0,0,0.06);"
+        style = sel_style if is_sel else nor_style
+        dot = "<span style='display:inline-block;width:8px;height:8px;border-radius:50%;background:#c0392b;margin-right:5px;'></span>" if is_sel else ""
+        js = "var t=document.querySelector('#agent_hidden_input textarea');t.value='AID';t.dispatchEvent(new Event('input',{bubbles:true}));".replace("AID", str(aid))
+        parts.append(
+            "<div onclick=\"" + js + "\" style=\"cursor:pointer;border-radius:10px;padding:10px 13px;transition:all 0.2s;" + style + "\">"
+            "<div style='font-size:11px;font-weight:700;color:#c0392b;margin-bottom:3px;font-family:monospace;'>" + dot + "Agent #" + str(aid) + "</div>"
+            "<div style='font-size:11px;color:#333;line-height:1.6;'>"
+            "<b>Age:</b> " + str(age) + " &nbsp; <b>Sex:</b> " + sex + "<br>"
+            "<b>Edu:</b> " + edu + "<br>"
+            "<b>Income:</b> " + inc + "</div></div>"
+        )
+    parts.append("</div>")
+    return "".join(parts)
+
+
 # ── UI ────────────────────────────────────────────────────────────────────────
 
 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[1]
+**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.
-
 """)
 
-    with gr.Row():
-        agent_dd = gr.Dropdown(
-            choices=[str(a) for a in sample_agents],
-            label="Select Agent",
-            value=str(sample_agents[0]),
-            scale=1
-        )
-        demo_label = gr.Textbox(
-            label="Ground Truth Demographics",
-            interactive=False,
-            scale=4
-        )
+    gr.Markdown("### Select Agent")
+    agent_cards = gr.HTML(value=build_agent_cards(SHOWCASE_AGENTS[0]))
+    agent_hidden = gr.Textbox(
+        value=str(SHOWCASE_AGENTS[0]),
+        visible=False,
+        elem_id="agent_hidden_input"
+    )
 
     with gr.Row():
         with gr.Column(scale=1):
@@ -568,16 +594,21 @@ with gr.Blocks(title="HiCoTraj Demo", theme=gr.themes.Soft()) as app:
             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_reset, inputs=agent_dd,
-        outputs=[map_out, summary_out, raw_out, demo_label, chain_out, step_state, run_btn]
+    def on_agent_click(agent_id):
+        cards_html = build_agent_cards(agent_id)
+        map_html, summary, raw_text, chain_html, step, btn = on_select_reset(agent_id)
+        return cards_html, map_html, summary, raw_text, chain_html, step, btn
+
+    agent_hidden.change(
+        fn=on_agent_click, inputs=agent_hidden,
+        outputs=[agent_cards, map_out, summary_out, raw_out, chain_out, step_state, run_btn]
     )
     app.load(
-        fn=on_select_reset, inputs=agent_dd,
-        outputs=[map_out, summary_out, raw_out, demo_label, chain_out, step_state, run_btn]
+        fn=on_select_reset, inputs=agent_hidden,
+        outputs=[map_out, summary_out, raw_out, chain_out, step_state, run_btn]
     )
     run_btn.click(
-        fn=handle_btn, inputs=[agent_dd, step_state],
+        fn=handle_btn, inputs=[agent_hidden, step_state],
         outputs=[chain_out, step_state, run_btn]
     )
 
@@ -590,8 +621,8 @@ with gr.Blocks(title="HiCoTraj Demo", theme=gr.themes.Soft()) as app:
             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],
+        fn=toggle_raw, inputs=[agent_hidden, raw_out],
         outputs=[raw_out, show_all_btn]
     )
 
-app.launch(show_error=True, share=True)
+app.launch(show_error=True)