Upload tool

requirements.txt

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

tool.py

@@ -1,7 +1,7 @@
 from smolagents.tools import Tool
-import string
 import json
 import pronouncing
+import string
 
 class ParodyWordSuggestionTool(Tool):
     name = "parody_word_suggester"
@@ -37,24 +37,42 @@ class ParodyWordSuggestionTool(Tool):
                 "suggestions": []
             }, indent=2)
         
-        target_phones = target_phones[0]  # Change: Don't split here
+        target_phones = target_phones[0]
+        target_phone_list = target_phones.split()
     
         # 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]  # Change: Don't split this either
+                word_phones = phones[0]
+                word_phone_list = word_phones.split()
+            
+                # Calculate position-aware similarity
+                matches = 0
+                length = max(len(word_phone_list), len(target_phone_list))
+            
+                # 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
             
-                # Calculate similarity using overlap (modifying to work with full phone strings)
-                overlap = len(set(phones.split()) & set(target_phones.split()))
-                total = min(len(phones.split()), len(target_phones.split()))
-                similarity = overlap / total if total > 0 else 0.0
+                similarity = matches / length if length > 0 else 0.0
             
                 if similarity >= min_similarity:
                     suggestions.append({
                         "word": word,
                         "similarity": round(similarity, 3),
-                        "syllables": pronouncing.syllable_count(phones)  # This should work now as phones is a string
+                        "syllables": pronouncing.syllable_count(word_phones),
+                        "phones": word_phones,
                     })
     
         # Sort by similarity score descending
@@ -63,6 +81,7 @@ class ParodyWordSuggestionTool(Tool):
         result = {
             "target": target,
             "target_syllables": pronouncing.syllable_count(target_phones),
+            "target_phones": target_phones,
             "suggestions": suggestions
         }