Update app.py

app.py

@@ -146,28 +146,35 @@ def correct_spelling(text):
 
 # Function to replace a word with its synonym
 def replace_with_synonyms(text):
-    words = text.split()
+    doc = nlp(text)
     replaced_words = []
-
-    for word in words:
-        synonyms = wordnet.synsets(word)
-        if synonyms:
-            # Take the first synonym if available
-            synonym = synonyms[0].lemmas()[0].name()
-            # Replace the word with its synonym if it's different
-            if synonym.lower() != word.lower():
-                replaced_words.append(synonym.replace('_', ' '))
+    for token in doc:
+        if token.pos_ in {"NOUN", "VERB", "ADJ", "ADV"}:  # Limit to specific POS
+            synonyms = wordnet.synsets(token.text)
+            if synonyms:
+                # Select synonyms with the same part of speech
+                relevant_synonyms = [syn for syn in synonyms if syn.pos() == token.pos_.lower()]
+                if relevant_synonyms:
+                    # Randomly choose a synonym from the relevant options
+                    synonym = random.choice(relevant_synonyms).lemmas()[0].name().replace('_', ' ')
+                    if synonym.lower() != token.text.lower():  # Avoid replacing with the same word
+                        replaced_words.append(synonym)
+                    else:
+                        replaced_words.append(token.text)
+                else:
+                    replaced_words.append(token.text)  # No relevant synonym found
             else:
-                replaced_words.append(word)
+                replaced_words.append(token.text)  # No synonyms available
         else:
-            replaced_words.append(word)
+            replaced_words.append(token.text)  # Non-replaceable tokens
 
     return ' '.join(replaced_words)
 
 # Main function for paraphrasing and grammar correction
 def paraphrase_and_correct(text):
-    paraphrased_text = replace_with_synonyms(text)  # Add synonym replacement here
-    cleaned_text = remove_redundant_words(paraphrased_text)
+     # Add synonym replacement here
+    cleaned_text = remove_redundant_words(text)
+    paraphrased_text = replace_with_synonyms(cleaned_text) 
     paraphrased_text = capitalize_sentences_and_nouns(cleaned_text)
     paraphrased_text = force_first_letter_capital(paraphrased_text)
     paraphrased_text = correct_article_errors(paraphrased_text)