Update agents/memory.py

agents/memory.py

@@ -4,18 +4,10 @@ class Memory:
         self.chat_history = []
         self.uploaded_files = []
         self.last_used_image = None
+        self.image_context = None   # 🔥 NEW
 
-    def add_message(self, role, message):
-        self.chat_history.append({
-            "role": role,
-            "message": message
-        })
+    def set_image_context(self, context):
+        self.image_context = context
 
-    def add_files(self, files):
-        self.uploaded_files.extend(files)
-
-    def get_last_image(self):
-        return self.last_used_image
-
-    def set_last_image(self, img_path):
-        self.last_used_image = img_path
+    def get_image_context(self):
+        return self.image_context