Deploy Karate Wiener (kimodo kata maker)

app.py

@@ -133,6 +133,8 @@ _DEFAULT_SCENE_PATH = Path(__file__).parent / "assets" / "default_scene.webp"
 DIT360_SPACE = os.environ.get("KIMODO_DIT360_SPACE", "Insta360-Research/DiT360").strip()
 KLEIN_SPACE = os.environ.get("KIMODO_KLEIN_SPACE", "polats/tiny-army-klein-zerogpu").strip()
 TRIPOSPLAT_SPACE = os.environ.get("KIMODO_TRIPOSPLAT_SPACE", "polats/kimodo-triposplat-zerogpu").strip()
+# tiny-aya (Cohere Tiny Aya) text model — used to suggest/improve dojo scene prompts.
+TINY_AYA_SPACE = os.environ.get("KIMODO_TINY_AYA_SPACE", "polats/tiny-army-tiny-aya-zerogpu").strip()
 
 # --- Pluggable side tabs ---------------------------------------------------
 # A NEW side tab is added by DROPPING FILES into ./tabs — NO app.py edit needed,
@@ -3571,6 +3573,45 @@ def generate_dojo_scene(payload_json: str | dict | None) -> str:
         return
 
 
+def improve_dojo_prompt(payload_json: str | dict | None) -> str:
+    """Use tiny-aya to turn a few keywords (or nothing) into a full dojo scene prompt.
+    Returns JSON: {"ok": true, "prompt": "..."} or {"ok": false, "error": "..."}."""
+    raw = payload_json
+    if not isinstance(raw, dict):
+        try:
+            raw = json.loads(raw or "{}")
+        except Exception:
+            raw = {"words": str(payload_json or "")}
+    words = (raw.get("words") or raw.get("prompt") or "").strip()
+    system = (
+        "You write a single vivid text-to-image prompt for a 360-degree room used as a "
+        "martial-arts dojo backdrop. Reply with ONLY the prompt on one line: no preamble, "
+        "no quotes, under 55 words. Describe a full room interior (floor, walls, lighting) "
+        "and end with: full room visible, high detail, no people, no text."
+    )
+    user = (f"Turn these ideas into a dojo scene prompt: {words}" if words
+            else "Invent a creative, unexpected dojo scene prompt.")
+    if not TINY_AYA_SPACE:
+        return json.dumps({"ok": False, "error": "tiny-aya space not configured"})
+    try:
+        from gradio_client import Client
+        client = Client(TINY_AYA_SPACE, token=_hf_token())
+        out = client.predict(system, user, 220, 0.9, api_name="/generate")
+        text = (out or "").strip()
+        # Prefer the first substantial line (drops any stray preamble the model adds).
+        best = ""
+        for line in text.splitlines():
+            cand = line.strip().strip('"').strip("'").lstrip("-*•0123456789. ").strip()
+            if len(cand) > len(best):
+                best = cand
+        text = " ".join((best or text).split())[:500]
+        if not text:
+            return json.dumps({"ok": False, "error": "empty suggestion"})
+        return json.dumps({"ok": True, "prompt": text})
+    except Exception as exc:
+        return json.dumps({"ok": False, "error": str(exc)[:180]})
+
+
 def _copy_creator_fields(dst: dict, src: dict | None) -> dict:
     if not src:
         return dst
@@ -6466,6 +6507,10 @@ with gr.Blocks(title="Kimodo ZeroGPU", css=_BLOCKS_CSS) as demo:
     dojo_payload = gr.Textbox(value="", label="Dojo payload", elem_id="dojo-payload", container=False)
     dojo_result = gr.Textbox(value="", label="Dojo result", elem_id="dojo-result", container=False)
     dojo_btn = gr.Button("Generate dojo scene", elem_id="dojo-btn")
+    # tiny-aya prompt helper: JSON {"words": "..."} -> JSON {"ok", "prompt"}.
+    dojo_suggest_payload = gr.Textbox(value="", label="Dojo suggest payload", elem_id="dojo-suggest-payload", container=False)
+    dojo_suggest_result = gr.Textbox(value="", label="Dojo suggest result", elem_id="dojo-suggest-result", container=False)
+    dojo_suggest_btn = gr.Button("Suggest dojo prompt", elem_id="dojo-suggest-btn")
     # Viewer is the only visible component — it fills the screen. Everything else
     # is wrapped in .kimodo-chrome and hidden (kept in the DOM so the drawers can
     # still read its inputs / fire its hidden buttons).
@@ -6608,6 +6653,12 @@ with gr.Blocks(title="Kimodo ZeroGPU", css=_BLOCKS_CSS) as demo:
         outputs=[dojo_result],
         api_name=False,
     )
+    dojo_suggest_btn.click(
+        fn=improve_dojo_prompt,
+        inputs=[dojo_suggest_payload],
+        outputs=[dojo_suggest_result],
+        api_name=False,
+    )
 
     # Display model drives the persistent viewer live via postMessage (no reload).
     # Clothing is handled by the fixed browser drawer so it can behave like kata's drawers.

comic_shots.py

