Update app.py

app.py

@@ -4,6 +4,7 @@ Persistent woodland civilisation simulation powered by Qwen2.5-1.5B.
 Build Small Hackathon 2026 — Thousand Token Wood track.
 Model: Qwen/Qwen2.5-1.5B-Instruct (≤4B → Tiny Titan badge!)
 """
+
 # ═══════════════════════════════════════════════════════════════════
 # 0 ▸ IMPORTS
 # ═══════════════════════════════════════════════════════════════════
@@ -17,13 +18,17 @@ import database
 
 try:
     import spaces
+
     _ZERO_GPU = True
 except ImportError:
+
     class _Stub:
         @staticmethod
         def GPU(fn=None, *, duration=120):
             return fn if callable(fn) else (lambda f: f)
-    spaces = _Stub(); _ZERO_GPU = False  # type: ignore
+
+    spaces = _Stub()
+    _ZERO_GPU = False  # type: ignore
 
 import torch
 from transformers import pipeline as hf_pipeline
@@ -31,30 +36,43 @@ from transformers import pipeline as hf_pipeline
 # ═══════════════════════════════════════════════════════════════════
 # 1 ▸ CONSTANTS
 # ═══════════════════════════════════════════════════════════════════
-CREATURES      = ["fox", "badger", "squirrel", "mole"]
-EVENT_TYPES    = ["trade", "gossip", "feud", "invention", "discovery", "ceremony"]
-CREATURE_EMOJI = {"fox":"🦊","badger":"🦡","squirrel":"🐿️","mole":"🐀"}
+CREATURES = ["fox", "badger", "squirrel", "mole"]
+EVENT_TYPES = ["trade", "gossip", "feud", "invention", "discovery", "ceremony"]
+CREATURE_EMOJI = {"fox": "🦊", "badger": "🦡", "squirrel": "🐿️", "mole": "🐀"}
 
 # Model — 1.5B primary → 3B fallback. BOTH qualify for ≤4B Tiny Titan badge!
-MODEL_PRIMARY  = "Qwen/Qwen2.5-1.5B-Instruct"
+MODEL_PRIMARY = "Qwen/Qwen2.5-1.5B-Instruct"
 MODEL_FALLBACK = "Qwen/Qwen2.5-3B-Instruct"
 
-REL_DELTAS = {"trade":+5,"gossip":-4,"feud":-10,"invention":+7,"discovery":+6,"ceremony":+3}
+REL_DELTAS = {
+    "trade": +5,
+    "gossip": -4,
+    "feud": -10,
+    "invention": +7,
+    "discovery": +6,
+    "ceremony": +3,
+}
 
 REL_TIERS = [
-    (0,  25,  "Sworn Enemies",      "⚔️"),
-    (26, 40,  "Very Suspicious",    "🦔"),
-    (41, 55,  "Wary Acquaintances", "🤝"),
-    (56, 70,  "Friendly",          "🌰"),
-    (71, 85,  "Close Companions",  "🌿"),
-    (86, 100, "Inseparable",       "🍄"),
+    (0, 25, "Sworn Enemies", "⚔️"),
+    (26, 40, "Very Suspicious", "🦔"),
+    (41, 55, "Wary Acquaintances", "🤝"),
+    (56, 70, "Friendly", "🌰"),
+    (71, 85, "Close Companions", "🌿"),
+    (86, 100, "Inseparable", "🍄"),
 ]
 
 WEIRD_OBJECTS = [
-    "half-eaten poem", "suspicious mushroom", "button that looks like the moon",
-    "forgotten birthday", "three secrets", "a fake acorn",
-    "map to somewhere that may not exist", "extremely formal apology note",
-    "small jar of preserved thunder", "second-hand prophecy",
+    "half-eaten poem",
+    "suspicious mushroom",
+    "button that looks like the moon",
+    "forgotten birthday",
+    "three secrets",
+    "a fake acorn",
+    "map to somewhere that may not exist",
+    "extremely formal apology note",
+    "small jar of preserved thunder",
+    "second-hand prophecy",
 ]
 
 LAWS = [
@@ -135,28 +153,50 @@ NARRATOR_PROMPT = (
     "MOLE: One cryptic sentence."
 )
 
-# ═══════════════════════════════════════════════════════════════════
-# 3 ▸ MODEL  (lazy-loaded inside @spaces.GPU context)
-# ═══════════════════════════════════════════════════════════════════
-_pipe = None
+
+import os as _os
+import requests as _req
+from huggingface_hub import InferenceClient as _InfClient
+
+_MODAL_URL = _os.getenv("MODAL_INFERENCE_URL", "").strip()
+_HF_TOKEN = _os.getenv("HF_TOKEN", "").strip()
+
+_USE_MODAL = bool(_MODAL_URL)
+_USE_HF_API = bool(_HF_TOKEN) and not _USE_MODAL
+_USE_LOCAL = not _USE_MODAL and not _USE_HF_API
+
+_hf_client = (
+    _InfClient(provider="hf-inference", api_key=_HF_TOKEN) if _USE_HF_API else None
+)
+_local_pipe = None
 _model_id_used = "none"
 
+_BACKEND = (
+    "Modal"
+    if _USE_MODAL
+    else "HF Inference API [no quota!]"
+    if _USE_HF_API
+    else "Local ZeroGPU [set HF_TOKEN to remove quota limit]"
+)
+print(f"[TinyC] Backend: {_BACKEND}", flush=True)
+
 
-def _load_pipeline() -> None:
-    global _pipe, _model_id_used
-    if _pipe is not None:
+def _load_local_pipeline() -> None:
+    global _local_pipe, _model_id_used
+    if _local_pipe is not None:
         return
     for mid in (MODEL_PRIMARY, MODEL_FALLBACK):
         try:
-            print(f"[TinyC] Loading {mid} …", flush=True)
-            _pipe = hf_pipeline(
-                "text-generation", model=mid,
-                torch_dtype=torch.float16,
+            print(f"[TinyC] Loading {mid} ...", flush=True)
+            _local_pipe = hf_pipeline(
+                "text-generation",
+                model=mid,
+                dtype=torch.float16,
                 device_map="auto",
                 trust_remote_code=True,
             )
             _model_id_used = mid
-            print(f"[TinyC] {mid} ready ✓", flush=True)
+            print(f"[TinyC] {mid} ready", flush=True)
             return
         except Exception as e:
             print(f"[TinyC] {mid} failed: {e}", flush=True)
@@ -164,22 +204,61 @@ def _load_pipeline() -> None:
 
 
 def _generate(system_prompt: str, user_prompt: str, max_new_tokens: int = 160) -> str:
-    assert _pipe is not None
+    messages = [
+        {"role": "system", "content": system_prompt},
+        {"role": "user", "content": user_prompt},
+    ]
+    if _USE_MODAL:
+        try:
+            r = _req.post(
+                _MODAL_URL,
+                json={
+                    "messages": messages,
+                    "max_new_tokens": max_new_tokens,
+                    "temperature": 0.88,
+                    "top_p": 0.92,
+                },
+                timeout=90,
+            )
+            r.raise_for_status()
+            d = r.json()
+            return (d.get("text") or d.get("generated_text") or "").strip()
+        except Exception as e:
+            print(f"[TinyC] Modal error: {e}", flush=True)
+            return ""
+    if _USE_HF_API:
+        try:
+            resp = _hf_client.chat.completions.create(
+                model=MODEL_PRIMARY,
+                messages=messages,
+                max_tokens=max_new_tokens,
+                temperature=0.88,
+                top_p=0.92,
+            )
+            return (resp.choices[0].message.content or "").strip()
+        except Exception as e:
+            print(f"[TinyC] HF API error: {e}", flush=True)
+            return ""
+    assert _local_pipe is not None
     try:
-        out = _pipe(
-            [{"role":"system","content":system_prompt},{"role":"user","content":user_prompt}],
+        out = _local_pipe(
+            messages,
             max_new_tokens=max_new_tokens,
-            temperature=0.88, top_p=0.92, do_sample=True,
+            temperature=0.88,
+            top_p=0.92,
+            do_sample=True,
             return_full_text=False,
         )
         return (out[0]["generated_text"] or "").strip()
     except Exception as e:
-        print(f"[TinyC] _generate error: {e}", flush=True)
+        print(f"[TinyC] local pipe error: {e}", flush=True)
         return ""
 
 
 def call_agent(agent_name: str, context: str) -> str:
-    result = _generate(AGENT_PROMPTS.get(agent_name, AGENT_PROMPTS["fox"]), context, 100)
+    result = _generate(
+        AGENT_PROMPTS.get(agent_name, AGENT_PROMPTS["fox"]), context, 100
+    )
     return result or f"{agent_name.capitalize()} had no comment at this time."
 
 
@@ -206,15 +285,19 @@ def _nudge_ctx(nudge_type, nudge_value, nudge_target, day_number):
 
 def _historical_ctx() -> str:
     nudges = database.get_recent_nudges(3)
-    if not nudges: return ""
-    lines = [f"Day {n['day_number']} {n['nudge_type']}: {n['nudge_value']}" for n in nudges]
+    if not nudges:
+        return ""
+    lines = [
+        f"Day {n['day_number']} {n['nudge_type']}: {n['nudge_value']}" for n in nudges
+    ]
     return "Recent influences: " + "; ".join(lines)
 
 
 def _past_headline_ctx() -> str:
     """Include last 2 headlines so newspaper can reference history."""
     hl = database.get_all_headlines()
-    if len(hl) < 2: return ""
+    if len(hl) < 2:
+        return ""
     recent = hl[:2]
     return "Past headlines: " + " | ".join(f"Day {d}: {h[:40]}" for d, h in recent)
 
@@ -265,41 +348,51 @@ def _parse_newspaper_full(raw: str, day_number: int) -> dict:
 
 
 def _run_simulation_step(nudge_type, nudge_value, nudge_target):
-    """Core simulation — runs within GPU context."""
-    _load_pipeline()
+    """Core simulation — backend-agnostic."""
     day_number = database.get_next_day_number()
-    creatures  = database.get_all_creatures()
+    creatures = database.get_all_creatures()
 
     current_nudge = _nudge_ctx(nudge_type, nudge_value, nudge_target, day_number)
-    historical    = _historical_ctx()
-    past_hl       = _past_headline_ctx()
-    combined_ctx  = " | ".join(filter(None, [current_nudge, historical]))
+    historical = _historical_ctx()
+    past_hl = _past_headline_ctx()
+    combined_ctx = " | ".join(filter(None, [current_nudge, historical]))
 
     # ── Generate 3 events ─────────────────────────────────────────
     event_records: list[dict] = []
     for _ in range(3):
-        actor  = random.choice(CREATURES)
+        actor = random.choice(CREATURES)
         target = random.choice([c for c in CREATURES if c != actor])
-        etype  = random.choice(EVENT_TYPES)
-        rel    = next((c for c in creatures if c["name"]==actor), {}).get(
-                     "relationship_scores", {}).get(target, 50)
+        etype = random.choice(EVENT_TYPES)
+        rel = (
+            next((c for c in creatures if c["name"] == actor), {})
+            .get("relationship_scores", {})
+            .get(target, 50)
+        )
 
         prompts = {
-            "trade":     f"Propose a trade with {target} (relationship {rel}/100). Be specific about what you're offering.",
-            "gossip":    f"Share gossip about {target} (relationship {rel}/100). Make it wonderfully absurd.",
-            "feud":      f"Describe your current feud with {target} (relationship {rel}/100). It must be about something trivial.",
+            "trade": f"Propose a trade with {target} (relationship {rel}/100). Be specific about what you're offering.",
+            "gossip": f"Share gossip about {target} (relationship {rel}/100). Make it wonderfully absurd.",
+            "feud": f"Describe your current feud with {target} (relationship {rel}/100). It must be about something trivial.",
             "invention": f"You've invented something that involves {target} somehow. Describe your invention.",
             "discovery": f"You've discovered something surprising about {target} or near them. What did you find?",
-            "ceremony":  f"You're organising a ceremony and {target} must be involved. Describe it.",
+            "ceremony": f"You're organising a ceremony and {target} must be involved. Describe it.",
         }
         agent_prompt = prompts[etype]
-        if combined_ctx: agent_prompt += f"\n\nWorld context: {combined_ctx}"
+        if combined_ctx:
+            agent_prompt += f"\n\nWorld context: {combined_ctx}"
 
         description = call_agent(actor, agent_prompt)
         if not description or len(description) < 8:
             description = f"{actor.capitalize()} had a {etype} with {target}."
 
-        event_records.append({"actor":actor,"action":etype,"target":target,"description":description})
+        event_records.append(
+            {
+                "actor": actor,
+                "action": etype,
+                "target": target,
+                "description": description,
+            }
+        )
         database.save_event(day_number, actor, etype, target, description)
 
         # ── Update relationships ──────────────────────────────────
@@ -308,11 +401,11 @@ def _run_simulation_step(nudge_type, nudge_value, nudge_target):
         for c in creatures:
             if c["name"] == actor:
                 sc = c["relationship_scores"]
-                sc[target] = max(0, min(100, sc.get(target,50) + delta))
+                sc[target] = max(0, min(100, sc.get(target, 50) + delta))
                 database.update_creature(actor, relationship_scores=sc)
             if c["name"] == target:
                 sc = c["relationship_scores"]
-                sc[actor] = max(0, min(100, sc.get(actor,50) + delta//2))
+                sc[actor] = max(0, min(100, sc.get(actor, 50) + delta // 2))
                 database.update_creature(target, relationship_scores=sc)
         creatures = database.get_all_creatures()
 
@@ -322,8 +415,10 @@ def _run_simulation_step(nudge_type, nudge_value, nudge_target):
         for e in event_records
     )
     extra = ""
-    if combined_ctx: extra += f"\nWorld context: {combined_ctx}"
-    if past_hl:      extra += f"\n{past_hl}"
+    if combined_ctx:
+        extra += f"\nWorld context: {combined_ctx}"
+    if past_hl:
+        extra += f"\n{past_hl}"
 
     raw_paper = _generate(
         NARRATOR_PROMPT,
@@ -345,122 +440,180 @@ def _run_simulation_step(nudge_type, nudge_value, nudge_target):
     parsed = _parse_newspaper_full(raw_paper, day_number)
     classified = random.choice(CLASSIFIEDS)
 
-    full_text = "\n\n".join([
-        parsed["headline"],
-        parsed["article"],
-        f"WEATHER: {parsed['weather']}",
-        "\n".join(f"{c.upper()}: {t}" for c, t in parsed["briefs"].items()),
-        f"CLASSIFIEDS: {classified}",
-    ])
+    full_text = "\n\n".join(
+        [
+            parsed["headline"],
+            parsed["article"],
+            f"WEATHER: {parsed['weather']}",
+            "\n".join(f"{c.upper()}: {t}" for c, t in parsed["briefs"].items()),
+            f"CLASSIFIEDS: {classified}",
+        ]
+    )
 
     database.save_day(day_number, parsed["headline"], full_text)
     return day_number, parsed, classified
 
 
-# ═══════════════════════════════════════════════════════════════════
-# 5 ▸ ZERОГPU WRAPPER
-# ═══════════════════════════════════════════════════════════════════
-@spaces.GPU(duration=120)
-def advance_day(nudge_type=None, nudge_value=None, nudge_target=None):
-    """Public GPU entry point — 1.5B model needs ≤60s on T4."""
-    return _run_simulation_step(nudge_type, nudge_value, nudge_target)
+if _USE_LOCAL:
+
+    @spaces.GPU(duration=60)
+    def advance_day(nudge_type=None, nudge_value=None, nudge_target=None):
+        _load_local_pipeline()
+        return _run_simulation_step(nudge_type, nudge_value, nudge_target)
+else:
+
+    def advance_day(nudge_type=None, nudge_value=None, nudge_target=None):
+        return _run_simulation_step(nudge_type, nudge_value, nudge_target)
 
 
 # ═══════════════════════════════════════════════════════════════════
 # 6 ▸ PIL NEWSPAPER IMAGE  (enhanced)
 # ═══════════════════════════════════════════════════════════════════
-_SERIF_B = ["/usr/share/fonts/truetype/liberation/LiberationSerif-Bold.ttf",
-            "/usr/share/fonts/truetype/dejavu/DejaVuSerif-Bold.ttf"]
-_SERIF_R = ["/usr/share/fonts/truetype/liberation/LiberationSerif-Regular.ttf",
-            "/usr/share/fonts/truetype/dejavu/DejaVuSerif.ttf"]
+_SERIF_B = [
+    "/usr/share/fonts/truetype/liberation/LiberationSerif-Bold.ttf",
+    "/usr/share/fonts/truetype/dejavu/DejaVuSerif-Bold.ttf",
+]
+_SERIF_R = [
+    "/usr/share/fonts/truetype/liberation/LiberationSerif-Regular.ttf",
+    "/usr/share/fonts/truetype/dejavu/DejaVuSerif.ttf",
+]
+
 
 def _tf(paths, sz):
     for p in paths:
-        try: return ImageFont.truetype(p, sz)
-        except: pass
+        try:
+            return ImageFont.truetype(p, sz)
+        except:
+            pass
     return ImageFont.load_default()
 
+
 def render_newspaper_image(parsed: dict, classified: str, day_number: int) -> str:
     W, H = 960, 720
-    PAPER=(245,232,200); INK=(20,8,2); BORDER=(70,40,8); SUBINK=(90,55,25); GREY=(130,100,70)
-
-    img = Image.new("RGB",(W,H),PAPER)
-    d   = ImageDraw.Draw(img)
-
-    f_mast  = _tf(_SERIF_B, 30);  f_hed  = _tf(_SERIF_B, 20)
-    f_body  = _tf(_SERIF_R, 12);  f_sm   = _tf(_SERIF_R, 10)
-    f_sub   = _tf(_SERIF_R, 11);  f_bold = _tf(_SERIF_B, 12)
+    PAPER = (245, 232, 200)
+    INK = (20, 8, 2)
+    BORDER = (70, 40, 8)
+    SUBINK = (90, 55, 25)
+    GREY = (130, 100, 70)
+
+    img = Image.new("RGB", (W, H), PAPER)
+    d = ImageDraw.Draw(img)
+
+    f_mast = _tf(_SERIF_B, 30)
+    f_hed = _tf(_SERIF_B, 20)
+    f_body = _tf(_SERIF_R, 12)
+    f_sm = _tf(_SERIF_R, 10)
+    f_sub = _tf(_SERIF_R, 11)
+    f_bold = _tf(_SERIF_B, 12)
 
     M = 18
-    d.rectangle([M,M,W-M,H-M], outline=BORDER, width=3)
-    d.rectangle([M+6,M+6,W-M-6,H-M-6], outline=BORDER, width=1)
+    d.rectangle([M, M, W - M, H - M], outline=BORDER, width=3)
+    d.rectangle([M + 6, M + 6, W - M - 6, H - M - 6], outline=BORDER, width=1)
 
     y = M + 14
     # Masthead
     mast = "THE TINYWICK HOLLOW GAZETTE"
-    bb = d.textbbox((0,0),mast,font=f_mast); tw=bb[2]-bb[0]
-    d.text(((W-tw)/2, y), mast, fill=INK, font=f_mast); y += bb[3]-bb[1]+3
+    bb = d.textbbox((0, 0), mast, font=f_mast)
+    tw = bb[2] - bb[0]
+    d.text(((W - tw) / 2, y), mast, fill=INK, font=f_mast)
+    y += bb[3] - bb[1] + 3
     # Sub
     sub = f"Est. Day 1  ✦  Day {day_number}  ✦  One Acorn  ✦  Woodland Readers Only"
-    bb=d.textbbox((0,0),sub,font=f_sm); d.text(((W-(bb[2]-bb[0]))/2,y),sub,fill=SUBINK,font=f_sm)
-    y+=bb[3]-bb[1]+5
+    bb = d.textbbox((0, 0), sub, font=f_sm)
+    d.text(((W - (bb[2] - bb[0])) / 2, y), sub, fill=SUBINK, font=f_sm)
+    y += bb[3] - bb[1] + 5
     # Weather strip
-    wx = parsed.get("weather","Overcast.")
+    wx = parsed.get("weather", "Overcast.")
     wx_line = f"☁ WEATHER: {wx}"
-    bb=d.textbbox((0,0),wx_line,font=f_sm); d.text(((W-(bb[2]-bb[0]))/2,y),wx_line,fill=GREY,font=f_sm)
-    y+=bb[3]-bb[1]+4
-    d.line([M+10,y,W-M-10,y],fill=BORDER,width=2)
-    d.line([M+10,y+4,W-M-10,y+4],fill=BORDER,width=1); y+=14
+    bb = d.textbbox((0, 0), wx_line, font=f_sm)
+    d.text(((W - (bb[2] - bb[0])) / 2, y), wx_line, fill=GREY, font=f_sm)
+    y += bb[3] - bb[1] + 4
+    d.line([M + 10, y, W - M - 10, y], fill=BORDER, width=2)
+    d.line([M + 10, y + 4, W - M - 10, y + 4], fill=BORDER, width=1)
+    y += 14
 
     # Headline
-    for line in textwrap.wrap(parsed.get("headline",""),width=50):
-        bb=d.textbbox((0,0),line,font=f_hed); d.text(((W-(bb[2]-bb[0]))/2,y),line,fill=INK,font=f_hed)
-        y+=bb[3]-bb[1]+2
-    y+=4; d.line([M+10,y,W-M-10,y],fill=BORDER,width=1); y+=10
+    for line in textwrap.wrap(parsed.get("headline", ""), width=50):
+        bb = d.textbbox((0, 0), line, font=f_hed)
+        d.text(((W - (bb[2] - bb[0])) / 2, y), line, fill=INK, font=f_hed)
+        y += bb[3] - bb[1] + 2
+    y += 4
+    d.line([M + 10, y, W - M - 10, y], fill=BORDER, width=1)
+    y += 10
 
     # Two-column article + sidebar
-    PAD=M+12; COL_GAP=24; SIDE_W=220
-    main_w = W - 2*PAD - COL_GAP - SIDE_W
-    half_w = (main_w-COL_GAP)//2
-    col1_x=PAD; col2_x=PAD+half_w+COL_GAP; side_x=PAD+main_w+COL_GAP
-    LH=15; MAX_Y=H-M-50; art_y=y
-
-    art_lines = textwrap.wrap(parsed.get("article",""), width=38)
-    mid = max(1,len(art_lines)//2)
-    ly=art_y
+    PAD = M + 12
+    COL_GAP = 24
+    SIDE_W = 220
+    main_w = W - 2 * PAD - COL_GAP - SIDE_W
+    half_w = (main_w - COL_GAP) // 2
+    col1_x = PAD
+    col2_x = PAD + half_w + COL_GAP
+    side_x = PAD + main_w + COL_GAP
+    LH = 15
+    MAX_Y = H - M - 50
+    art_y = y
+
+    art_lines = textwrap.wrap(parsed.get("article", ""), width=38)
+    mid = max(1, len(art_lines) // 2)
+    ly = art_y
     for line in art_lines[:mid]:
-        if ly+LH>MAX_Y: break
-        d.text((col1_x,ly),line,fill=INK,font=f_body); ly+=LH
-    d.line([col1_x+half_w+COL_GAP//2,art_y,col1_x+half_w+COL_GAP//2,min(ly,MAX_Y)],fill=GREY,width=1)
-    ry=art_y
+        if ly + LH > MAX_Y:
+            break
+        d.text((col1_x, ly), line, fill=INK, font=f_body)
+        ly += LH
+    d.line(
+        [
+            col1_x + half_w + COL_GAP // 2,
+            art_y,
+            col1_x + half_w + COL_GAP // 2,
+            min(ly, MAX_Y),
+        ],
+        fill=GREY,
+        width=1,
+    )
+    ry = art_y
     for line in art_lines[mid:]:
-        if ry+LH>MAX_Y: break
-        d.text((col2_x,ry),line,fill=INK,font=f_body); ry+=LH
+        if ry + LH > MAX_Y:
+            break
+        d.text((col2_x, ry), line, fill=INK, font=f_body)
+        ry += LH
 
     # Sidebar: In Brief
-    d.line([side_x-8,art_y,side_x-8,MAX_Y],fill=BORDER,width=1)
-    sy=art_y
-    d.text((side_x,sy),"IN BRIEF",fill=INK,font=f_bold); sy+=16
-    d.line([side_x,sy,W-M-14,sy],fill=GREY,width=1); sy+=6
-    briefs = parsed.get("briefs",{})
+    d.line([side_x - 8, art_y, side_x - 8, MAX_Y], fill=BORDER, width=1)
+    sy = art_y
+    d.text((side_x, sy), "IN BRIEF", fill=INK, font=f_bold)
+    sy += 16
+    d.line([side_x, sy, W - M - 14, sy], fill=GREY, width=1)
+    sy += 6
+    briefs = parsed.get("briefs", {})
     for cname in CREATURES:
-        em=CREATURE_EMOJI.get(cname,"?"); txt=briefs.get(cname,"")
-        header=f"{em} {cname.upper()}"
-        d.text((side_x,sy),header,fill=INK,font=f_bold); sy+=13
-        for bline in textwrap.wrap(txt,width=26):
-            if sy+12>MAX_Y: break
-            d.text((side_x,sy),bline,fill=INK,font=f_sm); sy+=12
-        sy+=4
+        em = CREATURE_EMOJI.get(cname, "?")
+        txt = briefs.get(cname, "")
+        header = f"{em} {cname.upper()}"
+        d.text((side_x, sy), header, fill=INK, font=f_bold)
+        sy += 13
+        for bline in textwrap.wrap(txt, width=26):
+            if sy + 12 > MAX_Y:
+                break
+            d.text((side_x, sy), bline, fill=INK, font=f_sm)
+            sy += 12
+        sy += 4
 
     # Classifieds footer
-    fy=H-M-38; d.line([M+10,fy,W-M-10,fy],fill=BORDER,width=1); fy+=4
-    d.text((M+14,fy),"CLASSIFIEDS",fill=INK,font=f_bold); fy+=14
-    for cl in textwrap.wrap(classified,width=100):
-        if fy+12>H-M-8: break
-        d.text((M+14,fy),cl,fill=INK,font=f_sm); fy+=12
-
-    path=f"/tmp/tinywick_day_{day_number}.png"
-    img.save(path,"PNG")
+    fy = H - M - 38
+    d.line([M + 10, fy, W - M - 10, fy], fill=BORDER, width=1)
+    fy += 4
+    d.text((M + 14, fy), "CLASSIFIEDS", fill=INK, font=f_bold)
+    fy += 14
+    for cl in textwrap.wrap(classified, width=100):
+        if fy + 12 > H - M - 8:
+            break
+        d.text((M + 14, fy), cl, fill=INK, font=f_sm)
+        fy += 12
+
+    path = f"/tmp/tinywick_day_{day_number}.png"
+    img.save(path, "PNG")
     return path
 
 
@@ -485,11 +638,13 @@ def export_agent_traces() -> str:
     headlines = database.get_all_headlines()
     for dn, _ in reversed(headlines):
         day = database.get_day(dn)
-        if day: trace["days"].append(dict(day))
+        if day:
+            trace["days"].append(dict(day))
         evts = database.get_events_for_day(dn)
         trace["events"].extend([{"day": dn, **e} for e in evts])
     path = "/tmp/tiny_civ_traces.json"
-    with open(path,"w") as f: json.dump(trace, f, indent=2, default=str)
+    with open(path, "w") as f:
+        json.dump(trace, f, indent=2, default=str)
     return path
 
 
@@ -659,21 +814,23 @@ KONAMI_JS = """<script>
 # ═══════════════════════════════════════════════════════════════════
 def _rel_tier(score: int) -> tuple[str, str]:
     for lo, hi, label, icon in REL_TIERS:
