Update memory.py

memory.py

@@ -1,14 +1,27 @@
-# memory.py
+import json
+import os
+
+MEMORY_FILE = "memory.json"
 
 def init_memory():
-    # Initialize memory as a simple dict with a 'texts' list
-    return {"texts": []}
-
-def add_to_memory(text, memory):
-    # Append text to memory, keep last 100 entries
-    memory["texts"].append(text)
-    if len(memory["texts"]) > 100:
-        memory["texts"] = memory["texts"][-100:]
+    if not os.path.exists(MEMORY_FILE):
+        with open(MEMORY_FILE, "w") as f:
+            json.dump({"history": []}, f)
+    with open(MEMORY_FILE, "r") as f:
+        return json.load(f)
+
+def save_memory(memory):
+    with open(MEMORY_FILE, "w") as f:
+        json.dump(memory, f, indent=2)
+
+def add_to_memory(prompt, response):
+    memory = init_memory()
+    memory["history"].append({"prompt": prompt, "response": response})
+    memory["history"] = memory["history"][-50:]  # limit to last 50
+    save_memory(memory)
+
+def get_memory():
+    return init_memory()["history"]
 
 def search_memory(query, memory):
     # Placeholder: Return last few memory entries related to query (simple all for now)