"""Fal tool configuration."""

from __future__ import annotations

from typing import TYPE_CHECKING

from mindroom.tool_system.declarations import ConfigField, SetupType, ToolCategory, ToolStatus
from mindroom.tool_system.registration import register_tool_with_metadata

if TYPE_CHECKING:
    from agno.tools.fal import FalTools


@register_tool_with_metadata(
    name="fal",
    display_name="Fal",
    description="AI model serving platform for media generation (images and videos)",
    category=ToolCategory.DEVELOPMENT,  # others category maps to DEVELOPMENT
    status=ToolStatus.REQUIRES_CONFIG,  # requires FAL_KEY API key
    setup_type=SetupType.API_KEY,  # uses API key authentication
    icon="FaRobot",  # AI/robot icon for AI model serving
    icon_color="text-purple-600",  # Purple for AI/ML services
    config_fields=[
        ConfigField(
            name="api_key",
            label="API Key",
            type="password",
            required=False,
            default=None,
        ),
        ConfigField(
            name="model",
            label="Model",
            type="text",
            required=False,
            default="fal-ai/hunyuan-video",
        ),
        ConfigField(
            name="enable_generate_media",
            label="Enable Generate Media",
            type="boolean",
            required=False,
            default=True,
        ),
        ConfigField(
            name="enable_image_to_image",
            label="Enable Image To Image",
            type="boolean",
            required=False,
            default=False,
        ),
        ConfigField(
            name="all",
            label="All",
            type="boolean",
            required=False,
            default=False,
        ),
    ],
    dependencies=["fal-client"],  # From agno requirements
    docs_url="https://docs.agno.com/tools/toolkits/others/fal",  # URL without .md extension
    function_names=("generate_media", "image_to_image", "on_queue_update"),
)
def fal_tools() -> type[FalTools]:
    """Return Fal tools for AI model serving and media generation."""
    from agno.tools.fal import FalTools

    return FalTools
