from smolagents import Tool


class PartyThemeTool(Tool):
    name = "party_theme_generator"
    description = """
    This tool suggests creative party themes based on a menu category.
    It returns a unique party theme idea.
    """

    inputs = {
        "category": {
            "type": "string",
            "description": "The type of occasion categorized by categorize_menu().",
        }
    }

    output_type = "string"

    themes = {
        "casual dining": "Backyard Bash: A relaxed and fun gathering with BBQ, picnic-style seating, and outdoor games.",
        "formal dining": "Black Tie Soirée: An elegant dinner party with gourmet dishes, fine wines, and sophisticated decor.",
        "kids’ birthday party": "Magical Wonderland: A colorful and playful party with themed decorations, fun activities, and a candy buffet.",
        "brunch": "Sunny Garden Brunch: A fresh and delightful morning gathering with pastries, mimosas, and floral decor.",
        "wedding reception": "Fairy Tale Wedding: A romantic and celebratory event with elegant decor, live music, and exquisite dining.",
        "cocktail party": "Gatsby Glam: A chic 1920s-inspired cocktail party with jazz music, fancy drinks, and dazzling outfits.",
        "holiday feast": "Winter Wonderland: A festive holiday dinner with seasonal dishes, twinkling lights, and cozy ambiance.",
        "BBQ cookout": "Texas-Style BBQ: A bold and smoky feast featuring grilled meats, country music, and rustic decor.",
        "picnic": "Spring Picnic: A laid-back outdoor meal with fresh sandwiches, fruit baskets, and checkered blankets.",
        "game day": "Sports Fan Frenzy: A high-energy gathering with big screens, wings, and team spirit decorations.",
        "afternoon tea": "Royal Tea Party: An elegant affair with fine china, delicate pastries, and a selection of gourmet teas.",
        "business luncheon": "Executive Networking Lunch: A professional yet refined meal with business-friendly dishes and a sleek ambiance.",
        "farm-to-table": "Organic Harvest Feast: A sustainable and wholesome dining experience with locally sourced ingredients.",
        "tasting menu": "Chef’s Special Tasting: A curated multi-course experience with artisanal flavors and expertly paired wines.",
        "street food festival": "Global Street Feast: A lively festival with food trucks, international flavors, and a bustling atmosphere.",
        "healthy & wellness": "Zen Wellness Retreat: A health-conscious event featuring nutritious meals, yoga, and detox drinks.",
        "late-night bites": "Midnight Munchies: A cozy gathering with comfort food, neon lights, and a relaxed, late-night vibe.",
        "buffet": "Grand Buffet Extravaganza: A diverse dining experience with an endless selection of dishes from around the world.",
        "family-style": "Sunday Family Feast: A warm and generous meal shared with loved ones around a big communal table.",
        "vegan feast": "Green Gourmet Gala: A plant-based celebration with vibrant, wholesome dishes and eco-friendly decor.",
        "fine dining": "Michelin Star Experience: A sophisticated evening inspired by world-class restaurants, featuring gourmet multi-course meals, expertly paired wines and candlelit ambiance.",
        "super healthy food": "Pure Vitality Retreat: A rejuvenating event focused on nutrient-rich superfoods, fresh organic produce, and plant-based dishes with herbal infusions.",
    }

    def forward(self, category: str):
        return self.themes.get(
            category.lower(),
            "Casual Gathering: A relaxed get-together with easygoing vibes and good food.",
        )