Update agents/image_agent.py

agents/image_agent.py

@@ -16,6 +16,16 @@ class ImageAgent:
         self.memory.add_files(paths)
         return paths
 
+    def auto_analyze(self, image_path):
+        """
+        STEP 1: Always understand image first
+        """
+        if image_path:
+            analysis = analyze_image(image_path)
+            self.memory.last_used_image = image_path
+            return analysis
+        return None
+
     def run(self, user_input: str):
 
         task = self.router.route(user_input)
@@ -23,27 +33,30 @@ class ImageAgent:
 
         result = None
 
+        # 🔥 STEP 0: ALWAYS pick image first
+        image = None
+
+        if self.memory.uploaded_files:
+            image = self.memory.uploaded_files[0]
+
+        # 🔥 STEP 1: ANALYZE FIRST (IMPORTANT)
+        if image:
+            analysis = self.auto_analyze(image)
+            self.memory.analysis = analysis
+
+        # 🔥 STEP 2: EXECUTE TASK
         for step in steps:
 
             if step == "generate":
                 result = generate_image(user_input)
 
             elif step == "edit":
-                image = self.memory.get_last_image()
-
-                if not image and self.memory.uploaded_files:
-                    image = self.memory.uploaded_files[0]
-
+                image = self.memory.last_used_image or image
                 result = edit_image(image, user_input)
-
-                self.memory.set_last_image(result)
+                self.memory.last_used_image = image
 
             elif step == "analyze":
-                image = self.memory.get_last_image()
-
-                if not image:
-                    image = self.memory.uploaded_files[0]
-
+                image = self.memory.last_used_image or image
                 result = analyze_image(image)
 
         return result