Create app.py

app.py

@@ -0,0 +1,226 @@
+import time
+import uuid
+import gradio as gr
+
+from core.config import SimConfig
+from core.sim import init_world, tick, svg_render, raycast_pov
+from core.kpi import compute_kpis
+from core.export import export_run
+
+TITLE = "ZEN Orchestrator Sandbox — Business-Grade Agent Orchestra Simulator"
+
+def new_run_id() -> str:
+    # Stable + readable enough for exports
+    return f"{int(time.time())}_{uuid.uuid4().hex[:8]}"
+
+def status_text(w):
+    k = compute_kpis(w)
+    lines = [
+        f"run_id={w.run_id} | seed={w.seed}",
+        f"day={k['day']} tick={k['tick']} outage={k['outage_active']}",
+        f"spend=${k['spend_usd']} | llm_calls={k['llm_calls']} | tool_calls={k['tool_calls']}",
+        f"done={k['done']} overdue={k['overdue']} sla_rate={k['sla_rate']} cost/done=${k['cost_per_done']}",
+    ]
+    if w.ledger.alerts:
+        lines.append(f"ALERT: {w.ledger.alerts[-1]}")
+    return "\n".join(lines)
+
+def ledger_text(w):
+    # CFO-friendly snapshot
+    top_agents = sorted(w.ledger.spend_by_agent.items(), key=lambda kv: -kv[1])[:6]
+    top_models = sorted(w.ledger.spend_by_model.items(), key=lambda kv: -kv[1])[:6]
+    lines = []
+    lines.append("Ledger Snapshot")
+    lines.append(f"Spend USD: {w.ledger.spend_usd:.6f}")
+    lines.append(f"Tokens prompt: {w.ledger.tokens_prompt} | completion: {w.ledger.tokens_completion} | cached: {w.ledger.tokens_cached_prompt}")
+    lines.append(f"Latency total ms: {w.ledger.latency_ms_total} | LLM calls: {w.ledger.llm_calls} | Tool calls: {w.ledger.tool_calls}")
+    lines.append("\nSpend by agent:")
+    for a, v in top_agents:
+        lines.append(f"- {a}: ${v:.6f}")
+    lines.append("\nSpend by model tier:")
+    for m, v in top_models:
+        lines.append(f"- {m}: ${v:.6f}")
+    return "\n".join(lines)
+
+def ui_events_text(w):
+    return "\n".join(w.ui_events[-20:])
+
+def rebuild(w):
+    arena = svg_render(w)
+    pov = raycast_pov(w, w.pov)
+    status = status_text(w)
+    tasks_table = w.tasks.to_compact_table(w.step, limit=14)
+    ledger = ledger_text(w)
+    events = ui_events_text(w)
+    return arena, pov, status, tasks_table, ledger, events
+
+def apply_settings(seed, minutes_per_tick, budget_soft, budget_hard, initial_tasks, new_tasks_per_day, rework_base):
+    cfg = SimConfig(
+        minutes_per_tick=int(minutes_per_tick),
+        budget_usd_soft=float(budget_soft),
+        budget_usd_hard=float(budget_hard),
+        initial_tasks=int(initial_tasks),
+        new_tasks_per_day=int(new_tasks_per_day),
+        rework_probability_base=float(rework_base),
+    )
+    rid = new_run_id()
+    w = init_world(seed=int(seed), run_id=rid, config=cfg)
+    return (*rebuild(w), w)
+
+def run_n_ticks(w, n):
+    n = max(1, int(n))
+    for _ in range(n):
+        if w.done:
+            break
+        tick(w)
+    return (*rebuild(w), w)
+
+def toggle_overlay(w, v):
+    w.overlay = bool(v)
+    return (*rebuild(w), w)
+
+def toggle_auto_camera(w, v):
+    w.auto_camera = bool(v)
+    w.ui_events.append(f"t={w.step}: auto_camera={w.auto_camera}")
+    return (*rebuild(w), w)
+
+def swap_pov(w):
+    names = list(w.agents.keys())
+    i = names.index(w.pov)
+    w.pov = names[(i + 1) % len(names)]
+    w.ui_events.append(f"t={w.step}: POV -> {w.pov}")
+    return (*rebuild(w), w)
+
+def export_clicked(w):
+    data, fname = export_run(w)
+    return gr.File(value=(fname, data))
+
+with gr.Blocks(title=TITLE) as demo:
+    gr.Markdown(
+        f"## {TITLE}\n"
+        "This is the **business spine**: event-sourcing + ledger accounting + time dilation + task SLA model.\n"
+        "Run weeks/months/years by changing **minutes per tick** and using **Run N ticks** or Autoplay."
+    )
+
+    w_state = gr.State(None)
+    autoplay_on = gr.State(False)
+
+    with gr.Row():
+        arena = gr.HTML(label="Arena (Animated SVG)")
+        with gr.Column():
+            pov_img = gr.Image(label="Agent POV (Body-cam Mode)", type="numpy", width=560, height=315)
+            status = gr.Textbox(label="Status + KPIs", lines=5)
+            tasks_box = gr.Textbox(label="Task Queue (top slice)", lines=12)
+
+    with gr.Row():
+        ledger_box = gr.Textbox(label="Ledger (cost/tokens/latency)", lines=12)
+        events_box = gr.Textbox(label="Run Log (UI)", lines=12)
+
+    with gr.Accordion("Scenario Controls (Business)", open=True):
+        with gr.Row():
+            seed = gr.Number(value=1337, precision=0, label="Seed (determinism)")
+            minutes_per_tick = gr.Slider(5, 240, value=30, step=5, label="Minutes per tick (time dilation)")
+        with gr.Row():
+            budget_soft = gr.Number(value=5.00, precision=2, label="Budget soft cap (USD)")
+            budget_hard = gr.Number(value=10.00, precision=2, label="Budget hard cap (USD)")
+        with gr.Row():
+            initial_tasks = gr.Slider(0, 200, value=24, step=1, label="Initial backlog size")
+            new_tasks_per_day = gr.Slider(0, 50, value=6, step=1, label="New tasks per simulated day")
+            rework_base = gr.Slider(0.0, 0.5, value=0.08, step=0.01, label="Base rework probability")
+
+        with gr.Row():
+            btn_apply = gr.Button("Apply / New Run")
+            btn_run = gr.Button("Run N ticks")
+            run_n = gr.Number(value=120, precision=0, label="N")
+
+    with gr.Accordion("Playback + Camera", open=True):
+        with gr.Row():
+            overlay = gr.Checkbox(value=True, label="POV Reticle Overlay")
+            auto_camera = gr.Checkbox(value=True, label="Auto Camera Cuts")
+            btn_swap_pov = gr.Button("Swap POV Agent")
+        with gr.Row():
+            autoplay_speed = gr.Slider(0.05, 0.8, value=0.18, step=0.01, label="Autoplay tick interval (sec)")
+            btn_play = gr.Button("▶ Start Autoplay")
+            btn_pause = gr.Button("⏸ Stop Autoplay")
+
+    with gr.Accordion("Exports", open=True):
+        btn_export = gr.Button("Export run (events.jsonl + metrics.csv + summary.json)")
+        export_file = gr.File(label="Download export ZIP")
+
+    timer = gr.Timer(value=0.18, active=False)
+
+    # Init on load
+    def on_load():
+        cfg = SimConfig()
+        w = init_world(seed=1337, run_id=new_run_id(), config=cfg)
+        return (*rebuild(w), w)
+
+    demo.load(on_load, inputs=[], outputs=[arena, pov_img, status, tasks_box, ledger_box, events_box, w_state], queue=True)
+
+    btn_apply.click(
+        apply_settings,
+        inputs=[seed, minutes_per_tick, budget_soft, budget_hard, initial_tasks, new_tasks_per_day, rework_base],
+        outputs=[arena, pov_img, status, tasks_box, ledger_box, events_box, w_state],
+        queue=True,
+    )
+
+    btn_run.click(
+        run_n_ticks,
+        inputs=[w_state, run_n],
+        outputs=[arena, pov_img, status, tasks_box, ledger_box, events_box, w_state],
+        queue=True,
+    )
+
+    overlay.change(
+        toggle_overlay,
+        inputs=[w_state, overlay],
+        outputs=[arena, pov_img, status, tasks_box, ledger_box, events_box, w_state],
+        queue=True,
+    )
+
+    auto_camera.change(
+        toggle_auto_camera,
+        inputs=[w_state, auto_camera],
+        outputs=[arena, pov_img, status, tasks_box, ledger_box, events_box, w_state],
+        queue=True,
+    )
+
+    btn_swap_pov.click(
+        swap_pov,
+        inputs=[w_state],
+        outputs=[arena, pov_img, status, tasks_box, ledger_box, events_box, w_state],
+        queue=True,
+    )
+
+    # Autoplay
+    def autoplay_start(interval: float):
+        return gr.update(value=float(interval), active=True), True
+
+    def autoplay_stop():
+        return gr.update(active=False), False
+
+    btn_play.click(autoplay_start, inputs=[autoplay_speed], outputs=[timer, autoplay_on], queue=True)
+    btn_pause.click(autoplay_stop, inputs=[], outputs=[timer, autoplay_on], queue=True)
+
+    def autoplay_tick(w, is_on: bool):
+        if not is_on:
+            return (*rebuild(w), w, is_on, gr.update())
+        if not w.done:
+            tick(w)
+        return (*rebuild(w), w, is_on, gr.update())
+
+    timer.tick(
+        autoplay_tick,
+        inputs=[w_state, autoplay_on],
+        outputs=[arena, pov_img, status, tasks_box, ledger_box, events_box, w_state, autoplay_on, timer],
+        queue=True,
+    )
+
+    btn_export.click(
+        export_clicked,
+        inputs=[w_state],
+        outputs=[export_file],
+        queue=True,
+    )
+
+demo.queue().launch(ssr_mode=False)