Update app.py

app.py

@@ -5,6 +5,7 @@ import spacy
 import subprocess
 import nltk
 from nltk.corpus import wordnet
+from gector.gec_model import GecBERTModel
 
 # Initialize the English text classification pipeline for AI detection
 pipeline_en = pipeline(task="text-classification", model="Hello-SimpleAI/chatgpt-detector-roberta")
@@ -25,6 +26,20 @@ except OSError:
     subprocess.run(["python", "-m", "spacy", "download", "en_core_web_sm"])
     nlp = spacy.load("en_core_web_sm")
 
+# Initialize GECToR model for grammar correction
+gector_model = GecBERTModel(vocab_path='data/output_vocabulary',
+                            model_paths=['https://grammarly-nlp-data.s3.amazonaws.com/gector/roberta_1_gector.th'],
+                            is_ensemble=False)
+
+# Function to correct grammar using GECToR
+def correct_grammar_with_gector(text):
+    corrected_sentences = []
+    sentences = [text]  # If you want to split into sentences, you can implement that here
+    for sentence in sentences:
+        preds = gector_model.handle_batch([sentence])
+        corrected_sentences.append(preds[0])
+    return ' '.join(corrected_sentences)
+
 # Function to get synonyms using NLTK WordNet (Humanifier)
 def get_synonyms_nltk(word, pos):
     synsets = wordnet.synsets(word, pos=pos)
@@ -94,7 +109,7 @@ def paraphrase_and_correct(text):
     
     return final_text
 
-# Gradio app setup with two tabs
+# Gradio app setup with three tabs
 with gr.Blocks() as demo:
     with gr.Tab("AI Detection"):
         t1 = gr.Textbox(lines=5, label='Text')
@@ -112,6 +127,14 @@ with gr.Blocks() as demo:
 
         # Connect the paraphrasing function to the button
         paraphrase_button.click(paraphrase_and_correct, inputs=text_input, outputs=output_text)
+    
+    with gr.Tab("Grammar Correction"):
+        grammar_input = gr.Textbox(lines=5, label="Input Text")
+        grammar_button = gr.Button("Correct Grammar")
+        grammar_output = gr.Textbox(label="Corrected Text")
+
+        # Connect the GECToR grammar correction function to the button
+        grammar_button.click(correct_grammar_with_gector, inputs=grammar_input, outputs=grammar_output)
 
-# Launch the app with both functionalities
+# Launch the app with all functionalities
 demo.launch()