-        if lo <= score <= hi: return label, icon
+        if lo <= score <= hi:
+            return label, icon
     return "Unknown", "?"
 
+
 def _html_paper(parsed: dict, classified: str, day_num: int) -> str:
-    hed   = (parsed.get("headline","") or "").replace("<","&lt;")
-    art   = (parsed.get("article","")  or "").replace("<","&lt;")
-    wx    = (parsed.get("weather","")  or "").replace("<","&lt;")
-    briefs = parsed.get("briefs",{})
+    hed = (parsed.get("headline", "") or "").replace("<", "&lt;")
+    art = (parsed.get("article", "") or "").replace("<", "&lt;")
+    wx = (parsed.get("weather", "") or "").replace("<", "&lt;")
+    briefs = parsed.get("briefs", {})
     emojis = "  ".join(f"{CREATURE_EMOJI[c]} {c.capitalize()}" for c in CREATURES)
-    classified_esc = classified.replace("<","&lt;")
+    classified_esc = classified.replace("<", "&lt;")
 
     sidebar_items = ""
     for c in CREATURES:
-        em  = CREATURE_EMOJI.get(c,"?")
-        txt = (briefs.get(c,"") or "").replace("<","&lt;")
+        em = CREATURE_EMOJI.get(c, "?")
+        txt = (briefs.get(c, "") or "").replace("<", "&lt;")
         sidebar_items += f"""
 <div class="sidebar-item">
   <div class="sidebar-creature-name">{em} {c.upper()}</div>
@@ -700,21 +857,24 @@ def _html_paper(parsed: dict, classified: str, day_num: int) -> str:
 </div>
 """
 
