Upload tool

requirements.txt

@@ -1,2 +1,2 @@
-smolagents
 pronouncing
+smolagents

tool.py

@@ -1,8 +1,8 @@
 from smolagents.tools import Tool
-import json
+import pronouncing
 import difflib
 import string
-import pronouncing
+import json
 
 class ParodyWordSuggestionTool(Tool):
     name = "parody_word_suggester"
@@ -10,32 +10,28 @@ class ParodyWordSuggestionTool(Tool):
     Returns similar-sounding words that rhyme, especially focusing on common vowel sounds."""
     inputs = {'target': {'type': 'string', 'description': 'The word you want to find rhyming alternatives for'}, 'word_list_str': {'type': 'string', 'description': 'JSON string of word list (e.g. \'["word1", "word2"]\')'}, 'min_similarity': {'type': 'string', 'description': 'Minimum similarity threshold (0.0-1.0)', 'nullable': True}}
     output_type = "string"
+    VOWEL_REF = "AH,UH,AX|AE,EH|IY,IH|AO,AA|UW,UH|AY,EY|OW,AO|AW,AO|OY,OW|ER,AXR"
 
-    def __init__(self):
-        self.vowel_groups = [
-            {'AH', 'UH', 'AX'},  # 'luck', 'buck', 'cuck'
-            {'AE', 'EH'},        # 'bat', 'bet'
-            {'IY', 'IH'},        # 'seat', 'sit'
-            {'AO', 'AA'},        # 'caught', 'cot'
-            {'UW', 'UH'},        # 'boot', 'put'
-            {'AY', 'EY'},        # 'bite', 'bait'
-            {'OW', 'AO'},        # 'boat', 'bought'
-            {'AW', 'AO'},        # 'bout', 'bought'
-            {'OY', 'OW'},        # 'boy', 'bow'
-            {'ER', 'AXR'},       # 'bird', 'hurt'
-        ]
+    def _get_vowel_groups(self):
+        """Convert the simple string format to usable groups."""
+        groups = []
+        group_strs = self.VOWEL_REF.split("|")
+        for group_str in group_strs:
+            groups.append(group_str.split(","))
+        return groups
 
 
     def _get_last_syllable(self, phones: list) -> tuple:
         """Extract the last syllable (vowel + remaining consonants)."""
         last_vowel_idx = -1
         last_vowel = None
+        vowel_groups = self._get_vowel_groups()
     
         # Find the last vowel
         for i, phone in enumerate(phones):
             # Strip stress markers for checking
             base_phone = phone.rstrip('012')
-            for group in self.vowel_groups:
+            for group in vowel_groups:
                 if base_phone in group:
                     last_vowel_idx = i
                     last_vowel = base_phone
@@ -58,7 +54,8 @@ class ParodyWordSuggestionTool(Tool):
         if v1 == v2:
             return True
         
-        for group in self.vowel_groups:
+        vowel_groups = self._get_vowel_groups()
+        for group in vowel_groups:
             if v1 in group and v2 in group:
                 return True
         return False
@@ -161,3 +158,7 @@ class ParodyWordSuggestionTool(Tool):
     
         return json.dumps(result, indent=2)
 
+
+    def __init__(self, *args, **kwargs):
+        self.is_initialized = False
+