@@ -0,0 +1,79 @@
+"""Shot grammar for the 4-koma stance comics: named shots -> deterministic camera,
+computed purely from the skeleton frame (bbox + hip heading). The LLM never sees any
+of this; it only ever picks captions. Cameras are plain pinhole (eye/target/fov/roll)
+so the same parameters port 1:1 to the three.js viewer later."""
+import numpy as np
+
+PARENTS = [-1, 0, 0, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 9, 9, 12, 13, 14, 16, 17, 18, 19]
+PEL, LHIP, RHIP, HEAD = 0, 1, 2, 15
+LEFTJ = {1, 4, 7, 10, 13, 16, 18, 20}
+RIGHTJ = {2, 5, 8, 11, 14, 17, 19, 21}
+
+# az = horizontal angle of the camera relative to the character's facing direction
+# (0 = head-on). el = elevation in degrees (negative = camera low, looking up).
+# margin = how much air around the joint bbox. roll = dutch tilt.
+SHOTS = {
+    "establishing": dict(az=38.0, el=10.0, margin=1.55, roll=0.0, fov=40.0),
+    "action":       dict(az=55.0, el=4.0,  margin=1.22, roll=-4.0, fov=46.0),
+    "gag":          dict(az=22.0, el=16.0, margin=1.12, roll=13.0, fov=50.0),
+    "hero":         dict(az=32.0, el=-16.0, margin=1.42, roll=0.0, fov=38.0),
+}
+
+
+def heading(Pf):
+    """Facing angle about Y from the hip line (0 = facing +Z), as in _build_stances."""
+    d = Pf[RHIP] - Pf[LHIP]
+    return float(np.arctan2(d[2], -d[0]))
+
+
+def make_camera(shot_name, Pf):
+    """Camera dict for one skeleton frame [22,3]. Frames the joint bbox with the
+    shot's margin, oriented relative to the character's heading."""
+    s = SHOTS[shot_name]
+    lo, hi = Pf.min(axis=0), Pf.max(axis=0)
+    target = (lo + hi) / 2.0
+    radius = float(np.linalg.norm(Pf - target, axis=1).max())
+    fov = np.radians(s["fov"])
+    dist = s["margin"] * radius / np.tan(fov / 2.0)
+    theta = heading(Pf) + np.radians(s["az"])
+    el = np.radians(s["el"])
+    direction = np.array([np.sin(theta) * np.cos(el), np.sin(el), np.cos(theta) * np.cos(el)])
+    eye = target + direction * dist
+    # Don't let the camera dip below the floor (hero shots on grounded poses).
+    eye[1] = max(eye[1], 0.12)
+    return dict(eye=eye, target=target, fov=fov, roll=np.radians(s["roll"]))
+
+
+def project(points, cam, W, H):
+    """Pinhole-project [N,3] world points -> ([N,2] pixel coords, [N] view depth)."""
+    pts = np.asarray(points, np.float64).reshape(-1, 3)
+    eye, target = cam["eye"], cam["target"]
+    fwd = target - eye
+    fwd = fwd / np.linalg.norm(fwd)
+    right = np.cross(fwd, np.array([0.0, 1.0, 0.0]))
+    right = right / (np.linalg.norm(right) + 1e-9)
+    up = np.cross(right, fwd)
+    r = cam["roll"]
+    if abs(r) > 1e-6:  # dutch tilt: rotate the basis about the view axis
+        c, s = np.cos(r), np.sin(r)
+        right, up = c * right + s * up, -s * right + c * up
+    rel = pts - eye
+    x = rel @ right
+    y = rel @ up
+    z = rel @ fwd
+    z = np.maximum(z, 1e-4)
+    f = (H / 2.0) / np.tan(cam["fov"] / 2.0)
+    px = W / 2.0 + f * x / z
+    py = H / 2.0 - f * y / z
+    return np.stack([px, py], axis=1), z
+
+
+def ground_grid(center, span=2.6, step=0.65):
+    """3D segments of a floor grid (y=0) around the character, for scene depth."""
+    segs = []
+    cx, cz = float(center[0]), float(center[2])
+    ticks = np.arange(-span, span + 1e-6, step)
+    for t in ticks:
+        segs.append(((cx - span, 0.0, cz + t), (cx + span, 0.0, cz + t)))
+        segs.append(((cx + t, 0.0, cz - span), (cx + t, 0.0, cz + span)))
+    return segs

tabs/dojo.drawer.html

@@ -1,73 +1,73 @@
-<aside id="kimodo-dojo-drawer" class="kimodo-drawer" aria-label="Dojo scene drawer">
+<aside id="kimodo-dojo-drawer" class="kimodo-drawer" aria-label="Dojo drawer">
   <h2>Dojo</h2>
-  <div class="kimodo-drawer-sub">Create a room scene for the viewer.</div>
 
-  <div class="kimodo-cz-seg kimodo-dojo-mode" id="kimodo-dojo-mode" aria-label="Dojo scene type">
-    <button type="button" class="kimodo-cz-segbtn active" data-mode="splat">Splat</button>
-    <button type="button" class="kimodo-cz-segbtn" data-mode="panorama">Panorama</button>
+  <!-- ============ LIBRARY VIEW: list of dojos (Mine / Community) ============ -->
+  <div id="kimodo-dojo-library" class="kimodo-dojo-view">
+    <div class="kimodo-cz-seg kimodo-cz-seg--sub" id="kimodo-dojo-scope" aria-label="Dojo scope">
+      <button type="button" class="kimodo-cz-segbtn active" data-scope="mine">Mine</button>
+      <button type="button" class="kimodo-cz-segbtn" data-scope="community">Community</button>
+    </div>
+    <button id="kimodo-dojo-add" class="kimodo-dojo-addbtn" type="button">＋ Add dojo</button>
+    <div id="kimodo-dojo-grid" class="kimodo-dojo-cards" aria-label="Dojo list"></div>
+    <div id="kimodo-dojo-empty" class="kimodo-drawer-sub kimodo-dojo-empty"></div>
   </div>
 
-  <div class="kimodo-dojo-settings" aria-label="Dojo room settings">
-    <label class="kimodo-dojo-check">
-      <input id="kimodo-dojo-skybox" type="checkbox" />
-      <span>Use panorama as skybox in splat</span>
-    </label>
-    <label class="kimodo-dojo-range" for="kimodo-dojo-room-size">
-      <span>Room size <output id="kimodo-dojo-room-size-value">8.0</output>m</span>
-      <input id="kimodo-dojo-room-size" type="range" min="3" max="14" value="8" step="0.5" />
-    </label>
-    <label class="kimodo-dojo-range" for="kimodo-dojo-room-height">
-      <span>Room height <output id="kimodo-dojo-room-height-value">3.2</output>m</span>
-      <input id="kimodo-dojo-room-height" type="range" min="2" max="6" value="3.2" step="0.2" />
-    </label>
-  </div>
-
-  <label class="kimodo-dojo-label" for="kimodo-dojo-prompt">Scene prompt</label>
-  <textarea id="kimodo-dojo-prompt" class="kimodo-dojo-prompt" rows="5"
-    spellcheck="true">A traditional martial arts dojo room, polished wooden floor, training mats, wooden wall panels, warm lantern light, full room visible, high detail, no people, no text.</textarea>
+  <!-- ============ CREATE VIEW: prompt + suggest + generate + progress ======= -->
+  <div id="kimodo-dojo-create" class="kimodo-dojo-view" hidden>
+    <button id="kimodo-dojo-back" class="kimodo-dojo-back" type="button" title="Back to dojos">‹ Dojos</button>
+    <div class="kimodo-drawer-sub">Describe a room scene for the viewer.</div>
 