+
 def _html_placeholder() -> str:
     founding = database.get_day(0)
     if founding:
-        text  = founding["full_newspaper_text"]
-        parts = text.split("\n\n",1)
-        hed   = parts[0]; art = parts[1] if len(parts)>1 else text
+        text = founding["full_newspaper_text"]
+        parts = text.split("\n\n", 1)
+        hed = parts[0]
+        art = parts[1] if len(parts) > 1 else text
         dummy_parsed = {
-            "headline": hed, "article": art,
+            "headline": hed,
+            "article": art,
             "weather": "Portentous, with scattered significance.",
             "briefs": {
                 "fox": "Forged three certificates before breakfast.",
                 "badger": "Insisted on thirteen amendments before lunch.",
                 "squirrel": "Invented a signing machine! It signed the wrong document!",
                 "mole": "Something is already happening underground.",
-            }
+            },
         }
         return _html_paper(dummy_parsed, "NOTICE: Civilisation now in progress.", 0)
     return """<div class="paper-wrap">
@@ -726,31 +886,36 @@ def _html_placeholder() -> str:
   <div class="paper-daybadge">— Day 0 —</div>
 </div>"""
 
+
 def _html_creatures() -> str:
     creatures = database.get_all_creatures()
     cards = ""
     for c in creatures:
