Upload tool

tool.py

@@ -1,7 +1,7 @@
 from smolagents.tools import Tool
-import json
-import pronouncing
 import string
+import pronouncing
+import json
 
 class ParodyWordSuggestionTool(Tool):
     name = "parody_word_suggester"
@@ -257,6 +257,7 @@ class ParodyWordSuggestionTool(Tool):
         target_phone_list = target_phones.split()
         target_vowel, target_end = self._get_last_syllable(target_phone_list)
     
+        # Check each word
         # Check each word
         for word in valid_words:
             word_phones = self._get_word_phones(word, custom_phones)
@@ -270,12 +271,11 @@ class ParodyWordSuggestionTool(Tool):
                     suggestions.append({
                         "word": word,
                         "similarity": similarity_result["similarity"],
-                        "rhyme_score": similarity_result["rhyme_score"],
-                        "length_score": similarity_result["length_score"],
-                        "stress_score": similarity_result["stress_score"],
-                        "base_word_diff": similarity_result["base_word_diff"],
-                        "has_common_suffix": similarity_result["has_common_suffix"],
-                        "suffix_match": similarity_result["suffix_match"],
+                        "vowel_score": similarity_result["vowel_score"],
+                        "consonant_score": similarity_result["consonant_score"],
+                        "pattern_score": similarity_result["pattern_score"],
+                        "primary_vowels": similarity_result["primary_vowels"],
+                        "consonants": similarity_result["consonants"],
                         "phones": word_phones,
                         "last_vowel": word_vowel,
                         "ending": " ".join(word_end) if word_end else "",