Update tools/detect_elements.py

tools/detect_elements.py

@@ -8,14 +8,12 @@ class DetectElementsTool(Tool):
     name = "detect_elements"
     description = "Detects table-like structures or text boxes in a screenshot using OpenCV."
     inputs = {
-        "screenshot_path": {"type": "str", "description": "Path to the screenshot"},
-        "element_type": {"type": "str", "default": "table", "description": "Type: 'table' or 'textbox'"}
+        "screenshot_path": {"type": "string", "description": "Path to the screenshot"},
+        "element_type": {"type": "string", "default": "table", "description": "Type: 'table' or 'textbox'"}
     }
-    output_type = "str"
+    output_type = "string"
 
-    def forward(self, **kwargs):
-        screenshot_path = kwargs.get("screenshot_path")
-        element_type = kwargs.get("element_type", "table")
+    def forward(self, screenshot_path, element_type="table"):
         try:
             if not os.path.exists(screenshot_path):
                 return f"Screenshot not found: {screenshot_path}"