Update app.py

app.py

@@ -1,11 +1,12 @@
 import gradio as gr
-from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
 import spacy
 import subprocess
 import nltk
 from nltk.corpus import wordnet
 from gensim import downloader as api
+import language_tool_python
 
 # Ensure necessary NLTK data is downloaded
 nltk.download('wordnet')
@@ -28,8 +29,8 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 tokenizer_ai = AutoTokenizer.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english")
 model_ai = AutoModelForSequenceClassification.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english").to(device)
 
-# Load Grammar Correction model (T5) from Hugging Face
-grammar_corrector = pipeline('text2text-generation', model='vennify/t5-base-grammar-correction')
+# Initialize LanguageTool
+tool = language_tool_python.LanguageTool('en-US')
 
 # AI detection function using DistilBERT
 def detect_ai_generated(text):
@@ -99,16 +100,22 @@ def paraphrase_with_spacy_nltk(text):
     
     return corrected_text
 
-# Combined function: Paraphrase -> Grammar Correction -> Capitalization
+# Function to correct grammar using LanguageTool
+def correct_grammar(text):
+    matches = tool.check(text)
+    corrected_text = language_tool_python.utils.correct(text, matches)
+    return corrected_text
+
+# Combined function: Paraphrase -> Capitalization -> Grammar Correction
 def paraphrase_and_correct(text):
     # Step 1: Paraphrase the text
     paraphrased_text = paraphrase_with_spacy_nltk(text)
     
-    # Step 2: Correct grammar using T5 model
-    corrected_text = grammar_corrector(paraphrased_text)[0]['generated_text']
+    # Step 2: Capitalize sentences and proper nouns
+    capitalized_text = capitalize_sentences_and_nouns(paraphrased_text)
     
-    # Step 3: Capitalize sentences and proper nouns
-    final_text = capitalize_sentences_and_nouns(corrected_text)
+    # Step 3: Correct grammar
+    final_text = correct_grammar(capitalized_text)
     
     return final_text