-        em   = CREATURE_EMOJI.get(c["name"],"?")
-        inv  = (", ".join(c["inventory"][:3]) + ("…" if len(c["inventory"])>3 else "")) or "nothing"
+        em = CREATURE_EMOJI.get(c["name"], "?")
+        inv = (
+            ", ".join(c["inventory"][:3]) + ("…" if len(c["inventory"]) > 3 else "")
+        ) or "nothing"
         rels = ""
         for other, score in sorted(c["relationship_scores"].items()):
             label, icon = _rel_tier(score)
-            rels += f'<span title="{label} ({score})">{CREATURE_EMOJI.get(other,"?")} {icon}</span> '
+            rels += f'<span title="{label} ({score})">{CREATURE_EMOJI.get(other, "?")} {icon}</span> '
         cards += f"""<div class="creature-card">
-  <span class="creature-name">{em} {c['name'].capitalize()}</span>
+  <span class="creature-name">{em} {c["name"].capitalize()}</span>
   <div class="creature-tier">{rels}</div>
   <div style="font-size:.78em;margin-top:3px;color:#5a3615;"><em>Carries:</em> {inv}</div>
 </div>"""
     return f'<div class="creature-grid">{cards}</div>'
 
+
 def _html_civ_stats() -> str:
     s = database.get_civ_stats()
