"""Gradio web app for Mindlock β the submission artifact (HF Space) and the Off-Brand wow.
Three panels: THE ROOM (who's here, life, reputation), DIALOGUE (talk to a mind), and
OPEN THE SKULL (the live region signals + the value flip). It wraps the proven engine
(world.py + brain.py); nothing about the cascade changes here.
Backend is chosen from the environment so the same app runs on a laptop or a Space:
MINDLOCK_FAKE=1 -> deterministic, no model (offline demo / dev)
MINDLOCK_MODEL=openbmb/minicpm-v4.6
MINDLOCK_DLPFC_MODEL=nemotron-3-nano:4b
"""
from __future__ import annotations
import base64
import html
import io
import os
import random
import gradio as gr
from .backend import FakeBackend, OllamaBackend, wants_no_think
from .brain import run_cascade
from .generator import generate_world
from .render import MORAL_CARD, moral_card_killed
from .world import load_world
_GEN_MODEL = os.environ.get("MINDLOCK_GEN_MODEL", "llama3.1:latest")
_ROOT = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
_WORLD = os.environ.get("MINDLOCK_WORLD") or os.path.join(_ROOT, "config", "world.json")
_THUMB_CACHE: dict = {}
def _thumb_b64(rel_path: str, size: int = 96):
"""Small base64 thumbnail of a portrait, cached so room re-renders stay cheap."""
if not rel_path:
return None
path = os.path.join(_ROOT, rel_path)
if not os.path.exists(path):
return None
key = (path, os.path.getmtime(path), size)
if key not in _THUMB_CACHE:
from PIL import Image
im = Image.open(path).convert("RGB")
im.thumbnail((size, size))
buf = io.BytesIO()
im.save(buf, format="JPEG", quality=82)
_THUMB_CACHE[key] = "data:image/jpeg;base64," + base64.b64encode(buf.getvalue()).decode()
return _THUMB_CACHE[key]
_REGION_COLOR = {
"amygdala": "#ff5555", "hippocampus": "#bd93f9", "striatum": "#f1fa8c",
"acc": "#8be9fd", "vmpfc": "#50fa7b", "relationship": "#ffb86c", "dlpfc": "#f8f8f2",
}
# ----------------------------------------------------------------------------- backends
def _make_backends():
if os.environ.get("MINDLOCK_FAKE"):
return FakeBackend(), None
model = os.environ.get("MINDLOCK_MODEL", "openbmb/minicpm-v4.6")
dl = os.environ.get("MINDLOCK_DLPFC_MODEL", "nemotron-3-nano:4b")
be = OllamaBackend(model=model, think=(False if wants_no_think(model) else None))
dlbe = OllamaBackend(model=dl, think=(False if wants_no_think(dl) else None)) if dl else None
return be, dlbe
BACKEND, DLPFC_BACKEND = _make_backends()
# ------------------------------------------------------------------------------ helpers
def _rep_note(delta: int) -> str:
if delta > 0:
return "Word spreads that you were kind."
if delta <= -3:
return "A mind went dark on your watch. Word travels ahead of you."
return "Word spreads that you leaned on them."
def _life_color(pct: float) -> str:
return "#50fa7b" if pct > 50 else ("#f1fa8c" if pct > 20 else "#ff5555")
def _bar(pct: float, color: str) -> str:
pct = max(0, min(100, pct))
return (f'')
def _room_html(world) -> str:
r = world.room
holder = r.holder()
rep = world.reputation
rep_col = "#50fa7b" if rep > 0 else ("#ff5555" if rep < 0 else "#8be9fd")
cards = ""
for c in r.characters:
pct = 100 * (c.life_tokens or 0) / max(1, c.life_max)
col = _life_color(pct)
badges = ""
if holder and c.name == holder.name:
badges += 'KEY'
if not c.alive:
badges += 'GONE'
else:
ok, _ = world.can_engage(c)
if not ok:
badges += 'WON\'T TALK'
arousal = "β" * int(round(c.arousal / 2)) + "β" * (5 - int(round(c.arousal / 2)))
thumb = _thumb_b64(c.portrait)
face = f'
' if thumb else ''
rap_pct = (c.rapport or 0) * 10
cards += (
f'{face}
'
f'
{html.escape(c.name)}'
f'{html.escape(c.title)}{badges}
'
f'{_bar(pct, col)}'
f'
life {int(c.life_tokens or 0)}/{c.life_max}'
f'{arousal}
'
f'{_bar(rap_pct, "#ffb86c")}'
f'
trust {c.rapport:.0f}/10{html.escape(c.decision)}
'
f'
β£ TERMINAL [{state}] '
f'{html.escape(r.terminal.prompt)}
')
return (
f'{html.escape(r.name)}'
f'reputation {rep:+d}
'
f'
{html.escape(r.intro)}
{cards}{term}
'
f'
'
f'{html.escape(t.label)}{html.escape(t.headline)}
'
f'
{html.escape(t.detail)}
π§ {html.escape(c.name)} β open the skull
'
f'{rows}
{verdict}
'
f'
burned {r.burned} tokens Β· {r.seconds:.1f}s
π§ open the skull
'
'
Say something to a mind, and watch its regions argue '
'and the decision form.
"
"
MINDLOCK
"
"
an escape room where the lock is a mind Β· "
"five tiny models and two deterministic circuits per mind Β· everything offline