Update tools/detect_elements.py

tools/detect_elements.py

@@ -4,6 +4,9 @@ import numpy as np
 import os
 import json
 from typing import Optional
+import logging
+
+logger = logging.getLogger(__name__)
 
 class DetectElementsTool(Tool):
     name = "detect_elements"
@@ -14,7 +17,14 @@ class DetectElementsTool(Tool):
     }
     output_type = "string"
 
+    def __init__(self):
+        super().__init__()
+        self.is_initialized = True  # No external dependencies like driver, so always True
+        logger.debug(f"DetectElementsTool initialized: is_initialized={self.is_initialized}")
+
     def forward(self, screenshot_path: Optional[str], element_type="table"):
+        if not self.is_initialized:
+            return "Error: DetectElementsTool is not initialized"
         try:
             if not os.path.exists(screenshot_path):
                 return f"Screenshot not found: {screenshot_path}"
@@ -55,4 +65,4 @@ class DetectElementsTool(Tool):
                 "output_image": output_path
             }) if detections else "No elements detected"
         except Exception as e:
-            return f"Failed to detect elements: {str(e)}" 
+            return f"Failed to detect elements: {str(e)}"