from smolagents import Tool
from typing import Any, Optional

class SimpleTool(Tool):
    name = "suggest_tool"
    description = "Suggest a menu based on the occasion."
    inputs = {'occasion': {'type': 'string', 'description': 'The type of occasion for the party. Allowed values are: "casual", "formal", "superhero", "custom".'}}
    output_type = "string"

    def forward(self, occasion: str) -> str:
        """
        Suggest a menu based on the occasion.
        Args:
            occasion (str): The type of occasion for the party. Allowed values are:
                            "casual", "formal", "superhero", "custom".
        """

        if occasion == "casual":
            return "Pizza, Snacks and Drinks"

        elif occasion == "formal":
            return "3 course dinner with wine and desert"

        elif occasion == "superhero":
            return "Buffet with high energy and healthy food"

        else:
            return "Custom Menu for the butler."