Update app.py with Agent Zero Canvas

app.py

@@ -2,8 +2,7 @@
 # requires-python = ">=3.10"
 # dependencies = [
 #     "marimo",
-#     "gqlalchemy",
-#     "plotly",
+#     "openai",
 # ]
 # ///
 
@@ -16,55 +15,75 @@ app = marimo.App()
 @app.cell
 def _():
     import marimo as mo
-    return mo,
+    import os
+    import json
+    return mo, os, json
 
 
 @app.cell
 def _(mo):
-    mo.md("# 🦀 VerdantClaw-Secure")
-    mo.md("**Status:** ✅ Online")
-    mo.md("---")
-    return
-
+    mo.md(
+        """
+        # ☤ Agent Zero Canvas
 
-@app.cell
-def _(mo):
-    mo.md("### 📊 Memgraph Query")
-    query = mo.ui.text_area(
-        label="Cypher Query:",
-        value="MATCH (n) RETURN n LIMIT 10",
-        full_width=True
+        Connected to **Agent Zero** via Hermes API (OpenAI-compatible).
+        """
     )
-    return query,
-
-
-@app.cell
-def _(mo, query):
-    from gqlalchemy import Memgraph
-    try:
-        memgraph = Memgraph(host='localhost', port=7687)
-        results = list(memgraph.execute(query.value))
-        mo.md(f"**Results:** {len(results)} rows")
-        mo.ui.table(results)
-    except Exception as e:
-        mo.md(f"❌ Memgraph error: {e}")
     return
 
 
 @app.cell
-def _(mo):
-    mo.md("---")
-    mo.md("### 💬 Chat")
-    chat = mo.ui.text_area(label="Message:", full_width=True)
-    return chat,
+def _(mo, os, json):
+    # --- API Key Resolution ---
+    # Priority: AGENT_ZERO_API_KEY > parsed from API_KEYS_JSON > fallback
+    api_key = os.environ.get("AGENT_ZERO_API_KEY", "")
+
+    if not api_key:
+        # Parse API_KEYS_JSON (same logic as verdant_claw entrypoint.sh)
+        keys_json = os.environ.get("API_KEYS_JSON", "")
+        if keys_json:
+            try:
+                keys = json.loads(keys_json)
+                # Use the "agent_zero" key if present, otherwise fall back to "openai"
+                api_key = keys.get("agent_zero", keys.get("openai", ""))
+            except json.JSONDecodeError:
+                pass
+
+    if not api_key:
+        api_key = "hermes-secret-key-123"  # default for local dev
+
+    # --- Base URL ---
+    # This MUST be the public tunnel URL for your Lightning AI Studio port 8642
+    base_url = os.environ.get(
+        "AGENT_ZERO_BASE_URL",
+        "https://8642-01kmke6kkwzc5svsxjvqje6yth.cloudspaces.litng.ai/v1"
+    )
+
+    chat = mo.ui.chat(
+        mo.ai.llm.openai(
+            "hermes-agent",
+            system_message="You are Agent Zero, an advanced AI assistant with access to tools including Hermes Agent for complex reasoning, a terminal, browser, and file system.",
+            api_key=api_key,
+            base_url=base_url,
+        ),
+        prompts=[
+            "What tools do you have available?",
+            "Use Hermes to analyze this problem",
+            "Help me write a Python script",
+        ],
+        show_configuration_controls=True,
+    )
+    chat
+    return base_url, api_key, chat
 
 
 @app.cell
 def _(chat, mo):
-    if chat.value:
-        mo.md(f"**You:** {chat.value}")
+    mo.md(
+        f"**Messages exchanged:** {len(chat.value) if chat.value else 0}"
+    )
     return
 
 
 if __name__ == "__main__":
-    app.run()
+    app.run()