Update app.py

app.py

@@ -56,4 +56,63 @@ dna_viewer = gr.Interface(
 
 # 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
+
+# --- Instantiate Core Systems ---
+agent = AgentCore()
+graph = ContextGraph()
+
+# --- Define main agent function ---
+def interact(prompt):
+    response = agent.run(prompt)
+    return response
+
+iface = gr.Interface(
+    fn=interact,
+    inputs="text",
+    outputs="text",
+    title="🧠 AgentOS MVP",
+    description="Self-learning agent with memory + telemetry."
+)
+
+# --- 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."
+)
+
+# --- Digital DNA Viewer Tab ---
+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 all interfaces ---
+app = gr.TabbedInterface(
+    [iface, dna_viewer, context_view],
+    ["AgentOS Core", "Digital DNA Viewer", "Context Query"]
+)
+
 app.launch()