-    bp  = s["best_pair"];  wp = s["worst_pair"]
+    bp = s["best_pair"]
+    wp = s["worst_pair"]
     dom = s["dominant"].capitalize()
     return f"""<div class="civ-stats">
-  <div class="stat-item"><span class="stat-label">Days</span><span class="stat-value">{s['total_days']}</span></div>
-  <div class="stat-item"><span class="stat-label">Events</span><span class="stat-value">{s['total_events']}</span></div>
-  <div class="stat-item"><span class="stat-label">Nudges</span><span class="stat-value">{s['total_nudges']}</span></div>
+  <div class="stat-item"><span class="stat-label">Days</span><span class="stat-value">{s["total_days"]}</span></div>
+  <div class="stat-item"><span class="stat-label">Events</span><span class="stat-value">{s["total_events"]}</span></div>
+  <div class="stat-item"><span class="stat-label">Nudges</span><span class="stat-value">{s["total_nudges"]}</span></div>
   <div class="stat-item"><span class="stat-label">Strongest bond</span>
     <span class="stat-value">{bp[0].capitalize()} &amp; {bp[1].capitalize()} ({bp[2]})</span></div>
   <div class="stat-item"><span class="stat-label">Bitterest feud</span>
@@ -758,18 +923,25 @@ def _html_civ_stats() -> str:
   <div class="stat-item"><span class="stat-label">Most popular</span><span class="stat-value">{dom}</span></div>
 </div>"""
 
