Upload tool

tool.py

@@ -5,17 +5,27 @@ import string
 
 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'}, '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}}
+    description = """Suggests rhyming funny words using CMU dictionary pronunciations.
+    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"
 
-    def forward(self, target: str, word_list_str: str, min_similarity: str = "0.7") -> str:
-        """Get phonetically similar funny word suggestions."""
+    def forward(self, target: str, word_list_str: str, min_similarity: str = "0.5") -> str:
+        """Get rhyming word suggestions."""
         import pronouncing
         import string
         import json
     
+        # Define vowel sound groups (common sounds that rhyme)
+        vowel_groups = {
+            'UW1': ['UW0', 'UW1', 'UW2'],  # oo sounds
+            'UW0': ['UW0', 'UW1', 'UW2'],
+            'UW2': ['UW0', 'UW1', 'UW2'],
+            'AH1': ['AH0', 'AH1', 'AH2'],  # uh sounds
+            'AH0': ['AH0', 'AH1', 'AH2'],
+            'AH2': ['AH0', 'AH1', 'AH2'],
+        }
+    
         target = target.lower().strip(string.punctuation)
         min_similarity = float(min_similarity)
         suggestions = []
@@ -40,6 +50,22 @@ class ParodyWordSuggestionTool(Tool):
         target_phones = target_phones[0]
         target_phone_list = target_phones.split()
     
+        # Focus on the vowel sound and end consonant(s)
+        target_end = []
+        found_vowel = False
+        for i in range(len(target_phone_list) - 1, -1, -1):
+            current_phone = target_phone_list[i]
+            target_end.insert(0, current_phone)
+            # Check if current phone contains a vowel
+            has_vowel = False
+            for vowel in ['A', 'E', 'I', 'O', 'U']:
+                if vowel in current_phone:
+                    has_vowel = True
+                    found_vowel = True
+                    break
+            if found_vowel:
+                break
+            
         # Check each word
         for word in words:
             phones = pronouncing.phones_for_word(word)
@@ -47,25 +73,51 @@ class ParodyWordSuggestionTool(Tool):
                 word_phones = phones[0]
                 word_phone_list = word_phones.split()
             
-                # Calculate position-aware similarity
+                # Get ending pattern (vowel + following consonants)
+                word_end = []
+                found_vowel = False
+                for i in range(len(word_phone_list) - 1, -1, -1):
+                    current_phone = word_phone_list[i]
+                    word_end.insert(0, current_phone)
+                    # Check if current phone contains a vowel
+                    has_vowel = False
+                    for vowel in ['A', 'E', 'I', 'O', 'U']:
+                        if vowel in current_phone:
+                            has_vowel = True
+                            found_vowel = True
+                            break
+                    if found_vowel:
+                        break
+            
+                # Calculate rhyme score
                 matches = 0
-                length = max(len(word_phone_list), len(target_phone_list))
+                total_checks = max(len(word_end), len(target_end))
             
-                # Compare phones at each position
-                for i in range(length):
-                    if i < len(word_phone_list) and i < len(target_phone_list):
-                        # Get current phones
-                        current_word_phone = word_phone_list[i]
-                        current_target_phone = target_phone_list[i]
-                    
-                        # Exact match at same position
-                        if current_word_phone == current_target_phone:
-                            matches += 1
-                        # Partial match (just the consonant or vowel part)
-                        elif current_word_phone.rstrip('012') == current_target_phone.rstrip('012'):
-                            matches += 0.5
+                for i in range(min(len(word_end), len(target_end))):
+                    w_phone = word_end[i]
+                    t_phone = target_end[i]
+                
+                    # Check for exact match
+                    if w_phone == t_phone:
+                        matches += 1
+                    else:
+                        # Check for vowel sound
+                        has_vowel = False
+                        for vowel in ['A', 'E', 'I', 'O', 'U']:
+                            if vowel in w_phone:
+                                has_vowel = True
+                                break
+                            
+                        if has_vowel:
+                            # Check vowel groups
+                            is_match = False
+                            for base_vowel in vowel_groups:
+                                if w_phone in vowel_groups[base_vowel] and t_phone in vowel_groups[base_vowel]:
+                                    is_match = True
+                                    matches += 1
+                                    break
             
-                similarity = matches / length if length > 0 else 0.0
+                similarity = matches / total_checks if total_checks > 0 else 0.0
             
                 if similarity >= min_similarity:
                     suggestions.append({