Upload tool

requirements.txt

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

tool.py

@@ -1,8 +1,8 @@
 from smolagents.tools import Tool
-import pronouncing
-import difflib
 import json
 import string
+import pronouncing
+import difflib
 
 class ParodyWordSuggestionTool(Tool):
     name = "parody_word_suggester"
@@ -11,39 +11,21 @@ class ParodyWordSuggestionTool(Tool):
     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 _are_vowels_similar(self, v1: str, v2: str) -> bool:
-        """Check if two vowel sounds are similar enough to rhyme."""
-        # Strip stress markers
-        v1 = v1.rstrip('012')
-        v2 = v2.rstrip('012')
-    
-        # Define groups of similar vowel sounds
-        similar_vowels = [
-            {'AH', 'UH'},  # Short u sounds
-            {'AE', 'EH'},  # Short e/a sounds
-            {'IY', 'IH'},  # Long/short i sounds
-            {'AO', 'AA'},  # Open o/a sounds
-            {'UW', 'UH'},  # Long/short oo sounds
-        ]
-    
-        # Direct match
-        if v1 == v2:
-            return True
-        
-        # Check if they're in the same similarity group
-        for group in similar_vowels:
-            if v1 in group and v2 in group:
+    def _has_vowel(self, phone: str) -> bool:
+        """Check if a phone contains a vowel."""
+        VOWELS = ['A', 'E', 'I', 'O', 'U']
+        for vowel in VOWELS:
+            if vowel in phone:
                 return True
-            
         return False
 
 
-    def _contains_vowel(self, phone: str, vowels: list) -> bool:
-        """Helper function to check if a phone contains any vowel from the list."""
-        for v in vowels:
-            if v in phone:
-                return True
-        return False
+    def _get_rhyme_part(self, phones: list) -> list:
+        """Get the rhyming part of a word (last vowel onwards)."""
+        for i, phone in enumerate(reversed(phones)):
+            if self._has_vowel(phone):
+                return phones[-(i+1):]
+        return phones
 
 
     def forward(self, target: str, word_list_str: str, min_similarity: str = "0.5") -> str:
@@ -53,8 +35,6 @@ class ParodyWordSuggestionTool(Tool):
         import json
         from difflib import SequenceMatcher
     
-        VOWEL_LETTERS = ['A', 'E', 'I', 'O', 'U']
-    
         target = target.lower().strip(string.punctuation)
         min_similarity = float(min_similarity)
         suggestions = []
@@ -75,17 +55,10 @@ class ParodyWordSuggestionTool(Tool):
                 "error": f"'{target}' not found in CMU dictionary",
                 "suggestions": []
             }, indent=2)
-        
+    
         target_phones = target_phones[0]
         target_phone_list = target_phones.split()
-    
-        # Find the last vowel in target
-        target_last_vowel = None
-        target_last_vowel_idx = -1
-        for i, phone in enumerate(target_phone_list):
-            if self._contains_vowel(phone, VOWEL_LETTERS):
-                target_last_vowel = phone
-                target_last_vowel_idx = i
+        target_rhyme_part = self._get_rhyme_part(target_phone_list)
     
         # Check each word
         for word in words:
@@ -94,29 +67,11 @@ class ParodyWordSuggestionTool(Tool):
             if phones:
                 word_phones = phones[0]
                 word_phone_list = word_phones.split()
-            
-                # Find last vowel in word
-                word_last_vowel = None
-                word_last_vowel_idx = -1
-                for i, phone in enumerate(word_phone_list):
-                    if self._contains_vowel(phone, VOWEL_LETTERS):
-                        word_last_vowel = phone
-                        word_last_vowel_idx = i
+                word_rhyme_part = self._get_rhyme_part(word_phone_list)
             
                 # 1. Rhyme score (most important - 60%)
-                rhyme_score = 0
-                if word_last_vowel and target_last_vowel:
-                    # Check if the vowels are similar
-                    if self._are_vowels_similar(word_last_vowel, target_last_vowel):
-                        # Check if the endings after the vowel match
-                        if (word_phone_list[word_last_vowel_idx:] == 
-                            target_phone_list[target_last_vowel_idx:]):
-                            rhyme_score = 1.0
-                        # Partial match for similar endings
-                        elif (len(word_phone_list) > word_last_vowel_idx + 1 and
-                              len(target_phone_list) > target_last_vowel_idx + 1 and
-                              word_phone_list[-1] == target_phone_list[-1]):
-                            rhyme_score = 0.8
+                # Perfect rhyme if the rhyming parts match exactly
+                rhyme_score = 1.0 if word_rhyme_part == target_rhyme_part else 0.0
             
                 # 2. Syllable match (25%)
                 target_syl = pronouncing.syllable_count(target_phones)
@@ -138,7 +93,8 @@ class ParodyWordSuggestionTool(Tool):
                         "syllable_match": syllable_score == 1.0,
                         "string_similarity": round(string_similarity, 3),
                         "syllables": word_syl,
-                        "phones": word_phones
+                        "phones": word_phones,
+                        "rhyme_part": " ".join(word_rhyme_part)
                     })
     
         # Sort by similarity score descending
@@ -148,6 +104,7 @@ class ParodyWordSuggestionTool(Tool):
             "target": target,
             "target_syllables": pronouncing.syllable_count(target_phones),
             "target_phones": target_phones,
+            "target_rhyme_part": " ".join(target_rhyme_part),
             "suggestions": suggestions
         }