Upload tool

tool.py

@@ -1,16 +1,16 @@
 from smolagents.tools import Tool
-import string
 import pronouncing
 import json
+import string
 
 class ParodyWordSuggestionTool(Tool):
     name = "parody_word_suggester"
     description = """Suggests phonetically similar funny words using CMU dictionary pronunciations.
     Returns similar words from a curated list of funny words, along with their similarity scores."""
-    inputs = {'target': {'type': 'string', 'description': 'The word you want to find funny alternatives for'}, 'word_list': {'type': 'string', 'description': 'JSON string of words list, e.g. \'["word1", "word2"]\''}, 'min_similarity': {'type': 'string', 'description': 'Minimum similarity threshold (0.0-1.0)', 'nullable': True}}
+    inputs = {'target': {'type': 'string', 'description': 'The word you want to find funny alternatives for'}, 'word_list_str': {'type': 'string', 'description': 'Comma-separated list of words to check for similarity'}, 'min_similarity': {'type': 'string', 'description': 'Minimum similarity threshold (0.0-1.0)', 'nullable': True}}
     output_type = "string"
 
-    def forward(self, target: str, word_list: str, min_similarity: str = "0.7") -> str:
+    def forward(self, target: str, word_list_str: str, min_similarity: str = "0.7") -> str:
         """Get phonetically similar funny word suggestions."""
         import pronouncing
         import string
@@ -20,19 +20,9 @@ class ParodyWordSuggestionTool(Tool):
         min_similarity = float(min_similarity)
         suggestions = []
     
-        # Parse word list from JSON string
-        try:
-            words = json.loads(word_list)
-            if not isinstance(words, list):
-                return json.dumps({
-                    "error": "word_list must be a JSON string representing a list of words",
-                    "suggestions": []
-                }, indent=2)
-        except json.JSONDecodeError:
-            return json.dumps({
-                "error": "Invalid JSON string for word_list",
-                "suggestions": []
-            }, indent=2)
+        # Parse comma-separated word list
+        word = ""
+        words = [word.strip() for word in word_list_str.split(',')]
         
         # Get target pronunciation
         target_phones = pronouncing.phones_for_word(target)