Update app.py

app.py

@@ -1,16 +1,45 @@
 import os
-import random
-import re
-import string
+import gradio as gr
+from transformers import pipeline
 import spacy
-from nltk.corpus import wordnet
+import subprocess
 import nltk
-import gradio as gr
+from nltk.corpus import wordnet
+from nltk.corpus import stopwords
+from nltk.tokenize import word_tokenize
+from spellchecker import SpellChecker
+import re
+import string
+import random
 
-# Ensure that necessary NLTK resources are downloaded
+# Download necessary NLTK data
 nltk.download('punkt')
+nltk.download('stopwords')
 nltk.download('averaged_perceptron_tagger')
+nltk.download('averaged_perceptron_tagger_eng')
 nltk.download('wordnet')
+nltk.download('omw-1.4')
+nltk.download('punkt_tab')
+
+# Initialize stopwords
+stop_words = set(stopwords.words("english"))
+
+# Words we don't want to replace
+exclude_tags = {'PRP', 'PRP$', 'MD', 'VBZ', 'VBP', 'VBD', 'VBG', 'VBN', 'TO', 'IN', 'DT', 'CC'}
+exclude_words = {'is', 'am', 'are', 'was', 'were', 'have', 'has', 'do', 'does', 'did', 'will', 'shall', 'should', 'would', 'could', 'can', 'may', 'might'}
+
+# Initialize the English text classification pipeline for AI detection
+pipeline_en = pipeline(task="text-classification", model="Hello-SimpleAI/chatgpt-detector-roberta")
+
+# Initialize the spell checker
+spell = SpellChecker()
+
+# Ensure the SpaCy model is installed
+try:
+    nlp = spacy.load("en_core_web_sm")
+except OSError:
+    subprocess.run(["python", "-m", "spacy", "download", "en_core_web_sm"])
+    nlp = spacy.load("en_core_web_sm")
 
 # Load SpaCy model
 nlp = spacy.load("en_core_web_sm")