Update app.py

app.py

@@ -189,6 +189,20 @@ def correct_spelling(text):
             corrected_words.append(word)
     return ' '.join(corrected_words)
 
+
+
+
+def correct_grammar_with_gpt(text):
+    # Initialize a pipeline for text generation with a suitable model for grammar correction
+    gpt_model = pipeline("text-generation", model="gpt2")  # Replace with a grammar correction model if available
+    
+    # Generate a corrected version of the text
+    corrected_output = gpt_model(text, max_length=100)[0]['generated_text']
+    
+    return corrected_output.strip()
+
+
+
 def paraphrase_and_correct(text):
     paragraphs = text.split("\n\n")  # Split by paragraphs
 
@@ -205,10 +219,11 @@ def paraphrase_and_correct(text):
         paraphrased_text = fix_possessives(paraphrased_text)
         paraphrased_text = correct_spelling(paraphrased_text)
         paraphrased_text = fix_punctuation_spacing(paraphrased_text)
-        processed_paragraphs.append(paraphrased_text)
 
-    return "\n\n".join(processed_paragraphs)  # Reassemble the text with paragraphs
+    # Use GPT for final grammar correction
+        final_corrected_text = correct_grammar_with_gpt(paraphrased_text)
 
+        return final_corrected_text
 # Gradio app setup
 with gr.Blocks() as demo:
     with gr.Tab("AI Detection"):