Upload tool

tool.py

@@ -1,41 +1,39 @@
 from smolagents.tools import Tool
-import json
+import string
 import pronouncing
+import json
 
 class ParodyWordSuggestionTool(Tool):
     name = "parody_word_suggester"
     description = """Suggests phonetically similar funny words using CMU dictionary pronunciations.
     Returns similar words from a curated list of funny words, along with their similarity scores."""
-    inputs = {'target': {'type': 'string', 'description': 'The word you want to find funny alternatives for'}, 'min_similarity': {'type': 'string', 'description': 'Minimum similarity threshold (0.0-1.0)', 'nullable': True}}
+    inputs = {'target': {'type': 'string', 'description': 'The word you want to find funny alternatives for'}, 'word_list': {'type': 'string', 'description': 'JSON string of words list, e.g. \'["word1", "word2"]\''}, 'min_similarity': {'type': 'string', 'description': 'Minimum similarity threshold (0.0-1.0)', 'nullable': True}}
     output_type = "string"
 
-    def __init__(self):
-        # Initialize with funny word list
-        self.funny_words = [
-            # Original words
-            "poop", "loop", "soup", "boop", "dupe", "goop",
-            "scoop", "whoop", "droop", "snoop", "coop",
-            "fart", "toot", "burp", "squish", "squash",
-            "wiggle", "jiggle", "wobble", "nincompoop", "dingleberry",
-            # Additional middle school humor words
-            "stinky", "booboo", "smelly", "goo", "turd", "pee", "peepee",
-            "potty", "butt", "tooty", "booger", "snot", "doody", "doodoo",
-            "weewee", "wedgie", "undies", "fartface", "poopyhead",
-            "butthead", "stinkbomb", "stinkface", "dork", "nerd",
-            "goober", "doofus", "dingus", "yahoo", "weirdo",
-            "stinkypants", "bubblebutt", "zonked", "zoop", "zoinks"
-        ]
-
-
-    def forward(self, target: str, min_similarity: str = "0.7") -> str:
+    def forward(self, target: str, word_list: str, min_similarity: str = "0.7") -> str:
         """Get phonetically similar funny word suggestions."""
         import pronouncing
+        import string
         import json
     
-        target = target.lower().strip()
+        target = target.lower().strip(string.punctuation)
         min_similarity = float(min_similarity)
         suggestions = []
     
+        # Parse word list from JSON string
+        try:
+            words = json.loads(word_list)
+            if not isinstance(words, list):
+                return json.dumps({
+                    "error": "word_list must be a JSON string representing a list of words",
+                    "suggestions": []
+                }, indent=2)
+        except json.JSONDecodeError:
+            return json.dumps({
+                "error": "Invalid JSON string for word_list",
+                "suggestions": []
+            }, indent=2)
+        
         # Get target pronunciation
         target_phones = pronouncing.phones_for_word(target)
         if not target_phones:
@@ -46,8 +44,8 @@ class ParodyWordSuggestionTool(Tool):
         
         target_phones = target_phones[0].split()
     
-        # Check each funny word
-        for word in self.funny_words:
+        # Check each word
+        for word in words:
             phones = pronouncing.phones_for_word(word)
             if phones:  # Only process if word is in dictionary
                 phones = phones[0].split()
@@ -75,3 +73,7 @@ class ParodyWordSuggestionTool(Tool):
     
         return json.dumps(result, indent=2)
 
+
+    def __init__(self, *args, **kwargs):
+        self.is_initialized = False
+