class Memory:

    def __init__(self):
        self.chat_history = []
        self.uploaded_files = []
        self.last_used_image = None
        self.image_context = None   # 🔥 NEW

    def set_image_context(self, context):
        self.image_context = context

    def get_image_context(self):
        return self.image_context