import json
import os

MEMORY_FILE = "memory.json"

def init_memory():
    if not os.path.exists(MEMORY_FILE):
        with open(MEMORY_FILE, "w") as f:
            json.dump({}, f)
    with open(MEMORY_FILE, "r") as f:
        return json.load(f)

def save_memory(data):
    with open(MEMORY_FILE, "w") as f:
        json.dump(data, f, indent=2)

def add_to_memory(session_id, prompt, response):
    memory = init_memory()
    if session_id not in memory:
        memory[session_id] = []
    memory[session_id].append({"prompt": prompt, "response": response})
    memory[session_id] = memory[session_id][-50:]
    save_memory(memory)

def get_memory(session_id):
    memory = init_memory()
    return memory.get(session_id, [])

def search_memory(query, memory):
    # Placeholder: Return last few memory entries related to query (simple all for now)
    return memory["texts"][-5:]

def get_summary(session_id, max_items=3):
    history = get_memory(session_id)
    return history[-max_items:] if history else []