Update app.py

app.py

@@ -1,75 +1,31 @@
 import gradio as gr
-from agentos_core_v4 import AgentCore   # ⬅️ switch to v4
-
-agent = AgentCore()
-
-def run_agent(prompt):
-    return agent.run(prompt)
-
-iface = gr.Interface(
-    fn=run_agent,
-    inputs="text",
-    outputs="text",
-    title="🧠 AgentOS MVP",
-    description="Self-learning agent with vector memory, telemetry, and personality."
-)
-
-iface.launch()
-
-if __name__ == "__main__":
-    iface.launch()
-from identity_core import IdentityCore
-import json
-
-def verify_identity():
-    identity = IdentityCore()
-    with open("agent_identities.json", "r") as f:
-        data = json.load(f)
-    last_id = list(data.keys())[-1]
-    sig = data[last_id]["signature"]
-    result = identity.verify_identity(last_id, sig)
-    return f"Verification result for {last_id}: {result}"
-verify_btn = gr.Interface(
-    fn=lambda: verify_identity(),
-    inputs=[],
-    outputs="text",
-    title="Verify Agent Identity",
-    description="Checks cryptographic DNA of the latest agent instance"
-)
-iface.launch(share=True)
-verify_btn.launch(share=True)
-import gradio as gr
 import json, os
-
-def show_identities():
-    if os.path.exists("agent_identities.json"):
-        with open("agent_identities.json", "r") as f:
-            return json.dumps(json.load(f), indent=2)
-    return "No digital DNA file found yet."
-
-dna_viewer = gr.Interface(
-    fn=show_identities,
-    inputs=[],
-    outputs="text",
-    title="🔐 Digital DNA Viewer",
-    description="View all agent identities and their signatures."
-)
-
-# Combine both UIs into tabs
-app = gr.TabbedInterface([iface, dna_viewer], ["AgentOS Core", "Digital DNA Viewer"])
-app.launch()
-import gradio as gr
 from agentos_core import AgentCore
 from context_graph import ContextGraph
-import json, os
+from identity_core import IdentityCore
+
+# === PATHS ===
+MEMORY_PATH = "data/memory.json"
+IDENTITY_PATH = "agent_identities.json"
+os.makedirs("data", exist_ok=True)
+
+# === LOAD OR INIT MEMORY ===
+if os.path.exists(MEMORY_PATH):
+    with open(MEMORY_PATH, "r") as f:
+        memory = json.load(f)
+else:
+    memory = {}
 
-# --- Instantiate Core Systems ---
-agent = AgentCore()
+# === CORE SYSTEMS ===
+agent = AgentCore(memory=memory)
 graph = ContextGraph()
+identity = IdentityCore()
 
-# --- Define main agent function ---
+# === SAVE MEMORY AFTER EACH RUN ===
 def interact(prompt):
     response = agent.run(prompt)
+    with open(MEMORY_PATH, "w") as f:
+        json.dump(agent.memory, f, indent=2)
     return response
 
 iface = gr.Interface(
@@ -77,10 +33,10 @@ iface = gr.Interface(
     inputs="text",
     outputs="text",
     title="🧠 AgentOS MVP",
-    description="Self-learning agent with memory + telemetry."
+    description="Self-learning agent with memory, telemetry, and identity verification."
 )
 
-# --- Context Query Tab ---
+# === CONTEXT QUERY TAB ===
 def query_context_ui(agent_id, keyword):
     return "\n".join(graph.query_context(agent_id, keyword))
 
@@ -95,10 +51,10 @@ context_view = gr.Interface(
     description="Search what the agent already knows or has linked in memory."
 )
 
-# --- Digital DNA Viewer Tab ---
+# === IDENTITY VIEWER TAB ===
 def show_identities():
-    if os.path.exists("agent_identities.json"):
-        with open("agent_identities.json", "r") as f:
+    if os.path.exists(IDENTITY_PATH):
+        with open(IDENTITY_PATH, "r") as f:
             return json.dumps(json.load(f), indent=2)
     return "No digital DNA file found yet."
 
@@ -107,13 +63,33 @@ dna_viewer = gr.Interface(
     inputs=[],
     outputs="text",
     title="🔐 Digital DNA Viewer",
-    description="View all agent identities and their signatures."
+    description="View all agent identities and their cryptographic signatures."
 )
 
-# --- Combine all interfaces ---
+# === VERIFY IDENTITY TAB ===
+def verify_identity():
+    if not os.path.exists(IDENTITY_PATH):
+        return "No identity file found."
+    with open(IDENTITY_PATH, "r") as f:
+        data = json.load(f)
+    last_id = list(data.keys())[-1]
+    sig = data[last_id]["signature"]
+    result = identity.verify_identity(last_id, sig)
+    return f"Verification result for {last_id}: {result}"
+
+verify_btn = gr.Interface(
+    fn=lambda: verify_identity(),
+    inputs=[],
+    outputs="text",
+    title="🧩 Verify Agent Identity",
+    description="Checks cryptographic DNA of the latest agent instance."
+)
+
+# === COMBINE ALL ===
 app = gr.TabbedInterface(
-    [iface, dna_viewer, context_view],
-    ["AgentOS Core", "Digital DNA Viewer", "Context Query"]
+    [iface, context_view, dna_viewer, verify_btn],
+    ["AgentOS Core", "Context Query", "Digital DNA Viewer", "Verify Identity"]
 )
 
-app.launch()
+if __name__ == "__main__":
+    app.launch(share=True)