tool.py

from typing import Any, Optional
from smolagents.tools import Tool

class SuperheroPartyThemeTool(Tool):
    name = "superhero_party_theme_generator"
    description = """
    This tool suggests creative superhero-themed party ideas based on a category. This is related to Indian Origin Flavour.
    It returns a unique party theme idea."""
    inputs = {'category': {'type': 'string', 'description': "The type of superhero party (e.g., 'desi superhero sammelan', 'bollywood khalnayak mehfil', 'digital diwali 2050')."}}
    output_type = "string"

    def forward(self, category: str):
        indian_vegetarian_themes = {
            "desi superhero sammelan": "A celebration of both traditional Indian heroes and modern superheroes with vegetarian dishes like 'Paneer Shaktiman', 'Wonder Woman Dahi Vada', and 'Green Lantern's Kiwi Jaljeera'.",
            "bollywood khalnayak mehfil": "A mysterious gathering inspired by iconic Indian villains featuring 'Mogambo's Malai Kofta', 'Gabbar Singh's Gunpowder Idli', and 'Poison Ivy's Paan Mojito'.",
            "digital diwali 2050": "A futuristic take on Indian celebration with high-tech presentation of traditional cuisine including 'Holographic Handi', 'Molecular Manchurian', and 'Quantum Kulfi'."
        }

        return indian_vegetarian_themes.get(category.lower(), "Themed party idea not found. Try 'desi superhero sammelan', 'bollywood khalnayak mehfil', or 'digital diwali 2050'.")

    def __init__(self, *args, **kwargs):
        self.is_initialized = False