Upload tool

requirements.txt

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

tool.py

@@ -1,8 +1,8 @@
 from smolagents.tools import Tool
+import json
 import pronouncing
 import difflib
 import string
-import json
 
 class ParodyWordSuggestionTool(Tool):
     name = "parody_word_suggester"
@@ -46,6 +46,14 @@ class ParodyWordSuggestionTool(Tool):
         return last_vowel, remaining
 
 
+    def _strip_stress(self, phones: list) -> list:
+        """Remove stress markers from phones."""
+        result = []
+        for phone in phones:
+            result.append(phone.rstrip('012'))
+        return result
+
+
     def _vowels_match(self, v1: str, v2: str) -> bool:
         """Check if two vowels are in the same group."""
         v1 = v1.rstrip('012')
@@ -110,12 +118,17 @@ class ParodyWordSuggestionTool(Tool):
                 # 1. Rhyme score (most important - 60%)
                 rhyme_score = 0.0
                 if word_vowel and target_vowel:
-                    # Check vowel match
+                    # Check if the vowels are similar
                     if self._vowels_match(word_vowel, target_vowel):
-                        # Perfect rhyme if endings match too
-                        if word_end == target_end:
+                        # Check if endings match (ignoring stress numbers)
+                        word_end_clean = self._strip_stress(word_end)
+                        target_end_clean = self._strip_stress(target_end)
+                    
+                        if word_end_clean == target_end_clean:
                             rhyme_score = 1.0
-                        # Partial rhyme if just the vowel matches
+                            # Extra boost for exact match
+                            if len(word) == len(target):
+                                rhyme_score = 1.2
                         else:
                             rhyme_score = 0.6
             
@@ -127,7 +140,7 @@ class ParodyWordSuggestionTool(Tool):
                 # 3. Overall similarity (15%)
                 string_similarity = SequenceMatcher(None, target, word).ratio()
             
-                # Combined score
+                # Combined score with phonetic similarity bonus
                 similarity = (rhyme_score * 0.6) + (syllable_score * 0.25) + (string_similarity * 0.15)
             
                 if similarity >= min_similarity:
@@ -141,7 +154,12 @@ class ParodyWordSuggestionTool(Tool):
                         "syllables": word_syl,
                         "phones": word_phones,
                         "last_vowel": word_vowel,
-                        "ending": " ".join(word_end) if word_end else ""
+                        "ending": " ".join(word_end) if word_end else "",
+                        "debug_info": {
+                            "word_end_clean": word_end_clean,
+                            "target_end_clean": target_end_clean,
+                            "exact_match": word_end_clean == target_end_clean
+                        }
                     })
     
         # Sort by similarity score descending