import os

from .nodes.agent import CreateAgent, RunAgent, ExpandAgent
from .nodes.config import (
    gtUIOpenAiStructureConfig,
    gtUIAmazonBedrockStructureConfig,
    gtUIGoogleStructureConfig,
    gtUIAnthropicStructureConfig,
)
from .nodes.tools import gtUIDateTime, gtUICalculator, gtUIWebScraper, gtUIFileManager
from .nodes.list import ToolList
from .nodes.tasks import (
    gtUIGroqPromptTask,
    gtUIPromptTask,
    gtUIToolTask,
    gtUITextSummaryTask,
    gtUIToolkitTask,
    gtUIImageQueryTask,
    gtUIPromptImageGenerationTask,
    gtUIPromptImageVariationTask,
)
from .nodes.drivers import (
    gtUIOpenAiImageGenerationDriver,
    gtUIAmazonBedrockStableDiffusionImageGenerationDriver,
    gtUIAmazonBedrockTitanImageGenerationDriver,
    gtUILeonardoImageGenerationDriver,
)

from .nodes.output_nodes import gtUIOutputStringNode, gtUIOutputImageNode
from .nodes.input_nodes import gtUIInputStringNode
from .nodes.html_node import HtmlNode
from .nodes.rules import gtUIRule
from .nodes.string_nodes import JoinStringListNode

NODE_CLASS_MAPPINGS = {}
NODE_DISPLAY_NAME_MAPPINGS = {}
WEB_DIRECTORY = "./js"

NODE_CLASS_MAPPINGS = {
    # "HtmlNode": HtmlNode,
    "gtUIInputNode": gtUIInputStringNode,
    "gtUIOpenAiImageGenerationDriver": gtUIOpenAiImageGenerationDriver,
    "gtUIAmazonBedrockStableDiffusionImageGenerationDriver": gtUIAmazonBedrockStableDiffusionImageGenerationDriver,
    "gtUIAmazonBedrockTitanImageGenerationDriver": gtUIAmazonBedrockTitanImageGenerationDriver,
    "gtUILeonardoImageGenerationDriver": gtUILeonardoImageGenerationDriver,
    "gtUIOutputStringNode": gtUIOutputStringNode,
    "gtUIOutputImageNode": gtUIOutputImageNode,
    "gtUIGroqPromptTask": gtUIGroqPromptTask,
    "CreateAgent": CreateAgent,
    "RunAgent": RunAgent,
    "ExpandAgent": ExpandAgent,
    "gtUIOpenAiStructureConfig": gtUIOpenAiStructureConfig,
    "gtUIAmazonBedrockStructureConfig": gtUIAmazonBedrockStructureConfig,
    "gtUIGoogleStructureConfig": gtUIGoogleStructureConfig,
    "gtUIAnthropicStructureConfig": gtUIAnthropicStructureConfig,
    "Calculator": gtUICalculator,
    "DateTime": gtUIDateTime,
    "WebScraper": gtUIWebScraper,
    "gtUIFileManager": gtUIFileManager,
    "ToolList": ToolList,
    "PromptTask": gtUIPromptTask,
    "ToolTask": gtUIToolTask,
    "ToolkitTask": gtUIToolkitTask,
    "ImageQueryTask": gtUIImageQueryTask,
    "PromptImageGenerationTask": gtUIPromptImageGenerationTask,
    "PromptImageVariationTask": gtUIPromptImageVariationTask,
    "TextSummaryTask": gtUITextSummaryTask,
    "JoinStringListNode": JoinStringListNode,
    "Rule": gtUIRule,
}

NODE_DISPLAY_NAME_MAPPINGS = {
    # "HtmlNode": "Griptape: HTML",
    "gtUIInputNode": "Griptape Input: String",
    "gtUIOpenAiImageGenerationDriver": "Griptape Driver: OpenAI Image Generation",
    "gtUIAmazonBedrockStableDiffusionImageGenerationDriver": "Griptape Driver: Amazon Bedrock Stable Diffusion Image Generation",
    "gtUIAmazonBedrockTitanImageGenerationDriver": "Griptape Driver: Amazon Bedrock Titan Image Generation",
    "gtUILeonardoImageGenerationDriver": "Griptape Driver: Leonardo Image Generation",
    "gtUIOutputStringNode": "Griptape Preview: String",
    "gtUIOutputImageNode": "Griptape Preview: Image",
    "gtUIGroqPromptTask": "Griptape Task: Groq Prompt",
    "CreateAgent": "Griptape: Create Agent",
    "RunAgent": "Griptape: Run Agent",
    "ExpandAgent": "Griptape: Expand Agent Nodes",
    "gtUIOpenAiStructureConfig": "Griptape Config: OpenAI",
    "gtUIAmazonBedrockStructureConfig": "Griptape Config: Amazon Bedrock",
    "gtUIGoogleStructureConfig": "Griptape Config: Google",
    "gtUIAnthropicStructureConfig": "Griptape Config: Anthropic",
    "Calculator": "Griptape Tool: Calculator",
    "DateTime": "Griptape Tool: DateTime",
    "WebScraper": "Griptape Tool: WebScraper",
    "gtUIFileManager": "Griptape Tool: FileManager",
    "ToolList": "Griptape List: Create Tool List",
    "PromptTask": "Griptape Task: Prompt",
    "ToolTask": "Griptape Task: Tool",
    "ToolkitTask": "Griptape Task: Toolkit",
    "ImageQueryTask": "Griptape Task: Image Query",
    "PromptImageGenerationTask": "Griptape Task: Image Generation",
    "PromptImageVariationTask": "Griptape Task: Image Variation",
    "TextSummaryTask": "Griptape Task: Text Summary",
    "JoinStringListNode": "Griptape: Join String List",
    "Rule": "Griptape: Create Rules",
}

__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]