Update app.py

app.py

@@ -145,6 +145,14 @@ def correct_spelling(text):
     return ' '.join(corrected_words)
 
 # Function to replace a word with its synonym
+def get_wordnet_synonyms(word):
+    """Get synonyms from WordNet for a given word."""
+    synonyms = set()
+    for syn in wn.synsets(word):
+        for lemma in syn.lemmas():
+            synonyms.add(lemma.name())  # Add synonym to set
+    return list(synonyms)
+
 def replace_with_synonyms(text):
     doc = nlp(text)
     replaced_words = []
@@ -152,23 +160,21 @@ def replace_with_synonyms(text):
     for index, token in enumerate(doc):
         # Replace every 2nd and 4th word
         if (index + 1) % 2 == 0 or (index + 1) % 4 == 0:
-            # Check if the token has synonyms in the dictionary
-            if token.text in synonyms_dict:
-                relevant_synonyms = synonyms_dict[token.text]
-                if relevant_synonyms:
-                    # Randomly choose a synonym from the available options
-                    synonym = random.choice(relevant_synonyms)
-                    # Add the synonym instead of the original word
-                    replaced_words.append(synonym)
-                else:
-                    replaced_words.append(token.text)  # No relevant synonym found
+            # Get synonyms from WordNet
+            relevant_synonyms = get_wordnet_synonyms(token.text)
+            if relevant_synonyms:
+                # Randomly choose a synonym from the available options
+                synonym = random.choice(relevant_synonyms)
+                # Add the synonym instead of the original word
+                replaced_words.append(synonym.replace('_', ' '))  # Replace underscores with spaces
             else:
-                replaced_words.append(token.text)  # No synonyms available
+                replaced_words.append(token.text)  # No relevant synonym found
         else:
             replaced_words.append(token.text)  # Non-replaceable tokens
 
     return ' '.join(replaced_words)
 
+
 # Main function for paraphrasing and grammar correction
 def paraphrase_and_correct(text):
      # Add synonym replacement here