cot

app.py

@@ -207,37 +207,51 @@ CHAIN_CSS = """
 .hct-s3 .hct-title { color: #b0302a; }
 
 /* ── Prompt pill ── */
-.hct-prompt-wrap { padding: 0 14px 8px; }
-.hct-prompt-toggle {
-  display: inline-flex; align-items: center; gap: 5px;
-  font-family: 'DM Mono', monospace; font-size: 9px;
-  letter-spacing: 0.08em; text-transform: uppercase;
-  padding: 3px 9px; border-radius: 20px; cursor: pointer;
-  border: 1px solid currentColor; opacity: 0.45;
-  transition: opacity 0.2s; background: transparent;
+/* ── Paper-style prompt+response layout ── */
+.hct-paper-wrap { padding: 0 12px 10px; }
+.hct-paper-prompt {
+  background: #fffef5;
+  border: 1.5px dashed #c8b870;
+  border-radius: 8px;
+  padding: 8px 11px;
+  margin-bottom: 0;
+  position: relative;
 }
-.hct-prompt-toggle:hover { opacity: 0.8; }
-.hct-s1 .hct-prompt-toggle { color: #3a4a80; }
-.hct-s2 .hct-prompt-toggle { color: #7a4a10; }
-.hct-s3 .hct-prompt-toggle { color: #b0302a; }
-.hct-prompt-box {
-  display: none;
+.hct-paper-response {
+  background: #f8fffe;
+  border: 1.5px solid #8ab8a8;
+  border-radius: 8px;
+  padding: 8px 11px;
   margin-top: 6px;
-  background: rgba(0,0,0,0.03);
-  border-radius: 6px;
-  padding: 8px 10px;
-  font-family: 'DM Mono', monospace;
-  font-size: 10px;
-  line-height: 1.65;
-  color: #556;
-  white-space: pre-wrap;
-  word-break: break-word;
-  max-height: 150px;
-  overflow-y: auto;
-  border-left: 2px solid currentColor;
-  opacity: 0.7;
+  position: relative;
+}
+.hct-s2 .hct-paper-prompt { background: #fffbf0; border-color: #d4a840; }
+.hct-s2 .hct-paper-response { background: #fffbf0; border-color: #c89050; }
+.hct-s3 .hct-paper-prompt { background: #fff8f7; border-color: #d4453a; border-style: dashed; }
+.hct-s3 .hct-paper-response { background: #fff8f7; border-color: #c03030; }
+.hct-paper-tag {
+  display: inline-block;
+  font-family: 'DM Mono', monospace; font-size: 8.5px; font-weight: 600;
+  letter-spacing: 0.1em; text-transform: uppercase;
+  padding: 1px 6px; border-radius: 3px; margin-bottom: 5px;
+}
+.hct-paper-prompt .hct-paper-tag  { background: #f0e48a; color: #7a6010; }
+.hct-s2 .hct-paper-prompt .hct-paper-tag { background: #f5d580; color: #7a5010; }
+.hct-s3 .hct-paper-prompt .hct-paper-tag { background: #fac8c4; color: #a03028; }
+.hct-paper-response .hct-paper-tag { background: #c8e8d8; color: #206048; }
+.hct-s2 .hct-paper-response .hct-paper-tag { background: #f0dcb0; color: #7a5010; }
+.hct-s3 .hct-paper-response .hct-paper-tag { background: #f8c8c4; color: #902828; }
+.hct-paper-text {
+  font-size: 11px; line-height: 1.6; color: #333;
+  white-space: pre-wrap; word-break: break-word;
+}
+.hct-paper-connector {
+  display: flex; align-items: center; justify-content: center;
+  height: 14px; margin: 0 20px;
+}
+.hct-paper-connector-line {
+  width: 1px; height: 100%; background: #aaa;
 }
-.hct-prompt-box.open { display: block; }
 
 /* ── Body ── */
 .hct-body { padding: 12px 14px; }
@@ -362,15 +376,6 @@ CHAIN_CSS = """
 .hct-s2 .hct-dot { background: #c08040; }
 .hct-s3 .hct-dot { background: #d4453a; }
 </style>
-<script>
-function hctTogglePrompt(id) {
-  var box = document.getElementById(id);
-  var btn = document.getElementById(id + '-btn');
-  if (!box) return;
-  var open = box.classList.toggle('open');
-  btn.textContent = open ? '▲ hide prompt' : '▼ show prompt';
-}
-</script>
 """
 
 
@@ -394,10 +399,14 @@ def _parse_s1(text):
             locations.append((m.group(1).strip(), int(m.group(2))))
             continue
 
-        # Duration
+        # Duration — 4 formats
         m2 = re.match(r'-?\s*(.+?):\s+(?:Average duration of\s*)?([\d.]+)\s+min(?:utes?)?\s+on average', s, re.IGNORECASE)
         if not m2:
             m2 = re.match(r'-?\s*(.+?):\s+Average duration of ([\d.]+)\s+min', s, re.IGNORECASE)
+        if not m2:
+            m2 = re.match(r'-?\s*Average duration at (.+?):\s+([\d.]+)\s+min', s, re.IGNORECASE)
+        if not m2:
+            m2 = re.search(r'\bat ([A-Za-z][^(,]+?)\s*\(average ([\d.]+)\s*min', s, re.IGNORECASE)
         if m2:
             dur_map[m2.group(1).strip()] = float(m2.group(2))
 
@@ -500,18 +509,33 @@ def _parse_s3(text):
     return pred, conf, (reasoning[:277] + '…' if len(reasoning) > 280 else reasoning)
 
 
+PROMPT_DESCRIPTIONS = {
+    1: (
+        "Given the agent's mobility trajectory data — including visited locations, visit frequencies, "
+        "stay durations, activity types, and temporal patterns — extract objective factual features "
+        "without any interpretation. Identify: (1) location inventory with visit counts and apparent "
+        "price tier; (2) temporal distribution across time-of-day and weekday/weekend; "
+        "(3) spatial activity radius and movement distances; (4) typical activity sequences and "
+        "transition patterns between location types."
+    ),
+    2: (
+        "Based on the trajectory features extracted in Step 1, perform behavioral abstraction across "
+        "four dimensions: (1) Routine & Schedule — infer work schedule type and daily structure; "
+        "(2) Economic Behavior — assess spending tier from venue choices and mobility costs; "
+        "(3) Social & Lifestyle — identify social engagement patterns, leisure activities, and "
+        "community involvement; (4) Routine Stability — evaluate consistency and regularity of "
+        "movement patterns over the observation period."
+    ),
+    3: (
+        "Synthesizing the factual features (Step 1) and behavioral patterns (Step 2), infer the "
+        "agent's household income bracket. Consider location economic indicators, transportation "
+        "mode signals, activity diversity, and lifestyle markers collectively. Output a single "
+        "income range prediction with supporting reasoning grounded in the observed mobility evidence."
+    ),
+}
+
 def _extract_prompt_instruction(prompt_text, stage):
-    if not prompt_text:
-        return ''
-    key = f'STEP {stage}:'
-    idx = prompt_text.find(key)
-    if idx != -1:
-        return prompt_text[idx:idx + 600].strip()
-    # fallback: first meaningful line
-    for line in prompt_text.strip().splitlines():
-        if line.strip():
-            return line.strip()[:300]
-    return ''
+    return PROMPT_DESCRIPTIONS.get(stage, '')
 
 
 # ── Body renderers ────────────────────────────────────────────────────────────
@@ -605,27 +629,37 @@ def render_chain(s1_text, s2_text, s3_text, status="idle",
     s2_body = _s2_body(s2_text if s2_on else '', s2_on)
     s3_body = _s3_body(s3_text if s3_on else '', s3_on)
 
-    def prompt_pill(pid, prompt_text, stage_num):
+    def paper_block(prompt_text, stage_num, response_text, resp_label):
+        def esc(t): return t.replace('&','&amp;').replace('<','&lt;').replace('>','&gt;')
+        # Prompt snippet
         instr = _extract_prompt_instruction(prompt_text, stage_num)
-        if not instr:
+        p_short = esc(instr[:220]) + ('……' if len(instr) > 220 else '') if instr else ''
+        # Response snippet
+        r_clean = response_text.strip()
+        r_short = esc(r_clean[:260]) + ('……' if len(r_clean) > 260 else '') if r_clean else ''
+
+        prompt_html = (f'<div class="hct-paper-prompt">'
+                       f'<div class="hct-paper-tag">Prompt {stage_num}</div>'
+                       f'<div class="hct-paper-text">{p_short}</div>'
+                       f'</div>') if p_short else ''
+        connector = '<div class="hct-paper-connector"><div class="hct-paper-connector-line"></div></div>' if (p_short and r_short) else ''
+        response_html = (f'<div class="hct-paper-response">'
+                         f'<div class="hct-paper-tag">{resp_label}</div>'
+                         f'<div class="hct-paper-text">{r_short}</div>'
+                         f'</div>') if r_short else ''
+        if not prompt_html and not response_html:
             return ''
-        safe = instr.replace('&', '&amp;').replace('<', '&lt;').replace('>', '&gt;')
-        return (f'<div class="hct-prompt-wrap">'
-                f'<span id="{pid}-btn" class="hct-prompt-toggle" '
-                f'onclick="hctTogglePrompt(\'{pid}\')">'
-                f'▼ show prompt</span>'
-                f'<div id="{pid}" class="hct-prompt-box">{safe}</div>'
-                f'</div>')
+        return f'<div class="hct-paper-wrap">{prompt_html}{connector}{response_html}</div>'
 
-    def stage(cls, num, title, body, on, prompt_text, stage_num):
+    def stage(cls, num, title, body, on, prompt_text, stage_num, response_text='', resp_label='Response'):
         dim_cls = 'active' if on else 'dim'
-        pill = prompt_pill(f'hct-p-{cls}', prompt_text, stage_num) if on and prompt_text else ''
+        paper = paper_block(prompt_text, stage_num, response_text, resp_label) if on and (prompt_text or response_text) else ''
         return (f'<div class="hct-stage hct-{cls} {dim_cls}">'
                 f'<div class="hct-head">'
                 f'<span class="hct-num">{num}</span>'
                 f'<span class="hct-title">{title}</span>'
                 f'</div>'
-                f'{pill}'
+                f'{paper}'
                 f'<div class="hct-body">{body}</div>'
                 f'</div>')
 
@@ -637,11 +671,11 @@ def render_chain(s1_text, s2_text, s3_text, status="idle",
                 f'<div class="hct-arrow-line"></div></div>')
 
     html  = CHAIN_CSS + '<div class="hct-root">'
-    html += stage('s1', 'Stage 01', 'Feature Extraction',    s1_body, s1_on, s1_prompt, 1)
+    html += stage('s1', 'Stage 01', 'Feature Extraction',    s1_body, s1_on, s1_prompt, 1, s1_text, 'Response 1')
     html += arrow('behavioral abstraction', s2_on)
-    html += stage('s2', 'Stage 02', 'Behavioral Analysis',   s2_body, s2_on, s2_prompt, 2)
+    html += stage('s2', 'Stage 02', 'Behavioral Analysis',   s2_body, s2_on, s2_prompt, 2, s2_text, 'Response 2')
     html += arrow('demographic inference', s3_on)
-    html += stage('s3', 'Stage 03', 'Demographic Inference', s3_body, s3_on, s3_prompt, 3)
+    html += stage('s3', 'Stage 03', 'Demographic Inference', s3_body, s3_on, s3_prompt, 3, s3_text, 'Response 3')
     html += '</div>'
     return html