Upload tool

requirements.txt

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

tool.py

@@ -11,6 +11,7 @@ class ParodyWordSuggestionTool(Tool):
     RHYME_WEIGHT = 0.5
     PHONE_SEQUENCE_WEIGHT = 0.3
     LENGTH_WEIGHT = 0.2
+    PHONE_GROUPS = "M,N,NG|P,B|T,D|K,G|F,V|TH,DH|S,Z|SH,ZH|L,R|W,Y|IY,IH|UW,UH|EH,AH|AO,AA|AE,AH|AY,EY|OW,UW"
 
     def _get_word_phones(self, word, custom_phones=None):
         """Get phones for a word, checking custom dictionary first."""
@@ -24,32 +25,93 @@ class ParodyWordSuggestionTool(Tool):
 
     def _get_primary_vowel(self, phones: list) -> str:
         """Get the primary stressed vowel from phone list."""
-        v = ""
-        for phone in phones:
-            if '1' in phone and any(v in phone for v in 'AEIOU'):
-                return phone.rstrip('012')
+        phone_str = ""
+        vowel_char = ""
+    
+        for phone_str in phones:
+            if '1' in phone_str and any(vowel_char in phone_str for vowel_char in 'AEIOU'):
+                return phone_str.rstrip('012')
         return None
 
 
+    def _get_phone_similarity(self, phone1: str, phone2: str) -> float:
+        """Calculate similarity between two phones."""
+        # Initialize variables
+        p1 = ""
+        p2 = ""
+        group_str = ""
+        group = []
+    
+        # Strip stress markers
+        p1 = phone1.rstrip('012')
+        p2 = phone2.rstrip('012')
+    
+        # Exact match
+        if p1 == p2:
+            return 1.0
+        
+        # Check similarity groups
+        for group_str in self.PHONE_GROUPS.split('|'):
+            group = group_str.split(',')
+            if p1 in group and p2 in group:
+                return 0.7
+            
+        # Check broader categories
+        if self._get_phone_type(p1) == self._get_phone_type(p2):
+            return 0.3
+        
+        return 0.0
+
+
+    def _get_phone_type(self, phone: str) -> str:
+        """Get the broad category of a phone."""
+        # Strip stress markers
+        phone = phone.rstrip('012')
+        v = ""
+    
+        # Vowels
+        if any(v in phone for v in 'AEIOU'):
+            return 'vowel'
+        
+        # Initialize fixed sets for categories
+        nasals = {'M', 'N', 'NG'}
+        stops = {'P', 'B', 'T', 'D', 'K', 'G'}
+        fricatives = {'F', 'V', 'TH', 'DH', 'S', 'Z', 'SH', 'ZH'}
+        liquids = {'L', 'R'}
+        glides = {'W', 'Y'}
+    
+        if phone in nasals:
+            return 'nasal'
+        if phone in stops:
+            return 'stop'
+        if phone in fricatives:
+            return 'fricative'
+        if phone in liquids:
+            return 'liquid'
+        if phone in glides:
+            return 'glide'
+        
+        return 'other'
+
+
     def _calculate_phone_sequence_similarity(self, phones1: list, phones2: list) -> float:
         """Calculate similarity based on matching phones in sequence."""
-        p = ""
         if not phones1 or not phones2:
             return 0.0
     
-        # Strip stress markers for comparison
-        clean_phones1 = [p.rstrip('012') for p in phones1]
-        clean_phones2 = [p.rstrip('012') for p in phones2]
+        # Initialize variables
+        total_similarity = 0.0
+        i = 0
+        similarity = 0.0
     
-        matches = 0
-        total_comparisons = max(len(clean_phones1), len(clean_phones2))
+        comparisons = max(len(phones1), len(phones2))
     
-        # Compare phones in sequence
-        for i in range(min(len(clean_phones1), len(clean_phones2))):
-            if clean_phones1[i] == clean_phones2[i]:
-                matches += 1
-            
-        return matches / total_comparisons if total_comparisons > 0 else 0.0
+        # Compare each position
+        for i in range(min(len(phones1), len(phones2))):
+            similarity = self._get_phone_similarity(phones1[i], phones2[i])
+            total_similarity += similarity
+        
+        return total_similarity / comparisons if comparisons > 0 else 0.0
 
 
     def _calculate_length_similarity(self, phones1: list, phones2: list) -> float:
@@ -61,9 +123,14 @@ class ParodyWordSuggestionTool(Tool):
 
     def _calculate_similarity(self, word1, phones1, word2, phones2):
         """Calculate similarity based on multiple factors."""
-        # Initialize phone lists
+        # Initialize phone lists and scores
         phone_list1 = phones1.split()
         phone_list2 = phones2.split()
+        rhyme_score = 0.0
+        phone_sequence_score = 0.0
+        length_score = 0.0
+        vowel1 = None
+        vowel2 = None
     
         # 1. Rhyme score (50%) - based on primary vowel
         vowel1 = self._get_primary_vowel(phone_list1)
@@ -109,6 +176,11 @@ class ParodyWordSuggestionTool(Tool):
         suggestions = []
         valid_words = []
         invalid_words = []
+        words = []
+        target_phones = ""
+        word_phones = ""
+        word = ""
+        similarity_result = {}
     
         # Parse JSON string to list
         try: