Update tools/detect_elements.py

tools/detect_elements.py

@@ -2,6 +2,7 @@ from smolagents.tools import Tool
 import cv2
 import numpy as np
 import os
+import json
 
 def detect_elements(screenshot_path, element_type="table"):
     """
@@ -57,12 +58,19 @@ def detect_elements(screenshot_path, element_type="table"):
         return f"Failed to detect elements: {str(e)}"
 
 # Register the tool
-tool = Tool(
+detect_elements_tool = 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": "str",
+            "description": "Path to the screenshot"
+        },
+        "element_type": {
+            "type": "str",
+            "default": "table",
+            "description": "Type: 'table' or 'textbox'"
+        }
     },
     output_type="str",
     function=detect_elements