Update app.py

app.py

@@ -24,7 +24,7 @@ nltk.download('punkt_tab')
 # Initialize stopwords
 stop_words = set(stopwords.words("english"))
 
-# Exclusion lists
+# 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'}
 
@@ -34,7 +34,7 @@ pipeline_en = pipeline(task="text-classification", model="Hello-SimpleAI/chatgpt
 # Initialize the spell checker
 spell = SpellChecker()
 
-# Load the SpaCy model
+# Ensure the SpaCy model is installed
 try:
     nlp = spacy.load("en_core_web_sm")
 except OSError:
@@ -213,19 +213,18 @@ def paraphrase_and_correct(text):
 
 # Gradio app setup
 with gr.Blocks() as demo:
-    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")
+    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')
 
-        submit_btn.click(paraphrase_and_correct, inputs=text_input, outputs=text_output)
+        button1.click(fn=predict_en, inputs=t1, outputs=[label1, score1])
 
-    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")
+    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')
 
-        ai_submit_btn.click(predict_en, inputs=ai_input, outputs=ai_output)
+        button2.click(fn=paraphrase_and_correct, inputs=t2, outputs=result2)
 
-# Launch the Gradio app
-demo.launch()