-  <div class="kimodo-dojo-presets" aria-label="Dojo scene presets">
-    <button class="kimodo-dojo-preset" type="button"
-      data-prompt="A traditional martial arts dojo room, polished wooden floor, training mats, wooden wall panels, warm lantern light, full room visible, high detail, no people, no text.">Classic dojo</button>
-    <button class="kimodo-dojo-preset" type="button"
-      data-prompt="An outdoor mountain dojo courtyard at sunrise, stone tiles, wooden gates, misty trees, full environment visible, high detail, no people, no text.">Mountain dojo</button>
-    <button class="kimodo-dojo-preset" type="button"
-      data-prompt="A futuristic neon martial arts training room, glossy floor, holographic wall panels, dramatic rim light, full room visible, no people, no text.">Neon dojo</button>
-  </div>
+    <label class="kimodo-dojo-label" for="kimodo-dojo-prompt">Scene prompt</label>
+    <textarea id="kimodo-dojo-prompt" class="kimodo-dojo-prompt" rows="5"
+      spellcheck="true" placeholder="A few words, or leave blank and tap Suggest…">A traditional martial arts dojo room, polished wooden floor, training mats, wooden wall panels, warm lantern light, full room visible, high detail, no people, no text.</textarea>
+    <button id="kimodo-dojo-improve" class="kimodo-dojo-improve" type="button"
+      title="Improve the prompt with tiny-aya (or suggest one if empty)">✦ Improve with tiny-aya</button>
 
-  <div class="kimodo-dojo-grid">
-    <label>
-      <span>Seed</span>
-      <input id="kimodo-dojo-seed" type="number" value="-1" step="1" />
-    </label>
-    <label>
-      <span>Steps</span>
-      <input id="kimodo-dojo-steps" type="number" value="28" min="1" max="100" step="1" />
-    </label>
-  </div>
+    <details class="kimodo-dojo-adv">
+      <summary>Advanced settings</summary>
+      <div class="kimodo-dojo-adv-body">
+        <div class="kimodo-cz-seg kimodo-dojo-mode" id="kimodo-dojo-mode" aria-label="Dojo scene type">
+          <button type="button" class="kimodo-cz-segbtn active" data-mode="splat">Splat</button>
+          <button type="button" class="kimodo-cz-segbtn" data-mode="panorama">Panorama</button>
+        </div>
+        <div class="kimodo-dojo-settings" aria-label="Dojo room settings">
+          <label class="kimodo-dojo-check">
+            <input id="kimodo-dojo-skybox" type="checkbox" />
+            <span>Use panorama as skybox in splat</span>
+          </label>
+          <label class="kimodo-dojo-range" for="kimodo-dojo-room-size">
+            <span>Room size <output id="kimodo-dojo-room-size-value">8.0</output>m</span>
+            <input id="kimodo-dojo-room-size" type="range" min="3" max="14" value="8" step="0.5" />
+          </label>
+          <label class="kimodo-dojo-range" for="kimodo-dojo-room-height">
+            <span>Room height <output id="kimodo-dojo-room-height-value">3.2</output>m</span>
+            <input id="kimodo-dojo-room-height" type="range" min="2" max="6" value="3.2" step="0.2" />
+          </label>
+        </div>
+        <div class="kimodo-dojo-grid2">
+          <label><span>Seed</span><input id="kimodo-dojo-seed" type="number" value="-1" step="1" /></label>
+          <label><span>Steps</span><input id="kimodo-dojo-steps" type="number" value="28" min="1" max="100" step="1" /></label>
+        </div>
+      </div>
+    </details>
 
-  <div class="kimodo-dojo-actions">
-    <button id="kimodo-dojo-generate" class="kimodo-dojo-primary" type="button">Generate Scene</button>
-    <button id="kimodo-dojo-clear" type="button">Clear</button>
-  </div>
+    <div class="kimodo-dojo-actions">
+      <button id="kimodo-dojo-generate" class="kimodo-dojo-primary" type="button">Generate Scene</button>
+      <button id="kimodo-dojo-clear" type="button">Clear</button>
+    </div>
 
-  <div id="kimodo-dojo-progress" class="kimodo-dojo-progress" aria-label="Dojo generation progress">
-    <div class="kimodo-dojo-step" data-step="flux-iso"><span></span>Flux isometric</div>
-    <div class="kimodo-dojo-step" data-step="flux-pano"><span></span>Flux 360</div>
-    <div class="kimodo-dojo-step" data-step="splat"><span></span>TripoSplat</div>
-    <div class="kimodo-dojo-step" data-step="viewer"><span></span>Viewer load</div>
-  </div>
-  <div id="kimodo-dojo-time" class="kimodo-dojo-time" aria-live="polite"></div>
-  <div id="kimodo-dojo-detail" class="kimodo-dojo-detail" aria-live="polite"></div>
-  <div id="kimodo-dojo-preview-wrap" class="kimodo-dojo-preview-wrap">
-    <figure>
-      <figcaption>Isometric</figcaption>
-      <img id="kimodo-dojo-preview" alt="Flux isometric source result" />
-    </figure>
-    <figure>
-      <figcaption>360</figcaption>
-      <img id="kimodo-dojo-preview-pano" alt="Flux 360 source result" />
-    </figure>
+    <div id="kimodo-dojo-progress" class="kimodo-dojo-progress" aria-label="Dojo generation progress">
+      <div class="kimodo-dojo-step" data-step="flux-iso"><span></span>Flux isometric</div>
+      <div class="kimodo-dojo-step" data-step="flux-pano"><span></span>Flux 360</div>
+      <div class="kimodo-dojo-step" data-step="splat"><span></span>TripoSplat</div>
+      <div class="kimodo-dojo-step" data-step="viewer"><span></span>Viewer load</div>
+    </div>
+    <div id="kimodo-dojo-time" class="kimodo-dojo-time" aria-live="polite"></div>
+    <div id="kimodo-dojo-detail" class="kimodo-dojo-detail" aria-live="polite"></div>
+    <div id="kimodo-dojo-preview-wrap" class="kimodo-dojo-preview-wrap">
+      <figure><figcaption>Isometric</figcaption><img id="kimodo-dojo-preview" alt="Flux isometric result" /></figure>
+      <figure><figcaption>360</figcaption><img id="kimodo-dojo-preview-pano" alt="Flux 360 result" /></figure>
+    </div>
+    <div id="kimodo-dojo-status" class="kimodo-drawer-sub kimodo-dojo-status">Ready.</div>
   </div>
