Update app.py

app.py

@@ -13,11 +13,6 @@ pipeline_en = pipeline(task="text-classification", model="Hello-SimpleAI/chatgpt
 # Initialize the spell checker
 spell = SpellChecker()
 
-# Function to predict the label and score for English text (AI Detection)
-def predict_en(text):
-    res = pipeline_en(text)[0]
-    return res['label'], res['score']
-
 # Ensure necessary NLTK data is downloaded
 nltk.download('wordnet')
 nltk.download('omw-1.4')
@@ -29,6 +24,11 @@ except OSError:
     subprocess.run(["python", "-m", "spacy", "download", "en_core_web_sm"])
     nlp = spacy.load("en_core_web_sm")
 
+# Function to predict the label and score for English text (AI Detection)
+def predict_en(text):
+    res = pipeline_en(text)[0]
+    return res['label'], res['score']
+
 # Function to get synonyms using NLTK WordNet
 def get_synonyms_nltk(word, pos):
     synsets = wordnet.synsets(word, pos=pos)
@@ -169,10 +169,42 @@ def correct_spelling(text):
         corrected_words.append(corrected_word)
     return ' '.join(corrected_words)
 
+# Function to rephrase text and replace words with synonyms while maintaining form
+def rephrase_with_synonyms(text):
+    doc = nlp(text)
+    rephrased_text = []
+
+    for token in doc:
+        pos_tag = None
+        if token.pos_ == "NOUN":
+            pos_tag = wordnet.NOUN
+        elif token.pos_ == "VERB":
+            pos_tag = wordnet.VERB
+        elif token.pos_ == "ADJ":
+            pos_tag = wordnet.ADJ
+        elif token.pos_ == "ADV":
+            pos_tag = wordnet.ADV
+        
+        if pos_tag:
+            synonym = get_synonym(token.text, pos_tag)
+            if token.pos_ == "VERB":
+                if token.morph.get("Tense") == "Past":
+                    synonym = synonym + 'ed'
+                elif token.tag_ == "VBG":  # Present participle
+                    synonym = synonym + 'ing'
+                elif token.tag_ == "VBZ":  # Third-person singular present
+                    synonym = synonym + 's'
+            elif token.pos_ == "NOUN" and token.tag_ == "NNS":  # Plural nouns
+                synonym += 's' if not synonym.endswith('s') else ""
+            rephrased_text.append(synonym)
+        else:
+            rephrased_text.append(token.text)
+
+    return ' '.join(rephrased_text)
+
 # Function to paraphrase and correct grammar
 def paraphrase_and_correct(text):
-    # Capitalize first to ensure proper noun capitalization
-    paraphrased_text = capitalize_sentences_and_nouns(text)  
+    paraphrased_text = capitalize_sentences_and_nouns(text)  # Capitalize first to ensure proper noun capitalization
     
     # Apply grammatical corrections
     paraphrased_text = correct_article_errors(paraphrased_text)
@@ -180,20 +212,14 @@ def paraphrase_and_correct(text):
     paraphrased_text = correct_tense_errors(paraphrased_text)
     paraphrased_text = correct_double_negatives(paraphrased_text)
     paraphrased_text = ensure_subject_verb_agreement(paraphrased_text)
-
-    # Replace words with synonyms while maintaining verb form
-    doc = nlp(paraphrased_text)
-    final_text = []
-    for token in doc:
-        if token.pos_ in {"VERB", "NOUN", "ADJ", "ADV"}:
-            final_text.append(replace_with_synonym(token))
-        else:
-            final_text.append(token.text)
+    
+    # Rephrase with synonyms while maintaining grammatical forms
+    paraphrased_text = rephrase_with_synonyms(paraphrased_text)
     
     # Correct spelling errors
-    final_text = correct_spelling(' '.join(final_text))
+    paraphrased_text = correct_spelling(paraphrased_text)
     
-    return final_text
+    return paraphrased_text
 
 # Gradio app setup with two tabs
 with gr.Blocks() as demo:
@@ -209,10 +235,10 @@ with gr.Blocks() as demo:
     with gr.Tab("Humanifier"):
         text_input = gr.Textbox(lines=5, label="Input Text")
         paraphrase_button = gr.Button("Paraphrase & Correct")
-        output_text = gr.Textbox(label="Paraphrased Text")
+        result_output = gr.Textbox(lines=10, label="Humanified Text")
 
         # Connect the paraphrasing function to the button
-        paraphrase_button.click(paraphrase_and_correct, inputs=text_input, outputs=output_text)
+        paraphrase_button.click(paraphrase_and_correct, inputs=[text_input], outputs=[result_output])
 
-# Launch the app with the remaining functionalities
+# Launch the app
 demo.launch()