Update app.py

app.py

@@ -16,15 +16,13 @@ import random
 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
+# Exclusion lists
 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'}
 
@@ -34,7 +32,7 @@ pipeline_en = pipeline(task="text-classification", model="Hello-SimpleAI/chatgpt
 # Initialize the spell checker
 spell = SpellChecker()
 
-# Ensure the SpaCy model is installed
+# Load the SpaCy model
 try:
     nlp = spacy.load("en_core_web_sm")
 except OSError:
@@ -213,20 +211,19 @@ def paraphrase_and_correct(text):
 
 # Gradio app setup
 with gr.Blocks() as demo:
-    with gr.Tab("AI Detection"):
-        t1 = gr.Textbox(lines=5, label='Text')
-        button1 = gr.Button("🤖 Predict!")
-        label1 = gr.Textbox(lines=1, label='Predicted Label 🎃')
-        score1 = gr.Textbox(lines=1, label='Prob')
+    with gr.Tab("Text Input"):
+        text_input = gr.TextArea(label="Input Text", placeholder="Enter text here...")
+        text_output = gr.TextArea(label="Output Text", interactive=False)
+        submit_btn = gr.Button("Submit")
 
-        button1.click(fn=predict_en, inputs=t1, outputs=[label1, score1])
+        submit_btn.click(paraphrase_and_correct, inputs=text_input, outputs=text_output)
 
-    with gr.Tab("Paraphrasing & Grammar Correction"):
-        t2 = gr.Textbox(lines=5, label='Enter text for paraphrasing and grammar correction')
-        button2 = gr.Button("🔄 Paraphrase and Correct")
-        result2 = gr.Textbox(lines=5, label='Corrected Text')
+    with gr.Tab("AI Detection"):
+        ai_input = gr.TextArea(label="Input Text for AI Detection", placeholder="Enter text here...")
+        ai_output = gr.Textbox(label="AI Detection Result", interactive=False)
+        ai_submit_btn = gr.Button("Submit for AI Detection")
 
-        button2.click(fn=paraphrase_and_correct, inputs=t2, outputs=result2)
+        ai_submit_btn.click(predict_en, inputs=ai_input, outputs=ai_output)
 
 # Launch the Gradio app
 demo.launch()