-  <div id="kimodo-dojo-status" class="kimodo-drawer-sub kimodo-dojo-status">Ready.</div>
 </aside>

tabs/dojo.tab.css

@@ -101,19 +101,19 @@
 .kimodo-dojo-actions button:hover:not(:disabled) {
   background: #3f3f48 !important;
 }
-.kimodo-dojo-grid {
+.kimodo-dojo-grid2 {
   display: grid;
   grid-template-columns: 1fr 1fr;
   gap: 8px;
-  margin: 10px 0;
+  margin: 10px 0 2px;
 }
-.kimodo-dojo-grid label {
+.kimodo-dojo-grid2 label {
   display: grid;
   gap: 4px;
   font-size: 11px;
   color: #b8bec6;
 }
-.kimodo-dojo-grid input {
+.kimodo-dojo-grid2 input {
   width: 100%;
   box-sizing: border-box;
   padding: 6px 7px;
@@ -226,3 +226,137 @@
   aspect-ratio: 1 / 1;
   object-fit: cover;
 }
+
+/* ============ Library / create restructure ============ */
+.kimodo-dojo-view[hidden] { display: none; }
+.kimodo-dojo-addbtn {
+  width: 100%;
+  padding: 9px;
+  margin-bottom: 10px;
+  font-size: 12.5px;
+  font-weight: 600;
+  cursor: pointer;
+  border-radius: 8px;
+  background: #4a3320 !important;
+  color: #ffe0ad !important;
+  border: 1px solid #b97736 !important;
+  box-shadow: none !important;
+}
+.kimodo-dojo-addbtn:hover { background: #5a3f27 !important; }
+.kimodo-dojo-cards {
+  display: grid;
+  grid-template-columns: 1fr 1fr;
+  gap: 8px;
+}
+.kimodo-dojo-card {
+  position: relative;
+  cursor: pointer;
+  border-radius: 8px;
+  overflow: hidden;
+  background: #202026;
+  border: 1px solid #34343c;
+}
+.kimodo-dojo-card:hover { border-color: #b97736; }
+.kimodo-dojo-card .thumb {
+  display: block;
+  width: 100%;
+  aspect-ratio: 1 / 1;
+  object-fit: cover;
+  background: #15151a;
+}
+.kimodo-dojo-card .ph {
+  display: flex;
+  align-items: center;
+  justify-content: center;
+  width: 100%;
+  aspect-ratio: 1 / 1;
+  font-size: 26px;
+  color: #6a5a3a;
+  background: #1b1b20;
+}
+.kimodo-dojo-card .nm {
+  padding: 5px 7px;
+  font-size: 10.5px;
+  line-height: 1.25;
+  color: #d7d3c8;
+  white-space: nowrap;
+  overflow: hidden;
+  text-overflow: ellipsis;
+}
+.kimodo-dojo-card .badge {
+  position: absolute;
+  left: 5px;
+  top: 5px;
+  padding: 1px 5px;
+  border-radius: 4px;
+  font-size: 8.5px;
+  font-weight: 700;
+  letter-spacing: .03em;
+  background: rgba(0, 0, 0, 0.6);
+  color: #ffd28a;
+}
+.kimodo-dojo-card .del {
+  position: absolute;
+  top: 4px;
+  right: 5px;
+  padding: 1px 5px;
+  font-size: 12px;
+  line-height: 1;
+  cursor: pointer;
+  color: #ccc !important;
+  background: rgba(0, 0, 0, 0.55) !important;
+  border: 0 !important;
+  border-radius: 5px;
+  box-shadow: none !important;
+}
+.kimodo-dojo-card .del:hover { color: #ff9a9a !important; }
+.kimodo-dojo-empty {
+  margin-top: 12px;
+  text-align: center;
+  color: #87909b;
+}
+.kimodo-dojo-back {
+  padding: 4px 2px;
+  margin-bottom: 4px;
+  font-size: 13px;
+  cursor: pointer;
+  color: #ffd28a !important;
+  background: none !important;
+  border: 0 !important;
+  box-shadow: none !important;
+}
+.kimodo-dojo-improve {
+  width: 100%;
+  margin: 7px 0 4px;
+  padding: 7px;
+  font-size: 11.5px;
+  font-weight: 600;
+  cursor: pointer;
+  border-radius: 7px;
+  background: #2a2440 !important;
+  color: #cbb6ff !important;
+  border: 1px solid #4a3a6a !important;
+  box-shadow: none !important;
+}
+.kimodo-dojo-improve:hover:not(:disabled) { background: #342c52 !important; }
+.kimodo-dojo-improve:disabled { opacity: .55; cursor: default; }
+.kimodo-dojo-adv {
+  margin: 8px 0;
+  border: 1px solid #34343c;
+  border-radius: 7px;
+  background: #1d1d22;
+  overflow: hidden;
+}
+.kimodo-dojo-adv > summary {
+  padding: 8px 10px;
+  font-size: 11.5px;
+  font-weight: 600;
+  color: #b8bec6;
+  cursor: pointer;
+  list-style: none;
+}
+.kimodo-dojo-adv > summary::-webkit-details-marker { display: none; }
+.kimodo-dojo-adv > summary::after { content: " ▾"; color: #7f8792; }
+.kimodo-dojo-adv[open] > summary::after { content: " ▴"; }
+.kimodo-dojo-adv-body { padding: 0 10px 8px; }
+.kimodo-dojo-adv-body .kimodo-dojo-mode { margin-top: 2px; }

tabs/dojo.tab.js

@@ -2,9 +2,29 @@
   let busy = false;
   let dojoMode = 'splat';
   let lastScene = null;
+  let dojoScope = 'mine';
+
+  // Community starters — tapping one opens the create view prefilled, so generating
+  // produces a real flux + splat scene (which then lands under "Mine").
+  const COMMUNITY_SEED = [
+    { name: 'Classic dojo', mode: 'splat', settings: { room_size: 8, room_height: 3.2 },
+      prompt: 'A traditional martial arts dojo room, polished wooden floor, training mats, wooden wall panels, warm lantern light, full room visible, high detail, no people, no text.' },
+    { name: 'Mountain dojo', mode: 'panorama', settings: { room_size: 10, room_height: 4 },
+      prompt: 'An outdoor mountain dojo courtyard at sunrise, stone tiles, wooden gates, misty trees, full environment visible, high detail, no people, no text.' },
+    { name: 'Neon dojo', mode: 'splat', settings: { room_size: 8, room_height: 3.2 },
+      prompt: 'A futuristic neon martial arts training room, glossy floor, holographic wall panels, dramatic rim light, full room visible, no people, no text.' }
+  ];
 
   function dojoEls() {
     return {
+      library: document.getElementById('kimodo-dojo-library'),
+      create: document.getElementById('kimodo-dojo-create'),
+      scope: document.getElementById('kimodo-dojo-scope'),
+      add: document.getElementById('kimodo-dojo-add'),
+      back: document.getElementById('kimodo-dojo-back'),
+      grid: document.getElementById('kimodo-dojo-grid'),
+      emptyMsg: document.getElementById('kimodo-dojo-empty'),
+      improve: document.getElementById('kimodo-dojo-improve'),
       prompt: document.getElementById('kimodo-dojo-prompt'),
       mode: document.getElementById('kimodo-dojo-mode'),
       roomSize: document.getElementById('kimodo-dojo-room-size'),
@@ -91,10 +111,7 @@
     if (els.detail) els.detail.textContent = bestProgressText(stage);
   }
 
-  function hiddenInput(id) {
-    return inputInside(id);
-  }
-
+  function hiddenInput(id) { return inputInside(id); }
   function writeHidden(id, value) {
     const input = hiddenInput(id);
     if (!input) return false;
@@ -103,12 +120,20 @@
     input.dispatchEvent(new Event('change', { bubbles: true }));
     return true;
   }
-
   function readHidden(id) {
     const input = hiddenInput(id);
     return input ? input.value || '' : '';
   }
 
+  function roomSettings() {
+    const els = dojoEls();
+    return {
+      room_size: Math.max(1, Number(els.roomSize && els.roomSize.value) || 8),
+      room_height: Math.max(1, Number(els.roomHeight && els.roomHeight.value) || 3.2),
+      use_panorama_skybox: !!(els.skybox && els.skybox.checked)
+    };
+  }
+
   function sendScene(scene) {
     const frame = previewFrame();
     if (!scene) return;
@@ -119,15 +144,6 @@
     if (frame && frame.contentWindow) frame.contentWindow.postMessage({ kind: 'dojo-scene', scene: payload, image: payload.image }, '*');
   }
 
-  function roomSettings() {
-    const els = dojoEls();
-    return {
-      room_size: Math.max(1, Number(els.roomSize && els.roomSize.value) || 8),
-      room_height: Math.max(1, Number(els.roomHeight && els.roomHeight.value) || 3.2),
-      use_panorama_skybox: !!(els.skybox && els.skybox.checked)
-    };
-  }
-
   function updateRoomLabels() {
     const els = dojoEls();
     const settings = roomSettings();
@@ -172,6 +188,7 @@
     const els = dojoEls();
     if (els.generate) els.generate.disabled = busy;
     if (els.clear) els.clear.disabled = busy;
+    if (els.improve) els.improve.disabled = busy;
   }
 
   function waitForResult(previousValue, onStage) {
@@ -213,144 +230,280 @@
     };
   }
 
+  // ---------- Saved dojos (localStorage) ----------
+  function loadMyDojos() { try { return JSON.parse(localStorage.getItem('kimodoMyDojos') || '[]'); } catch (e) { return []; } }
+  function saveMyDojos(list) {
+    let arr = (list || []).slice(0, 16);
+    // Scene payloads carry data-URL splats (can be MBs); drop the oldest until it fits.
+    while (arr.length) {
+      try { localStorage.setItem('kimodoMyDojos', JSON.stringify(arr)); return true; }
+      catch (e) { arr = arr.slice(0, arr.length - 1); }
+    }
+    try { localStorage.setItem('kimodoMyDojos', '[]'); } catch (e) {}
+    return false;
+  }
+  function dojoName(prompt) {
+    const head = (prompt || '').replace(/[.,].*$/, '').trim().replace(/^(a|an|the)\s+/i, '');
+    const name = head.split(/\s+/).slice(0, 4).join(' ');
+    return (name || 'Dojo').slice(0, 30);
+  }
+  function saveGeneratedDojo(scene, cfg) {
+    const thumb = scene.isometric_image || scene.source_image || scene.image || '';
+    const entry = {
+      id: 'd' + Date.now().toString(36),
+      name: dojoName(cfg.prompt),
+      prompt: cfg.prompt,
+      mode: cfg.mode,
+      settings: { room_size: cfg.room_size, room_height: cfg.room_height, use_panorama_skybox: !!cfg.use_panorama_skybox, seed: cfg.seed, steps: cfg.steps },
+      scene: scene,
+      thumb: thumb,
+      ts: Date.now()
+    };
+    const arr = loadMyDojos();
+    arr.unshift(entry);
+    saveMyDojos(arr);
+    return entry;
+  }
+
+  // ---------- View + library ----------
+  function showView(view) {
+    const els = dojoEls();
+    if (els.library) els.library.hidden = (view !== 'library');
+    if (els.create) els.create.hidden = (view !== 'create');
+    if (view === 'library') renderLibrary();
+  }
+
+  function syncModeButtons() {
+    const els = dojoEls();
+    if (!els.mode) return;
+    els.mode.querySelectorAll('.kimodo-cz-segbtn').forEach((b) => b.classList.toggle('active', b.dataset.mode === dojoMode));
+  }
+
+  function applySettings(settings) {
+    const els = dojoEls();
+    if (!settings) return;
+    if (els.roomSize && settings.room_size) els.roomSize.value = settings.room_size;
+    if (els.roomHeight && settings.room_height) els.roomHeight.value = settings.room_height;
+    if (els.skybox) els.skybox.checked = !!settings.use_panorama_skybox;
+    if (els.seed && settings.seed != null) els.seed.value = settings.seed;
+    if (els.steps && settings.steps != null) els.steps.value = settings.steps;
+    updateRoomLabels();
+  }
+
+  function openCreate(seed) {
+    showView('create');
+    resetProgress();
+    const els = dojoEls();
+    if (seed) {
+      if (els.prompt) { els.prompt.value = seed.prompt || ''; els.prompt.dispatchEvent(new Event('input', { bubbles: true })); }
+      dojoMode = seed.mode || 'splat';
+      syncModeButtons();
+      applySettings(seed.settings);
+      setStatus('Loaded "' + (seed.name || 'starter') + '". Tweak and Generate.');
+    } else {
+      setStatus('Ready.');
+    }
+  }
+
+  function mineCard(entry) {
+    const card = document.createElement('div');
+    card.className = 'kimodo-dojo-card';
+    card.title = entry.name || 'dojo';
+    if (entry.thumb) {
+      const img = document.createElement('img');
+      img.className = 'thumb'; img.src = entry.thumb; img.alt = entry.name || 'dojo';
+      card.appendChild(img);
+    } else {
+      const ph = document.createElement('div'); ph.className = 'ph'; ph.textContent = '🏯'; card.appendChild(ph);
+    }
+    const tag = document.createElement('span'); tag.className = 'badge'; tag.textContent = (entry.mode === 'panorama' ? '360' : 'SPLAT'); card.appendChild(tag);
+    const del = document.createElement('button'); del.className = 'del'; del.type = 'button'; del.textContent = '✕'; del.title = 'Delete';
+    del.onclick = (e) => { e.stopPropagation(); if (!window.confirm('Delete this dojo?')) return; saveMyDojos(loadMyDojos().filter((x) => x.id !== entry.id)); renderLibrary(); };
+    card.appendChild(del);
+    const nm = document.createElement('div'); nm.className = 'nm'; nm.textContent = entry.name || 'Dojo'; card.appendChild(nm);
+    card.onclick = () => applySavedDojo(entry);
+    return card;
+  }
+
+  function communityCard(seed) {
+    const card = document.createElement('div');
+    card.className = 'kimodo-dojo-card';
+    card.title = seed.name;
+    const ph = document.createElement('div'); ph.className = 'ph'; ph.textContent = '🏯'; card.appendChild(ph);
+    const tag = document.createElement('span'); tag.className = 'badge'; tag.textContent = 'STARTER'; card.appendChild(tag);
+    const nm = document.createElement('div'); nm.className = 'nm'; nm.textContent = seed.name; card.appendChild(nm);
+    card.onclick = () => openCreate(seed);
+    return card;
+  }
+
+  function renderLibrary() {
+    const els = dojoEls();
+    if (els.scope) els.scope.querySelectorAll('.kimodo-cz-segbtn').forEach((b) => b.classList.toggle('active', b.dataset.scope === dojoScope));
+    if (!els.grid) return;
+    els.grid.innerHTML = '';
+    if (dojoScope === 'mine') {
+      const arr = loadMyDojos();
+      arr.forEach((e) => els.grid.appendChild(mineCard(e)));
+      if (els.emptyMsg) els.emptyMsg.textContent = arr.length ? '' : 'No dojos yet — tap ＋ Add dojo to create one.';
+    } else {
+      COMMUNITY_SEED.forEach((s) => els.grid.appendChild(communityCard(s)));
+      if (els.emptyMsg) els.emptyMsg.textContent = '';
+    }
+  }
+
+  function applySavedDojo(entry) {
+    const scene = entry.scene;
+    if (!scene) { openCreate(entry); return; }   // recipe-only fallback → re-generate
+    dojoMode = scene.render_mode || (scene.splat || scene.splat_url ? 'splat' : 'panorama');
+    syncModeButtons();
+    applySettings(entry.settings);
+    lastScene = scene;
+    sendScene(scene);
+    setStatus('Applied "' + (entry.name || 'dojo') + '".');
+  }
+
+  // ---------- tiny-aya prompt suggestion ----------
+  function improvePrompt() {
+    const els = dojoEls();
+    const payloadInput = hiddenInput('dojo-suggest-payload');
+    const trigger = document.querySelector('#dojo-suggest-btn button, #dojo-suggest-btn');
+    if (!payloadInput || !trigger) { setStatus('Suggest backend missing.'); return; }
+    const words = (els.prompt && els.prompt.value || '').trim();
+    if (els.improve) { els.improve.disabled = true; els.improve.textContent = '✦ Thinking…'; }
+    setStatus(words ? 'Improving your prompt with tiny-aya…' : 'Suggesting a prompt with tiny-aya…');
+    writeHidden('dojo-suggest-result', '');
+    writeHidden('dojo-suggest-payload', JSON.stringify({ words: words }));
+    const started = Date.now();
+    const timer = window.setInterval(() => {
+      const val = readHidden('dojo-suggest-result');
+      if (val) { window.clearInterval(timer); finishImprove(val); }
+      else if (Date.now() - started > 150000) { window.clearInterval(timer); finishImprove('{"ok":false,"error":"timed out"}'); }
+    }, 300);
+    window.setTimeout(() => trigger.click(), 40);
+
+    function finishImprove(val) {
+      if (els.improve) { els.improve.disabled = busy; els.improve.textContent = '✦ Improve with tiny-aya'; }
+      try {
+        const r = JSON.parse(val);
+        if (r.ok && r.prompt) {
+          if (els.prompt) { els.prompt.value = r.prompt; els.prompt.dispatchEvent(new Event('input', { bubbles: true })); }
+          setStatus('Prompt suggested ✦');
+        } else {
+          setStatus('Suggest failed: ' + (r.error || 'unknown'));
+        }
+      } catch (e) { setStatus('Suggest failed.'); }
+    }
+  }
+
+  // ---------- Generate ----------
+  async function runGenerate() {
+    if (busy) return;
+    const cfg = currentConfig();
+    if (!cfg.prompt) { setStatus('Add a scene prompt first (or tap Improve).'); return; }
+    const payloadInput = hiddenInput('dojo-payload');
+    const trigger = document.querySelector('#dojo-btn button, #dojo-btn');
+    if (!payloadInput || !trigger) { setStatus('Dojo backend controls are missing.'); return; }
+    setBusy(true);
+    resetProgress();
+    setStep('flux-iso', 'active');
+    setStatus('Starting isometric Flux generation...');
+    writeHidden('dojo-result', '');
+    writeHidden('dojo-payload', JSON.stringify(cfg));
+    try {
+      trigger.click();
+      const value = await waitForResult('', (stage) => {
+        setTiming(stage);
+        if (stage.source_image || stage.isometric_image) showSourceImage(stage.source_image || stage.isometric_image, 'isometric');
+        if (stage.panorama_image) showSourceImage(stage.panorama_image, 'panorama');
+        if (stage.stage === 'flux-iso-start' || stage.stage === 'flux-iso-progress') {
+          setStep('flux-iso', 'active');
+          setStatus(stage.status || 'Generating isometric Flux image...');
+        } else if (stage.stage === 'flux-iso-done') {
+          setStep('flux-iso', 'done'); setStep('flux-pano', 'active');
+          showSourceImage(stage.source_image || stage.isometric_image, 'isometric');
+          setStatus(stage.status || 'Isometric ready. Starting 360 Flux...');
+        } else if (stage.stage === 'flux-pano-start' || stage.stage === 'flux-pano-progress') {
+          setStep('flux-iso', 'done'); setStep('flux-pano', 'active');
+          setStatus(stage.status || 'Generating 360 Flux image...');
+        } else if (stage.stage === 'flux-pano-done') {
+          setStep('flux-pano', 'done'); setStep('splat', 'active');
+          showSourceImage(stage.panorama_image, 'panorama');
+          setStatus(stage.status || '360 ready. Starting TripoSplat...');
+        } else if (stage.stage === 'splat-start' || stage.stage === 'splat-progress') {
+          setStep('flux-iso', 'done'); setStep('flux-pano', 'done'); setStep('splat', 'active');
+          setStatus(bestProgressText(stage) || stage.status || 'Generating TripoSplat...');
+        } else if (stage.stage === 'splat-done') {
+          setStep('splat', 'done'); setStep('viewer', 'active');
+          setStatus(stage.status || 'TripoSplat ready. Loading viewer...');
+        } else if (stage.stage === 'fallback') {
+          setStep('flux-iso', 'fail'); setStep('flux-pano', 'fail'); setStep('splat', 'fail');
+          setStatus(stage.warning || stage.error || 'Fallback generated.');
+        }
+      });
+      const scene = JSON.parse(value);
+      if (scene && (scene.image || scene.splat || scene.splat_url)) {
+        Object.assign(scene, roomSettings());
+        lastScene = scene;
+        sendScene(scene);
+        // A real scene (klein-triposplat / dit360) is saved under "Mine".
+        if (scene.source === 'klein-triposplat' || scene.source === 'dit360') {
+          const entry = saveGeneratedDojo(scene, cfg);
+          setStatus('Saved "' + entry.name + '" to My dojos. ' + (scene.info || ''));
+        } else {
+          setStatus((scene.warning || 'Using default scene.') + ' Default scene applied.');
+        }
+      } else {
+        setStatus((scene && scene.error) || 'Scene generation failed.');
+      }
+    } catch (e) {
+      console.error('dojo scene generation failed', e);
+      setStatus('Scene generation failed: ' + e.message);
+    } finally {
+      setBusy(false);
+    }
+  }
+
   function initDojo() {
     const drawer = document.getElementById('kimodo-dojo-drawer');
     const els = dojoEls();
-    if (!drawer || drawer.dataset.ready === '1') return;
+    if (!drawer || drawer.dataset.ready === '1') { renderLibrary(); return; }
     drawer.dataset.ready = '1';
     updateRoomLabels();
+    showView('library');
 
-    if (els.mode) {
-      els.mode.addEventListener('click', (e) => {
-        const button = e.target.closest('.kimodo-cz-segbtn[data-mode]');
-        if (!button || busy) return;
-        dojoMode = button.dataset.mode || 'splat';
-        els.mode.querySelectorAll('.kimodo-cz-segbtn').forEach((b) => b.classList.toggle('active', b === button));
-        setStatus(dojoMode === 'panorama' ? 'Panorama mode selected.' : 'Splat mode selected.');
-        if (lastScene) sendScene(lastScene);
-      });
-    }
+    if (els.scope) els.scope.addEventListener('click', (e) => {
+      const b = e.target.closest('.kimodo-cz-segbtn[data-scope]');
+      if (!b) return;
+      dojoScope = b.dataset.scope || 'mine';
+      renderLibrary();
+    });
+    if (els.add) els.add.addEventListener('click', () => openCreate(null));
+    if (els.back) els.back.addEventListener('click', () => showView('library'));
+    if (els.improve) els.improve.addEventListener('click', () => { if (!busy) improvePrompt(); });
+
+    if (els.mode) els.mode.addEventListener('click', (e) => {
+      const button = e.target.closest('.kimodo-cz-segbtn[data-mode]');
+      if (!button || busy) return;
+      dojoMode = button.dataset.mode || 'splat';
+      syncModeButtons();
+      setStatus(dojoMode === 'panorama' ? 'Panorama mode selected.' : 'Splat mode selected.');
+      if (lastScene) sendScene(lastScene);
+    });
 
     ['roomSize', 'roomHeight', 'skybox'].forEach((key) => {
       const input = els[key];
       if (!input) return;
-      input.addEventListener('input', () => {
-        updateRoomLabels();
-        sendRoomSettings();
-      });
-      input.addEventListener('change', () => {
-        updateRoomLabels();
-        if (lastScene) sendScene(lastScene);
-      });
+      input.addEventListener('input', () => { updateRoomLabels(); sendRoomSettings(); });
+      input.addEventListener('change', () => { updateRoomLabels(); if (lastScene) sendScene(lastScene); });
     });
 
-    drawer.querySelectorAll('.kimodo-dojo-preset').forEach((button) => {
-      button.addEventListener('click', () => {
-        if (els.prompt) {
-          els.prompt.value = button.dataset.prompt || '';
-          els.prompt.dispatchEvent(new Event('input', { bubbles: true }));
-        }
-        setStatus('Preset loaded.');
-      });
+    if (els.generate) els.generate.addEventListener('click', runGenerate);
+    if (els.clear) els.clear.addEventListener('click', () => {
+      const frame = previewFrame();
+      if (frame && frame.contentWindow) frame.contentWindow.postMessage({ kind: 'dojo-clear' }, '*');
+      resetProgress();
+      setStatus('Scene cleared.');
     });
-
-    if (els.generate) {
-      els.generate.addEventListener('click', async () => {
-        if (busy) return;
-        const cfg = currentConfig();
-        if (!cfg.prompt) {
-          setStatus('Add a scene prompt first.');
-          return;
-        }
-        const payloadInput = hiddenInput('dojo-payload');
-        const trigger = document.querySelector('#dojo-btn button, #dojo-btn');
-        if (!payloadInput || !trigger) {
-          setStatus('Dojo backend controls are missing.');
-          return;
-        }
-        setBusy(true);
-        resetProgress();
-        setStep('flux-iso', 'active');
-        setStatus('Starting isometric Flux generation...');
-        writeHidden('dojo-result', '');
-        const previous = '';
-        writeHidden('dojo-payload', JSON.stringify(cfg));
-        try {
-          trigger.click();
-          const value = await waitForResult(previous, (stage) => {
-            setTiming(stage);
-            if (stage.source_image || stage.isometric_image) showSourceImage(stage.source_image || stage.isometric_image, 'isometric');
-            if (stage.panorama_image) showSourceImage(stage.panorama_image, 'panorama');
-            if (stage.stage === 'flux-iso-start') {
-              setStep('flux-iso', 'active');
-              setStatus(stage.status || 'Generating isometric Flux image...');
-            } else if (stage.stage === 'flux-iso-progress') {
-              setStep('flux-iso', 'active');
-              setStatus(stage.status || 'Generating isometric Flux image...');
-            } else if (stage.stage === 'flux-iso-done') {
-              setStep('flux-iso', 'done');
-              setStep('flux-pano', 'active');
-              showSourceImage(stage.source_image || stage.isometric_image, 'isometric');
-              setStatus(stage.status || 'Isometric Flux image ready. Starting 360 Flux image...');
-            } else if (stage.stage === 'flux-pano-start') {
-              setStep('flux-iso', 'done');
-              setStep('flux-pano', 'active');
-              setStatus(stage.status || 'Generating 360 Flux image...');
-            } else if (stage.stage === 'flux-pano-progress') {
-              setStep('flux-pano', 'active');
-              setStatus(stage.status || 'Generating 360 Flux image...');
-            } else if (stage.stage === 'flux-pano-done') {
-              setStep('flux-pano', 'done');
-              setStep('splat', 'active');
-              showSourceImage(stage.panorama_image, 'panorama');
-              setStatus(stage.status || '360 Flux image ready. Starting TripoSplat...');
-            } else if (stage.stage === 'splat-start') {
-              setStep('splat', 'active');
-              setStatus(stage.status || 'Generating TripoSplat...');
-            } else if (stage.stage === 'splat-progress') {
-              setStep('flux-iso', 'done');
-              setStep('flux-pano', 'done');
-              setStep('splat', 'active');
-              setStatus(bestProgressText(stage) || stage.status || 'Generating TripoSplat...');
-            } else if (stage.stage === 'splat-done') {
-              setStep('splat', 'done');
-              setStep('viewer', 'active');
-              setStatus(stage.status || 'TripoSplat ready. Loading viewer...');
-            } else if (stage.stage === 'fallback') {
-              setStep('flux-iso', 'fail');
-              setStep('flux-pano', 'fail');
-              setStep('splat', 'fail');
-              setStatus(stage.warning || stage.error || 'Fallback generated.');
-            }
-          });
-          const scene = JSON.parse(value);
-          if (scene && (scene.image || scene.splat || scene.splat_url)) {
-            Object.assign(scene, roomSettings());
-            lastScene = scene;
-            sendScene(scene);
-            if (scene.source === 'klein-triposplat') setStatus((dojoMode === 'panorama' ? 'Panorama applied. ' : 'Splat applied. ') + (scene.info || ''));
-            else if (scene.source === 'dit360') setStatus(scene.stage === 'fallback' ? 'Panorama fallback applied.' : 'Panorama applied.');
-            else setStatus((scene.warning || 'Using default scene.') + ' Default scene applied.');
-          } else {
-            setStatus((scene && scene.error) || 'Scene generation failed.');
-          }
-        } catch (e) {
-          console.error('dojo scene generation failed', e);
-          setStatus('Scene generation failed: ' + e.message);
-        } finally {
-          setBusy(false);
-        }
-      });
-    }
-
-    if (els.clear) {
-      els.clear.addEventListener('click', () => {
-        const frame = previewFrame();
-        if (frame && frame.contentWindow) frame.contentWindow.postMessage({ kind: 'dojo-clear' }, '*');
-        resetProgress();
-        setStatus('Scene cleared.');
-      });
-    }
   }
 
   window.addEventListener('kimodo-drawer-open', (e) => {