memory.py

import os
import json
import datetime

MEMORY_FILE = "memory/conversations.json"
TOPICS_FILE = "memory/learned_topics.json"

def load_memory():
    os.makedirs("memory", exist_ok=True)
    if os.path.exists(MEMORY_FILE):
        with open(MEMORY_FILE, "r", encoding="utf-8") as f:
            return json.load(f)
    return []

def save_memory(history, rating=None):
    memory = load_memory()
    session = {
        "date": datetime.datetime.now().strftime("%Y-%m-%d %H:%M"),
        "messages": history,
        "rating": rating
    }
    memory.append(session)
    with open(MEMORY_FILE, "w", encoding="utf-8") as f:
        json.dump(memory, f, indent=2, ensure_ascii=False)
    print("[Memory] Session saved!")

def get_memory_context():
    memory = load_memory()
    if not memory:
        return ""
    
    context = ""
    # All recent conversations use 
    for session in memory[-5:]:
        for msg in session["messages"]:
            if msg["role"] == "user":
                context += f"Past Q: {msg['content']}\n"
            elif msg["role"] == "assistant":
                context += f"Past A: {msg['content'][:300]}\n"
        context += "\n"
    return context[:3000]

def save_learned_topic(topic):
    os.makedirs("memory", exist_ok=True)
    topics = []
    if os.path.exists(TOPICS_FILE):
        with open(TOPICS_FILE, "r") as f:
            topics = json.load(f)
    if topic not in topics:
        topics.append(topic)
        with open(TOPICS_FILE, "w") as f:
            json.dump(topics, f, indent=2)

def get_learned_topics():
    if os.path.exists(TOPICS_FILE):
        with open(TOPICS_FILE, "r") as f:
            return json.load(f)
    return []

def get_feedback():
    while True:
        try:
            rating = input("\n[Feedback] Rate Me (1-5 OR Enter skip): ").strip()
            if rating == "":
                return None
            rating = int(rating)
            if 1 <= rating <= 5:
                return rating
            print("write in between 1 to 5")
        except ValueError:
            return None