+
 def _archive_choices():
     hl = database.get_all_headlines()
-    return [(f"Day {dn}: {h[:44]}{'…' if len(h)>44 else ''}", dn) for dn,h in hl] or []
+    return [
+        (f"Day {dn}: {h[:44]}{'…' if len(h) > 44 else ''}", dn) for dn, h in hl
+    ] or []
+
+
+def _status(msg):
+    return f'<div class="status-strip">{msg}</div>'
 
-def _status(msg): return f'<div class="status-strip">{msg}</div>'
 
 def _konami_html() -> str:
     import html as _h
+
     details = "".join(
-        f"<details><summary>{CREATURE_EMOJI.get(n,'')}<strong> {n.upper()}</strong></summary>"
+        f"<details><summary>{CREATURE_EMOJI.get(n, '')}<strong> {n.upper()}</strong></summary>"
         f"<pre>{_h.escape(p)}</pre></details>"
-        for n,p in AGENT_PROMPTS.items()
+        for n, p in AGENT_PROMPTS.items()
     )
     details += f"<details><summary>📰 <strong>NARRATOR</strong></summary><pre>{_h.escape(NARRATOR_PROMPT)}</pre></details>"
     return f"""
@@ -794,58 +966,95 @@ def _render_all(day_num, parsed, classified, status_msg):
         _html_civ_stats(),
         gr.update(choices=_archive_choices(), value=None),
         _status(status_msg),
-        day_num, parsed, classified,
+        day_num,
+        parsed,
+        classified,
     )
 
+
 def _safe_advance(nudge_type=None, nudge_value=None, nudge_target=None):
     try:
         day_num, parsed, classified = advance_day(nudge_type, nudge_value, nudge_target)
-        model_tag = f" [{_model_id_used.split('/')[-1]}]" if _model_id_used!="none" else ""
-        return _render_all(day_num, parsed, classified,
-            f"✓ Day {day_num} published to the Gazette.{model_tag}")
+        model_tag = (
+            f" [{_model_id_used.split('/')[-1]}]" if _model_id_used != "none" else ""
+        )
+        return _render_all(
+            day_num,
+            parsed,
+            classified,
+            f"✓ Day {day_num} published to the Gazette.{model_tag}",
+        )
     except Exception:
         tb = traceback.format_exc()
         print(tb)
         err_parsed = {
             "headline": "THE GAZETTE'S PRINTING PRESS HAS JAMMED",
             "article": "Our correspondents report a technical malfunction. The editor is inconsolable. "
-                       "Beatrice Badger suspects sabotage. Reginald Fox denies everything.",
+            "Beatrice Badger suspects sabotage. Reginald Fox denies everything.",
             "weather": "Stormy, with a chance of errors.",
             "briefs": {c: "Unavailable for comment." for c in CREATURES},
         }
-        return _render_all(0, err_parsed, "LOST: One functioning simulation. — The Editor",
-            "✗ Simulation error — the printing press is jammed. Check logs.")
+        return _render_all(
+            0,
+            err_parsed,
+            "LOST: One functioning simulation. — The Editor",
+            "✗ Simulation error — the printing press is jammed. Check logs.",
+        )
+
+
+def handle_advance():
+    return _safe_advance()
+
+
+def handle_rumour(c, r):
+    return _safe_advance("rumour", r, c)
+
+
+def handle_donation(o):
+    return _safe_advance("donation", o)
+
+
+def handle_law(l):
+    return _safe_advance("law", l)
 
-def handle_advance():      return _safe_advance()
-def handle_rumour(c,r):    return _safe_advance("rumour",r,c)
-def handle_donation(o):    return _safe_advance("donation",o)
-def handle_law(l):         return _safe_advance("law",l)
 
 def handle_archive_view(day_num):
     if day_num is None:
-        return '<div class="archive-area"><em>Select a day to read its edition.</em></div>'
+        return (
+            '<div class="archive-area"><em>Select a day to read its edition.</em></div>'
+        )
     day = database.get_day(int(day_num))
-    if not day: return '<div class="archive-area"><em>Edition not found.</em></div>'
-    txt = day["full_newspaper_text"]; parts = txt.split("\n\n",1)
-    hed = parts[0]; art = parts[1] if len(parts)>1 else txt
+    if not day:
+        return '<div class="archive-area"><em>Edition not found.</em></div>'
+    txt = day["full_newspaper_text"]
+    parts = txt.split("\n\n", 1)
+    hed = parts[0]
+    art = parts[1] if len(parts) > 1 else txt
     # Reconstruct parsed from stored text
     parsed = _parse_newspaper_full(txt, day_num)
     classified_match = re.search(r"CLASSIFIEDS:\s*(.+)", txt)
     cl = classified_match.group(1) if classified_match else random.choice(CLASSIFIEDS)
     return f'<div class="archive-area">{_html_paper(parsed, cl, day_num)}</div>'
 
+
 def handle_share(day_num_state, parsed_state, classified_state):
-    if not parsed_state or not parsed_state.get("headline"): return gr.update(visible=False)
+    if not parsed_state or not parsed_state.get("headline"):
+        return gr.update(visible=False)
     try:
-        path = render_newspaper_image(parsed_state, classified_state or "", day_num_state or 0)
+        path = render_newspaper_image(
+            parsed_state, classified_state or "", day_num_state or 0
+        )
         return gr.update(visible=True, value=path)
-    except Exception: return gr.update(visible=False)
+    except Exception:
+        return gr.update(visible=False)
+
 
 def handle_export_traces():
     try:
         path = export_agent_traces()
         return gr.update(visible=True, value=path)
-    except Exception: return gr.update(visible=False)
+    except Exception:
+        return gr.update(visible=False)
 
 
 # =================================================================
@@ -856,108 +1065,169 @@ def handle_export_traces():
 database.init_db()
 _latest = database.get_latest_day()
 if _latest:
-    _INIT_DAY    = _latest["day_number"]
+    _INIT_DAY = _latest["day_number"]
     _INIT_PARSED = _parse_newspaper_full(_latest["full_newspaper_text"], _INIT_DAY)
-    _cl_m        = re.search(r"CLASSIFIEDS:\s*(.+)", _latest["full_newspaper_text"])
-    _INIT_CL     = _cl_m.group(1) if _cl_m else random.choice(CLASSIFIEDS)
+    _cl_m = re.search(r"CLASSIFIEDS:\s*(.+)", _latest["full_newspaper_text"])
+    _INIT_CL = _cl_m.group(1) if _cl_m else random.choice(CLASSIFIEDS)
 else:
-    _INIT_DAY = 0; _INIT_PARSED = {}; _INIT_CL = ""
+    _INIT_DAY = 0
+    _INIT_PARSED = {}
+    _INIT_CL = ""
 
 # -- Gradio version-aware css/theme routing -----------------------
 _GR_MAJOR = int(gr.__version__.split(".")[0])
-if   _GR_MAJOR >= 6: _BKW: dict = {};                              _LKW: dict = {"css": NEWSPAPER_CSS}
-elif _GR_MAJOR >= 5: _BKW = {"css": NEWSPAPER_CSS};              _LKW = {}
-else:                _BKW = {"css": NEWSPAPER_CSS, "theme": gr.themes.Base(primary_hue="orange", neutral_hue="stone")}; _LKW = {}
+if _GR_MAJOR >= 6:
+    _BKW: dict = {}
+    _LKW: dict = {"css": NEWSPAPER_CSS}
+elif _GR_MAJOR >= 5:
+    _BKW = {"css": NEWSPAPER_CSS}
+    _LKW = {}
+else:
+    _BKW = {
+        "css": NEWSPAPER_CSS,
+        "theme": gr.themes.Base(primary_hue="orange", neutral_hue="stone"),
+    }
+    _LKW = {}
 
 # ------------------------------------------------------------------
 with gr.Blocks(title="Tiny Civilization — The Tinywick Hollow Gazette", **_BKW) as demo:
-
     gr.HTML(_konami_html())
 
     gr.HTML(
         '<div style="text-align:center;padding:8px 0 2px;">'
-        '<h1 style="font-family:\'Playfair Display\',Georgia,serif;color:#160800;font-size:1.9em;margin:0 0 2px;">'
-        '🦊 Tiny Civilization 🐀</h1>'
+        "<h1 style=\"font-family:'Playfair Display',Georgia,serif;color:#160800;font-size:1.9em;margin:0 0 2px;\">"
+        "🦊 Tiny Civilization 🐀</h1>"
         '<p style="font-family:Georgia,serif;color:#5a3615;font-style:italic;margin:0;font-size:.87em;">'
-        'A persistent woodland civilisation. One day. One acorn. One absurd headline at a time.'
+        "A persistent woodland civilisation. One day. One acorn. One absurd headline at a time."
         '&nbsp;|&nbsp;<kbd title="Konami Code">↑↑↓↓←→←→BA</kbd> for secrets.'
-        '</p></div>'
+        "</p></div>"
     )
 
-    day_state     = gr.State(_INIT_DAY)
-    parsed_state  = gr.State(_INIT_PARSED)
+    day_state = gr.State(_INIT_DAY)
+    parsed_state = gr.State(_INIT_PARSED)
     classif_state = gr.State(_INIT_CL)
 
-    status_html    = gr.HTML(value=_status(
-        f"Day {_INIT_DAY} in the archive — next: Day {_INIT_DAY + 1}."
-        if _INIT_DAY >= 0 else "No days yet. Press Advance Day to begin."))
+    status_html = gr.HTML(
+        value=_status(
+            f"Day {_INIT_DAY} in the archive — next: Day {_INIT_DAY + 1}."
+            if _INIT_DAY >= 0
+            else "No days yet. Press Advance Day to begin."
+        )
+    )
     civ_stats_html = gr.HTML(value=_html_civ_stats())
 
     with gr.Row(equal_height=False):
-
         with gr.Column(scale=3):
             newspaper_display = gr.HTML(
-                value=(_html_paper(_INIT_PARSED, _INIT_CL, _INIT_DAY)
-                       if _INIT_PARSED else _html_placeholder()))
+                value=(
+                    _html_paper(_INIT_PARSED, _INIT_CL, _INIT_DAY)
+                    if _INIT_PARSED
+                    else _html_placeholder()
+                )
+            )
 
-            with gr.Accordion('📜 Archive — Past Editions', open=False):
+            with gr.Accordion("📜 Archive — Past Editions", open=False):
                 archive_dd = gr.Dropdown(
-                    choices=_archive_choices(), value=None,
-                    label='Select a past day', container=False)
+                    choices=_archive_choices(),
+                    value=None,
+                    label="Select a past day",
+                    container=False,
+                )
                 archive_display = gr.HTML(
-                    value='<div class="archive-area"><em>Select a day to read its edition.</em></div>')
+                    value='<div class="archive-area"><em>Select a day to read its edition.</em></div>'
+                )
 
         with gr.Column(scale=1, min_width=260):
             gr.HTML('<div class="section-title">📰 Editorial Desk</div>')
-            advance_btn = gr.Button('📅 Advance Day  (no nudge)', variant='primary', size='lg')
+            advance_btn = gr.Button(
+                "📅 Advance Day  (no nudge)", variant="primary", size="lg"
+            )
 
             gr.HTML('<hr style="border-color:#8a6030;margin:8px 0;">')
             gr.HTML('<div class="section-title">✉ Nudge the Story</div>')
-            gr.HTML('<p style="font-size:.78em;color:#5a3615;text-align:center;'
-                    'font-style:italic;margin:0 0 6px;">Each nudge advances one day.</p>')
-
-            with gr.Accordion('🗣️ Spread a Rumour', open=False):
-                rumour_creature = gr.Dropdown(choices=CREATURES, value=CREATURES[0],
-                                              label='About which creature?')
-                rumour_type_dd  = gr.Dropdown(choices=RUMOUR_TYPES, value=RUMOUR_TYPES[0],
-                                              label='What rumour?')
-                rumour_btn      = gr.Button('📢 Spread It', variant='secondary')
-
-            with gr.Accordion('🎁 Donate a Weird Object', open=False):
-                donation_dd  = gr.Dropdown(choices=WEIRD_OBJECTS, value=WEIRD_OBJECTS[0],
-                                           label='Which object?')
-                donation_btn = gr.Button('🎁 Donate It', variant='secondary')
+            gr.HTML(
+                '<p style="font-size:.78em;color:#5a3615;text-align:center;'
+                'font-style:italic;margin:0 0 6px;">Each nudge advances one day.</p>'
+            )
 
-            with gr.Accordion('⚖️ Propose a New Law', open=False):
-                law_dd  = gr.Dropdown(choices=LAWS, value=LAWS[0], label='Which law?')
-                law_btn = gr.Button('⚖️ Propose It', variant='secondary')
+            with gr.Accordion("🗣️ Spread a Rumour", open=False):
+                rumour_creature = gr.Dropdown(
+                    choices=CREATURES, value=CREATURES[0], label="About which creature?"
+                )
+                rumour_type_dd = gr.Dropdown(
+                    choices=RUMOUR_TYPES, value=RUMOUR_TYPES[0], label="What rumour?"
+                )
+                rumour_btn = gr.Button("📢 Spread It", variant="secondary")
+
+            with gr.Accordion("🎁 Donate a Weird Object", open=False):
+                donation_dd = gr.Dropdown(
+                    choices=WEIRD_OBJECTS, value=WEIRD_OBJECTS[0], label="Which object?"
+                )
+                donation_btn = gr.Button("🎁 Donate It", variant="secondary")
+
+            with gr.Accordion("⚖️ Propose a New Law", open=False):
+                law_dd = gr.Dropdown(choices=LAWS, value=LAWS[0], label="Which law?")
+                law_btn = gr.Button("⚖️ Propose It", variant="secondary")
 
             gr.HTML('<hr style="border-color:#8a6030;margin:8px 0;">')
-            share_btn    = gr.Button('🖼️ Share as Image', variant='secondary')
-            img_output   = gr.Image(label='Front Page PNG', visible=False, type='filepath')
-            export_btn   = gr.Button('📡 Export Agent Traces (JSON)', variant='secondary')
-            trace_output = gr.File(label='Agent Traces JSON', visible=False)
+            share_btn = gr.Button("🖼️ Share as Image", variant="secondary")
+            img_output = gr.Image(
+                label="Front Page PNG", visible=False, type="filepath"
+            )
+            export_btn = gr.Button("📡 Export Agent Traces (JSON)", variant="secondary")
+            trace_output = gr.File(label="Agent Traces JSON", visible=False)
 
-            gr.HTML('<p style="font-size:.70em;color:#6a4818;text-align:center;'
-                    'margin-top:8px;font-style:italic;">'
-                    'Model: Qwen2.5-1.5B ≤4B 🐜&nbsp;|&nbsp;Local only 🔌&nbsp;|&nbsp;Custom UI 🎨</p>')
+            gr.HTML(
+                '<p style="font-size:.70em;color:#6a4818;text-align:center;'
+                'margin-top:8px;font-style:italic;">'
+                "Model: Qwen2.5-1.5B ≤4B 🐜&nbsp;|&nbsp;Local only 🔌&nbsp;|&nbsp;Custom UI 🎨</p>"
+            )
 
-    gr.HTML('<div class="section-title" style="margin-top:12px;">Woodland Residents</div>')
+    gr.HTML(
+        '<div class="section-title" style="margin-top:12px;">Woodland Residents</div>'
+    )
     creature_display = gr.HTML(value=_html_creatures())
 
-    _OUT = [newspaper_display, creature_display, civ_stats_html,
-            archive_dd, status_html, day_state, parsed_state, classif_state]
-
-    advance_btn.click( fn=handle_advance,   inputs=[],                                outputs=_OUT, api_name=False)
-    rumour_btn.click(  fn=handle_rumour,    inputs=[rumour_creature, rumour_type_dd], outputs=_OUT, api_name=False)
-    donation_btn.click(fn=handle_donation,  inputs=[donation_dd],                    outputs=_OUT, api_name=False)
-    law_btn.click(     fn=handle_law,       inputs=[law_dd],                         outputs=_OUT, api_name=False)
-    archive_dd.change( fn=handle_archive_view, inputs=[archive_dd], outputs=[archive_display], api_name=False)
-    share_btn.click(   fn=handle_share,    inputs=[day_state, parsed_state, classif_state], outputs=[img_output], api_name=False)
-    export_btn.click(  fn=handle_export_traces, inputs=[], outputs=[trace_output], api_name=False)
+    _OUT = [
+        newspaper_display,
+        creature_display,
+        civ_stats_html,
+        archive_dd,
+        status_html,
+        day_state,
+        parsed_state,
+        classif_state,
+    ]
+
+    advance_btn.click(fn=handle_advance, inputs=[], outputs=_OUT, api_name=False)
+    rumour_btn.click(
+        fn=handle_rumour,
+        inputs=[rumour_creature, rumour_type_dd],
+        outputs=_OUT,
+        api_name=False,
+    )
+    donation_btn.click(
+        fn=handle_donation, inputs=[donation_dd], outputs=_OUT, api_name=False
+    )
+    law_btn.click(fn=handle_law, inputs=[law_dd], outputs=_OUT, api_name=False)
+    archive_dd.change(
+        fn=handle_archive_view,
+        inputs=[archive_dd],
+        outputs=[archive_display],
+        api_name=False,
+    )
+    share_btn.click(
+        fn=handle_share,
+        inputs=[day_state, parsed_state, classif_state],
+        outputs=[img_output],
+        api_name=False,
+    )
+    export_btn.click(
+        fn=handle_export_traces, inputs=[], outputs=[trace_output], api_name=False
+    )
 
 # ═══════════════════════════════════════════════════════════════════
 # 13 ▸ ENTRY POINT
 # ═══════════════════════════════════════════════════════════════════
 if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860, **_LKW)
+    demo.launch(server_name="0.0.0.0", server_port=7860, **_LKW)