updated - memory_store, persistent memory in api_server, app.js

api_server.py

@@ -206,9 +206,9 @@ def chat(req: ChatRequest, background_tasks: BackgroundTasks):
     # -----------------------------
     output = handle_turn(state, message)
 
-    # -----------------------------------
-    # SAVE AFTER EVERY TURN (PERSISTENCE)
-    # -----------------------------------
+    # -----------------------------------------
+    # SAVE STATE AFTER EVERY TURN (PERSISTENCE)
+    # -----------------------------------------
     save_state(session_id, state)
 
     # -----------------------------
@@ -239,15 +239,13 @@ def chat(req: ChatRequest, background_tasks: BackgroundTasks):
 # ---- GET SESSION DATA, PAST CONVERSATION ---- 
 @app.get("/session/{session_id}")
 def get_session(session_id: str):
-    state = load_state(session_id)
 
-    if not state:
+    data = load_state(session_id)
+
+    if not data:
         return {"error": "Session not found"}
 
-    return {
-        "history": state.history,
-        "agent_histories": state.agent_histories
-    }
+    return data
 
 
 

core/memory_store.py

@@ -3,6 +3,7 @@
 
 import json
 import os
+from datetime import datetime
 
 MEMORY_DIR = "memory"
 
@@ -10,37 +11,39 @@ if not os.path.exists(MEMORY_DIR):
     os.makedirs(MEMORY_DIR)
 
 
+def generate_label(history):
+    """
+    Simple label generator (can upgrade later with LLM)
+    """
+    for msg in history:
+        if msg["role"] == "client":
+            return msg["content"][:40]
+    return "Conversation"
+
+
 def save_state(session_id, state):
-    path = f"{MEMORY_DIR}/{session_id}.json"
+
+    filepath = os.path.join(MEMORY_DIR, f"{session_id}.json")
 
     data = {
+        "session_id": session_id,
+        "created_at": getattr(state, "created_at", datetime.now().isoformat()),
+        "last_updated": datetime.now().isoformat(),
+        "label": generate_label(state.history),
         "history": state.history,
-        "agent_histories": state.agent_histories,
-        "initial_query": state.initial_query,
-        "current_agent": state.current_agent,
-        "stage": state.stage
+        "agents": getattr(state, "agent_logs", {})
     }
 
-    with open(path, "w") as f:
-        json.dump(data, f)
+    with open(filepath, "w", encoding="utf-8") as f:
+        json.dump(data, f, indent=2)
 
 
 def load_state(session_id):
-    path = f"{MEMORY_DIR}/{session_id}.json"
 
-    if not os.path.exists(path):
-        return None
+    filepath = os.path.join(MEMORY_DIR, f"{session_id}.json")
 
-    with open(path, "r") as f:
-        data = json.load(f)
-
-    from core.conversation_manager import ConversationState
-
-    state = ConversationState()
-    state.history = data.get("history", [])
-    state.agent_histories = data.get("agent_histories", {})
-    state.initial_query = data.get("initial_query")
-    state.current_agent = data.get("current_agent", "jung")
-    state.stage = data.get("stage", "jung")
+    if not os.path.exists(filepath):
+        return None
 
-    return state
+    with open(filepath, "r", encoding="utf-8") as f:
+        return json.load(f)

web/app.js

@@ -135,10 +135,6 @@ function switchToConversationMode() {
 //function loadSession() {
 async function loadSession() {
 
-    document.getElementById("session_display").value = input
-    window.conversationStarted = true
-    switchToConversationMode()
-
     const input = document.getElementById("session_input").value
 
     if (!input) return
@@ -153,19 +149,29 @@ async function loadSession() {
         return
     }
 
-    // Restore chat visually
+    // Set session display
+    document.getElementById("session_display").value = input
+
+    // Switch UI mode
+    window.conversationStarted = true
+    switchToConversationMode()
+
+    // Clear chat - Restore chat visually
     const chatBox = document.getElementById("chat")
     chatBox.innerHTML = ""
 
+    // Restore messages
     data.history.forEach(msg => {
+
         if (msg.role === "client") {
             appendMessage("user", msg.content)
         } else {
             updateThinking(msg.content, "jung")
         }
+
     })
 
-    //console.log("♻️ Loading session:", input)
+    console.log("♻️ Session loaded:", input)
 }
 
 
@@ -431,41 +437,39 @@ function formatEpistemic(text) {
         .replace(/\n/g, "<br>")
 }
 
-// -------------------------------
+// ------------
 // END SESSION
-// -------------------------------
+// ------------
 async function endSession() {
 
     console.log("🛑 Ending session")
 
+    const sessionId = window.session_id
+
     await fetch(window.location.origin + "/chat", {
         method: "POST",
         headers: { "Content-Type": "application/json" },
         body: JSON.stringify({
-            session_id: window.session_id,
+            session_id: sessionId,
             message: "end session"
         })
     })
 
+    // 🔥 POPUP MESSAGE
+    if (sessionId) {
+        alert(`Session ended.\n\nUse Session ID: ${sessionId} to resume later.`)
+    }
+
     window.session_id = null
+    window.conversationStarted = false
+
     document.getElementById("chat").innerHTML = ""
+    document.getElementById("session_display").value = ""
 }
 
 // ---------------------------------------------------------
 // ENTER(SEND ACTION) & SHIFT+ENTER(INSERT NEW LINE) CONTROL
 // ---------------------------------------------------------
-//document.getElementById("main_input").addEventListener("keydown", function (e) {
-//    if (e.key === "Enter" && !e.shiftKey) {
-//        e.preventDefault()
-
-//        if (window.conversationStarted) {
-//            fuel()
-//        } else {
-//            fire()
-//        }
-//    }
-//})
-
 window.addEventListener("DOMContentLoaded", () => {
     const input = document.getElementById("main_input")