from .detection_model import run_detection
from .segmentation_model import run_segmentation

__all__ = ["run_detection", "run_segmentation"]

FUNCTION_SCHEMA = [
    {
        "type": "function",
        "function": {
            "name": "run_detection",
            "description": "Detect objects in an image and return bounding boxes and labels.",
            "parameters": {
                "type": "object",
                "properties": {
                    "image_path":  {"type": "string", "description": "Local path to the image file."},
                    "is_visualize":{"type": "bool",   "description": "If true draw bboxes and save next to image."}
                },
                "required": ["image_path"]
            },
        },
    },
    {
        "type": "function",
        "function": {
            "name": "run_segmentation",
            "description": "Segment objects in an image and return binary masks.",
            "parameters": {
                "type": "object",
                "properties": {
                    "image_path": {"type": "string", "description": "Local path to the image file."}
                },
                "required": ["image_path"]
            },
        },
    },
]