Update main.py

main.py

@@ -14,6 +14,7 @@ Capability types:
 """
 
 import asyncio
+import httpx
 import json
 import os
 import sqlite3
@@ -32,10 +33,12 @@ from fastapi.responses import HTMLResponse, JSONResponse, StreamingResponse
 # Config
 # ---------------------------------------------------------------------------
 
-DB_PATH = Path(os.getenv("FORGE_DB", "/tmp/forge.db"))
-PORT = int(os.getenv("PORT", "7860"))
-FORGE_KEY = os.getenv("FORGE_KEY", "")        # optional publish auth key
-BASE_URL = os.getenv("FORGE_BASE_URL", "https://chris4k-agent-forge.hf.space")
+DB_PATH     = Path(os.getenv("FORGE_DB", "/tmp/forge.db"))
+PORT        = int(os.getenv("PORT", "7860"))
+FORGE_KEY   = os.getenv("FORGE_KEY", "")
+BASE_URL    = os.getenv("FORGE_BASE_URL", "https://chris4k-agent-forge.hf.space")
+PULSE_URL   = os.getenv("PULSE_URL",    "https://chris4k-agent-pulse.hf.space")
+PROMPTS_URL = os.getenv("PROMPTS_URL",  "https://chris4k-agent-prompts.hf.space")
 
 VALID_TYPES = {
     "skill", "prompt", "workflow", "knowledge",
@@ -897,6 +900,173 @@ async def api_tags():
     return JSONResponse({"tags": sorted(tags)})
 
 
+# ---------------------------------------------------------------------------
+# Agent Config Handler — Sprint 5
+# ---------------------------------------------------------------------------
+
+AGENT_DEFAULTS = {
+    "heartbeat_seconds": 0,
+    "cost_mode":         "balanced",
+    "max_react_steps":   6,
+    "color":             "#ff6b00",
+    "tags":              [],
+    "enabled":           True,
+}
+
+COST_MODES  = ["cheap", "balanced", "best"]
+AGENT_COLORS = ["#ff6b00","#0ea5e9","#2ed573","#ff9500","#ff6b9d","#8b5cf6","#10b981","#f59e0b"]
+
+async def _push(url: str, path: str, payload: dict) -> dict:
+    try:
+        async with httpx.AsyncClient(timeout=8) as c:
+            r = await c.post(url + path, json=payload)
+            r.raise_for_status()
+            return r.json()
+    except Exception as e:
+        return {"ok": False, "error": str(e)}
+
+async def _patch(url: str, path: str, payload: dict) -> dict:
+    try:
+        async with httpx.AsyncClient(timeout=8) as c:
+            r = await c.patch(url + path, json=payload)
+            r.raise_for_status()
+            return r.json()
+    except Exception as e:
+        return {"ok": False, "error": str(e)}
+
+async def _get(url: str, path: str) -> dict | list | None:
+    try:
+        async with httpx.AsyncClient(timeout=6) as c:
+            r = await c.get(url + path)
+            r.raise_for_status()
+            return r.json()
+    except Exception:
+        return None
+
+
+@app.post("/api/agents/register", status_code=201)
+async def register_agent(request: Request):
+    """
+    One-shot agent registration.
+    Pushes to PULSE (agent config) AND agent-prompts (persona) simultaneously.
+
+    Body:
+      name           str   required — agent identifier (slug)
+      persona        str   required — system prompt / persona description
+      heartbeat_seconds int  0=manual only
+      cost_mode      str   cheap | balanced | best
+      max_react_steps int
+      color          str   hex colour for UI
+      tags           list[str]
+      enabled        bool
+    """
+    body = await request.json()
+    name = (body.get("name") or "").strip().lower().replace(" ", "_")
+    if not name:
+        raise HTTPException(status_code=400, detail="name is required")
+
+    persona = (body.get("persona") or "").strip()
+    if not persona:
+        raise HTTPException(status_code=400, detail="persona is required")
+
+    agent_cfg = {
+        "name":              name,
+        "persona":           persona,
+        "heartbeat_seconds": int(body.get("heartbeat_seconds", 0)),
+        "cost_mode":         body.get("cost_mode", "balanced"),
+        "max_react_steps":   int(body.get("max_react_steps", 6)),
+        "color":             body.get("color", "#ff6b00"),
+        "tags":              body.get("tags", []),
+        "enabled":           bool(body.get("enabled", True)),
+    }
+
+    results = {}
+
+    # 1. Push agent config to PULSE
+    pulse_r = await _push(PULSE_URL, "/api/agents", agent_cfg)
+    results["pulse"] = pulse_r
+
+    # 2. Push persona to agent-prompts
+    prompts_payload = {
+        "agent":       name,
+        "name":        f"{name.upper()} Agent",
+        "system_prompt": persona,
+        "model_pref":  agent_cfg["cost_mode"],
+        "max_steps":   agent_cfg["max_react_steps"],
+        "tools":       [],
+        "config":      {"color": agent_cfg["color"], "tags": agent_cfg["tags"]},
+    }
+    prompts_r = await _push(PROMPTS_URL, "/api/personas", prompts_payload)
+    results["prompts"] = prompts_r
+
+    # 3. Also store as a FORGE config capability (self-registry)
+    cap_id = f"agent_config_{name}"
+    try:
+        conn = get_db()
+        now  = int(time.time())
+        vid  = str(uuid.uuid4())[:8]
+        conn.execute("""
+            INSERT OR REPLACE INTO capabilities
+              (id, name, description, type, payload, author, tags, version,
+               created_at, updated_at, verified, download_count)
+            VALUES (?,?,?,?,?,?,?,?,?,?,0,0)
+        """, (cap_id,
+              f"Agent Config: {name}",
+              f"Registered agent configuration for {name}",
+              "config",
+              json.dumps(agent_cfg),
+              "forge-config-handler",
+              json.dumps(["agent", name, "config"]),
+              vid, now, now))
+        conn.commit()
+        results["forge_cap"] = cap_id
+    except Exception as e:
+        results["forge_cap"] = f"warning: {e}"
+
+    pulse_ok   = isinstance(pulse_r,   dict) and "error" not in pulse_r
+    prompts_ok = isinstance(prompts_r, dict) and "error" not in prompts_r
+    all_ok     = pulse_ok or prompts_ok   # partial success is still useful
+
+    return JSONResponse(
+        status_code=201 if all_ok else 207,
+        content={"ok": all_ok, "agent": name, "results": results}
+    )
+
+
+@app.get("/api/agents")
+async def list_agents():
+    """Proxy PULSE /api/agents list. Falls back to FORGE config store."""
+    live = await _get(PULSE_URL, "/api/agents")
+    if live is not None:
+        return JSONResponse(live if isinstance(live, list) else [live])
+    # Fallback: pull from forge config store
+    conn  = get_db()
+    rows  = conn.execute(
+        "SELECT payload FROM capabilities WHERE type='config' AND id LIKE 'agent_config_%'"
+    ).fetchall()
+    agents = []
+    for row in rows:
+        try: agents.append(json.loads(row["payload"]))
+        except Exception: pass
+    return JSONResponse(agents)
+
+
+@app.delete("/api/agents/{agent_name}")
+async def delete_agent(agent_name: str):
+    """Disable an agent in PULSE."""
+    r = await _push(PULSE_URL, f"/api/agents/{agent_name}/disable", {})
+    return JSONResponse({"ok": True, "pulse": r})
+
+
+@app.post("/api/agents/{agent_name}/trigger")
+async def trigger_agent(agent_name: str, request: Request):
+    """Manually trigger an agent tick via PULSE."""
+    body = await request.json()
+    r    = await _push(PULSE_URL, f"/api/trigger/{agent_name}", body)
+    return JSONResponse({"ok": True, "pulse": r})
+
+
+
 @app.get("/api/health")
 async def health():
     stats = db_stats()
@@ -1134,6 +1304,7 @@ pre{background:#0a0a14;border:1px solid var(--border);border-radius:6px;padding:
       <button class="type-btn" onclick="showTab('browse')">&#128270; Browse</button>
       <button class="type-btn" onclick="showTab('publish')">&#128228; Publish</button>
       <button class="type-btn" onclick="showTab('api')">&#128225; API Docs</button>
+      <button class="type-btn" onclick="showTab('agents')">&#129302; Agents</button>
     </nav>
 
     <div class="content">
@@ -1210,6 +1381,87 @@ pre{background:#0a0a14;border:1px solid var(--border);border-radius:6px;padding:
         <div class="result-msg" id="pubResult"></div>
       </div>
 
+      <!-- AGENTS TAB -->
+      <div id="tab-agents" style="display:none">
+        <h2 style="font-family:'Space Mono',monospace;color:var(--accent);margin-bottom:1.25rem;font-size:1.1rem">&#129302; Agent Config Handler</h2>
+        <p style="font-family:'DM Mono',monospace;font-size:0.78rem;color:var(--muted);margin-bottom:1.5rem">
+          Register or update an agent in PULSE + agent-prompts in one shot.
+        </p>
+
+        <!-- Live agents grid -->
+        <div style="margin-bottom:1.5rem">
+          <div style="display:flex;align-items:center;gap:1rem;margin-bottom:.75rem">
+            <span style="font-family:'DM Mono',monospace;font-size:.7rem;color:var(--muted);text-transform:uppercase;letter-spacing:.15em">Live Agents</span>
+            <button class="btn btn-secondary" id="refreshAgentsBtn" style="padding:.25rem .75rem;font-size:.75rem">&#8635; Refresh</button>
+          </div>
+          <div id="agentGrid" style="display:grid;grid-template-columns:repeat(auto-fill,minmax(220px,1fr));gap:.75rem"></div>
+        </div>
+
+        <!-- Registration form -->
+        <div style="background:var(--surface2);border:1px solid var(--border);border-radius:8px;padding:1.5rem;max-width:720px">
+          <h3 style="font-family:'Space Mono',monospace;font-size:.9rem;color:var(--accent);margin-bottom:1.25rem">&#43; Register / Update Agent</h3>
+
+          <div style="display:grid;grid-template-columns:1fr 1fr;gap:1rem;margin-bottom:1rem">
+            <div>
+              <label style="font-family:'DM Mono',monospace;font-size:.72rem;color:var(--muted);display:block;margin-bottom:.35rem">Agent Name (slug) *</label>
+              <input id="ag-name" class="search-input" placeholder="researcher" style="width:100%">
+            </div>
+            <div>
+              <label style="font-family:'DM Mono',monospace;font-size:.72rem;color:var(--muted);display:block;margin-bottom:.35rem">Cost Mode</label>
+              <select id="ag-cost" style="width:100%;background:var(--surface);border:1px solid var(--border);color:var(--text);padding:.5rem .75rem;border-radius:4px;font-family:'DM Mono',monospace;font-size:.8rem">
+                <option value="cheap">cheap (fast)</option>
+                <option value="balanced" selected>balanced</option>
+                <option value="best">best (slow)</option>
+              </select>
+            </div>
+          </div>
+
+          <div style="display:grid;grid-template-columns:1fr 1fr;gap:1rem;margin-bottom:1rem">
+            <div>
+              <label style="font-family:'DM Mono',monospace;font-size:.72rem;color:var(--muted);display:block;margin-bottom:.35rem">Heartbeat (seconds, 0=manual)</label>
+              <input id="ag-hb" type="number" min="0" value="0" class="search-input" style="width:100%">
+            </div>
+            <div>
+              <label style="font-family:'DM Mono',monospace;font-size:.72rem;color:var(--muted);display:block;margin-bottom:.35rem">Max ReAct Steps</label>
+              <input id="ag-steps" type="number" min="1" max="20" value="6" class="search-input" style="width:100%">
+            </div>
+          </div>
+
+          <div style="display:grid;grid-template-columns:1fr 1fr;gap:1rem;margin-bottom:1rem">
+            <div>
+              <label style="font-family:'DM Mono',monospace;font-size:.72rem;color:var(--muted);display:block;margin-bottom:.35rem">Tags (comma-separated)</label>
+              <input id="ag-tags" class="search-input" placeholder="research,analysis" style="width:100%">
+            </div>
+            <div>
+              <label style="font-family:'DM Mono',monospace;font-size:.72rem;color:var(--muted);display:block;margin-bottom:.35rem">UI Color</label>
+              <div style="display:flex;gap:.4rem;align-items:center;flex-wrap:wrap;margin-top:.2rem">
+                <input type="color" id="ag-color" value="#ff6b00" style="width:36px;height:36px;border:none;background:none;cursor:pointer;padding:0">
+                <span style="font-family:'DM Mono',monospace;font-size:.72rem;color:var(--muted)" id="ag-color-lbl">#ff6b00</span>
+              </div>
+            </div>
+          </div>
+
+          <div style="margin-bottom:1rem">
+            <label style="font-family:'DM Mono',monospace;font-size:.72rem;color:var(--muted);display:block;margin-bottom:.35rem">Persona / System Prompt *</label>
+            <textarea id="ag-persona" class="search-input" rows="6"
+              placeholder="You are a deep research specialist. Your job is to..."
+              style="width:100%;resize:vertical;min-height:120px;font-family:'DM Mono',monospace;font-size:.8rem;line-height:1.5"></textarea>
+          </div>
+
+          <div style="display:flex;align-items:center;gap:1rem;margin-bottom:1rem">
+            <label style="display:flex;align-items:center;gap:.5rem;cursor:pointer;font-family:'DM Mono',monospace;font-size:.78rem;color:var(--muted)">
+              <input type="checkbox" id="ag-enabled" checked style="accent-color:var(--accent)"> Enabled
+            </label>
+          </div>
+
+          <div style="display:flex;gap:.75rem;align-items:center">
+            <button class="btn btn-primary" id="agRegisterBtn">&#128640; Register Agent</button>
+            <button class="btn btn-secondary" id="agClearBtn">Clear</button>
+            <span id="agResult" style="font-family:'DM Mono',monospace;font-size:.78rem"></span>
+          </div>
+        </div>
+      </div>
+
       <!-- API TAB -->
       <div id="tab-api" style="display:none">
         <h2 style="font-family:'Space Mono',monospace;color:var(--accent);margin-bottom:1.25rem;font-size:1.1rem">FORGE API v2</h2>
@@ -1466,9 +1718,10 @@ function showList() {
 }
 
 function showTab(tab) {
-  ['browse','publish','api'].forEach(t => {
+  ['browse','publish','api','agents'].forEach(t => {
     document.getElementById('tab-'+t).style.display = t===tab?'block':'none';
   });
+  if (tab === 'agents') loadAgents();
 }
 
 // Publish
@@ -1552,6 +1805,140 @@ function escHtml(s) {
 
 // Init
 loadStats();
+// ── Agents tab ──────────────────────────────────────────────────────
+
+async function loadAgents() {
+  const grid = document.getElementById('agentGrid');
+  grid.innerHTML = '<span style="font-family:'DM Mono',monospace;font-size:.8rem;color:var(--muted)">Loading&#8230;</span>';
+  try {
+    const r = await fetch('/api/agents');
+    const agents = await r.json();
+    if (!agents || !agents.length) {
+      grid.innerHTML = '<span style="font-family:'DM Mono',monospace;font-size:.8rem;color:var(--muted)">No agents registered yet.</span>';
+      return;
+    }
+    grid.innerHTML = agents.map(a => {
+      const color = a.color || '#ff6b00';
+      const enabled = a.enabled !== false;
+      return `<div style="background:var(--surface2);border:1px solid var(--border);border-left:3px solid ${color};border-radius:6px;padding:.85rem">
+        <div style="display:flex;align-items:center;gap:.5rem;margin-bottom:.4rem">
+          <span style="width:8px;height:8px;border-radius:50%;background:${enabled?'#00ff88':'#ef4444'};flex-shrink:0"></span>
+          <span style="font-family:'Space Mono',monospace;font-size:.82rem;font-weight:700;color:${color}">${esc(a.name||'?')}</span>
+        </div>
+        <div style="font-family:'DM Mono',monospace;font-size:.68rem;color:var(--muted);margin-bottom:.5rem;line-height:1.4">${esc((a.persona||'').slice(0,100))}${(a.persona||'').length>100?'&#8230;':''}</div>
+        <div style="display:flex;gap:.5rem;flex-wrap:wrap">
+          <span style="font-family:'DM Mono',monospace;font-size:.65rem;background:var(--surface);border:1px solid var(--border);padding:.1rem .4rem;border-radius:3px;color:var(--muted)">${esc(a.cost_mode||'balanced')}</span>
+          <span style="font-family:'DM Mono',monospace;font-size:.65rem;background:var(--surface);border:1px solid var(--border);padding:.1rem .4rem;border-radius:3px;color:var(--muted)">steps:${a.max_react_steps||6}</span>
+          ${a.heartbeat_seconds?`<span style="font-family:'DM Mono',monospace;font-size:.65rem;background:var(--surface);border:1px solid var(--border);padding:.1rem .4rem;border-radius:3px;color:var(--muted)">hb:${a.heartbeat_seconds}s</span>`:''}
+        </div>
+        <div style="margin-top:.6rem;display:flex;gap:.4rem">
+          <button onclick="triggerAgent('${esc(a.name||'')}')" style="font-family:'DM Mono',monospace;font-size:.65rem;background:var(--surface);border:1px solid var(--border);color:var(--accent);padding:.2rem .5rem;border-radius:3px;cursor:pointer">&#9654; Trigger</button>
+          <button onclick="prefillAgent(${JSON.stringify(a)})" style="font-family:'DM Mono',monospace;font-size:.65rem;background:var(--surface);border:1px solid var(--border);color:var(--muted);padding:.2rem .5rem;border-radius:3px;cursor:pointer">&#9998; Edit</button>
+        </div>
+      </div>`;
+    }).join('');
+  } catch(e) {
+    grid.innerHTML = '<span style="font-family:'DM Mono',monospace;font-size:.8rem;color:var(--red)">Error loading agents: ' + esc(String(e)) + '</span>';
+  }
+}
+
+async function registerAgent() {
+  const name    = document.getElementById('ag-name').value.trim();
+  const persona = document.getElementById('ag-persona').value.trim();
+  if (!name || !persona) {
+    setAgResult('error', 'Name and persona are required.');
+    return;
+  }
+  const tagsRaw = document.getElementById('ag-tags').value.trim();
+  const tags    = tagsRaw ? tagsRaw.split(',').map(t=>t.trim()).filter(Boolean) : [];
+  const payload = {
+    name:              name,
+    persona:           persona,
+    heartbeat_seconds: parseInt(document.getElementById('ag-hb').value)||0,
+    cost_mode:         document.getElementById('ag-cost').value,
+    max_react_steps:   parseInt(document.getElementById('ag-steps').value)||6,
+    color:             document.getElementById('ag-color').value,
+    tags:              tags,
+    enabled:           document.getElementById('ag-enabled').checked,
+  };
+  const btn = document.getElementById('agRegisterBtn');
+  btn.disabled = true;
+  btn.textContent = 'Registering&#8230;';
+  setAgResult('muted','Pushing to PULSE + agent-prompts&#8230;');
+  try {
+    const r = await fetch('/api/agents/register', {
+      method:'POST', headers:{'Content-Type':'application/json'},
+      body: JSON.stringify(payload)
+    });
+    const d = await r.json();
+    if (d.ok) {
+      setAgResult('green', '&#10003; Agent &ldquo;' + esc(name) + '&rdquo; registered!');
+      clearAgentForm();
+      setTimeout(loadAgents, 800);
+    } else {
+      setAgResult('red', '&#10007; ' + JSON.stringify(d.results).slice(0,200));
+    }
+  } catch(e) {
+    setAgResult('red', 'Network error: ' + esc(String(e)));
+  } finally {
+    btn.disabled = false;
+    btn.textContent = '&#128640; Register Agent';
+  }
+}
+
+async function triggerAgent(name) {
+  try {
+    const r = await fetch('/api/agents/' + encodeURIComponent(name) + '/trigger', {
+      method:'POST', headers:{'Content-Type':'application/json'},
+      body: JSON.stringify({content:'Manual trigger from FORGE UI', trigger_type:'manual'})
+    });
+    const d = await r.json();
+    setAgResult(d.ok?'green':'red', d.ok ? '&#9654; Triggered ' + esc(name) : 'Trigger failed: ' + JSON.stringify(d));
+  } catch(e) { setAgResult('red', String(e)); }
+}
+
+function prefillAgent(a) {
+  document.getElementById('ag-name').value    = a.name || '';
+  document.getElementById('ag-persona').value = a.persona || '';
+  document.getElementById('ag-hb').value      = a.heartbeat_seconds || 0;
+  document.getElementById('ag-cost').value    = a.cost_mode || 'balanced';
+  document.getElementById('ag-steps').value   = a.max_react_steps || 6;
+  document.getElementById('ag-color').value   = a.color || '#ff6b00';
+  document.getElementById('ag-color-lbl').textContent = a.color || '#ff6b00';
+  document.getElementById('ag-tags').value    = (a.tags||[]).join(', ');
+  document.getElementById('ag-enabled').checked = a.enabled !== false;
+  document.getElementById('agRegisterBtn').scrollIntoView({behavior:'smooth',block:'nearest'});
+}
+
+function clearAgentForm() {
+  ['ag-name','ag-persona','ag-tags'].forEach(id => document.getElementById(id).value = '');
+  document.getElementById('ag-hb').value     = '0';
+  document.getElementById('ag-cost').value   = 'balanced';
+  document.getElementById('ag-steps').value  = '6';
+  document.getElementById('ag-color').value  = '#ff6b00';
+  document.getElementById('ag-color-lbl').textContent = '#ff6b00';
+  document.getElementById('ag-enabled').checked = true;
+  setAgResult('','');
+}
+
+function setAgResult(type, msg) {
+  const el = document.getElementById('agResult');
+  const colors = {green:'var(--green)',red:'var(--red)',muted:'var(--muted)',error:'var(--red)','':`var(--text)`};
+  el.style.color = colors[type] || 'var(--text)';
+  el.innerHTML = msg;
+}
+
+// Wire up listeners once DOM ready
+document.addEventListener('DOMContentLoaded', () => {
+  document.getElementById('agRegisterBtn').addEventListener('click', registerAgent);
+  document.getElementById('agClearBtn').addEventListener('click', clearAgentForm);
+  document.getElementById('refreshAgentsBtn').addEventListener('click', loadAgents);
+  document.getElementById('ag-color').addEventListener('input', e => {
+    document.getElementById('ag-color-lbl').textContent = e.target.value;
+  });
+});
+
+
 doSearch();
 </script>
 </body>
@@ -1568,4 +1955,4 @@ async def root():
 # ---------------------------------------------------------------------------
 
 if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=PORT, log_level="info")
+    uvicorn.run(app, host="0.0.0.0", port=PORT, log_level="info")