MoinRomanticbot / memory.py
syedmoinms's picture
Update memory.py
87fb4ba verified
raw
history blame contribute delete
563 Bytes
import json
MEMORY_FILE = "memory.json"
def load_memory():
try:
with open(MEMORY_FILE, "r", encoding="utf-8") as f:
return json.load(f)
except FileNotFoundError:
return {}
def save_memory(memory):
with open(MEMORY_FILE, "w", encoding="utf-8") as f:
json.dump(memory, f, indent=4)
def update_memory(user_input, bot_response):
memory = load_memory()
memory[user_input] = bot_response
save_memory(memory)
def check_memory(user_input):
memory = load_memory()
return memory.get(user_input, None)