Update app.py

app.py

@@ -1,96 +1,21 @@
 import os
-os.environ["TRANSFORMERS_CACHE"] = "/data/cache"import gradio as gr
-import json, os
-from agentos_core import AgentCore
-from context_graph import ContextGraph
-from identity_core import IdentityCore
+os.environ["TRANSFORMERS_CACHE"] = "/data/cache"
 
-# === PATHS ===
-MEMORY_PATH = "data/memory.json"
-IDENTITY_PATH = "agent_identities.json"
-os.makedirs("data", exist_ok=True)
+import gradio as gr
+from agentos_core_v4 import AgentCore   # or agentos_core if you're not on v4
 
-# === LOAD OR INIT MEMORY ===
-if os.path.exists(MEMORY_PATH):
-    with open(MEMORY_PATH, "r") as f:
-        memory = json.load(f)
-else:
-    memory = {}
+agent = AgentCore()
 
-# === CORE SYSTEMS ===
-agent = AgentCore(memory=memory)
-graph = ContextGraph()
-identity = IdentityCore()
-
-# === 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
+def run_agent(prompt):
+    return agent.run(prompt)
 
 iface = gr.Interface(
-    fn=interact,
+    fn=run_agent,
     inputs="text",
     outputs="text",
     title="🧠 AgentOS MVP",
-    description="Self-learning agent with memory, telemetry, and identity verification."
-)
-
-# === CONTEXT QUERY TAB ===
-def query_context_ui(agent_id, keyword):
-    return "\n".join(graph.query_context(agent_id, keyword))
-
-context_view = gr.Interface(
-    fn=query_context_ui,
-    inputs=[
-        gr.Textbox(label="Agent ID"),
-        gr.Textbox(label="Keyword")
-    ],
-    outputs="text",
-    title="🕸 Context Memory Query",
-    description="Search what the agent already knows or has linked in memory."
-)
-
-# === IDENTITY VIEWER TAB ===
-def show_identities():
-    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."
-
-dna_viewer = gr.Interface(
-    fn=show_identities,
-    inputs=[],
-    outputs="text",
-    title="🔐 Digital DNA Viewer",
-    description="View all agent identities and their cryptographic signatures."
-)
-
-# === 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, context_view, dna_viewer, verify_btn],
-    ["AgentOS Core", "Context Query", "Digital DNA Viewer", "Verify Identity"]
+    description="Self-learning agent with vector memory, telemetry, and personality."
 )
 
 if __name__ == "__main__":
-    app.launch(share=True)
+    iface.launch()