Update app.py

app.py

@@ -1,11 +1,8 @@
-import os
 import gradio as gr
-from transformers import pipeline
 import spacy
 import subprocess
 import nltk
 from nltk.corpus import wordnet
-from gensim import downloader as api
 
 # Ensure necessary NLTK data is downloaded
 nltk.download('wordnet')
@@ -18,24 +15,6 @@ except OSError:
     subprocess.run(["python", "-m", "spacy", "download", "en_core_web_sm"])
     nlp = spacy.load("en_core_web_sm")
 
-# Load a smaller Word2Vec model from Gensim's pre-trained models
-word_vectors = api.load("glove-wiki-gigaword-50")
-
-# Load the English AI detection pipeline using the Hello-SimpleAI model
-pipeline_en = pipeline(task="text-classification", model="Hello-SimpleAI/chatgpt-detector-roberta")
-
-# AI detection function using the Hello-SimpleAI/chatgpt-detector-roberta model
-def detect_ai_generated(text):
-    res = pipeline_en(text)[0]
-    label = res['label']  # "LABEL_0" or "LABEL_1"
-    score = res['score'] * 100  # Convert probability to percentage
-    
-    # Map the model's label to human-readable label
-    human_readable_label = "AI" if label == "LABEL_1" else "Human"
-    
-    # Return formatted string with label and percentage score
-    return f"The content is {score:.2f}% {human_readable_label} Written", score
-
 # Function to get synonyms using NLTK WordNet
 def get_synonyms_nltk(word, pos):
     synsets = wordnet.synsets(word, pos=pos)
@@ -105,19 +84,16 @@ def paraphrase_and_correct(text):
     
     return final_text
 
-# Gradio interface definition
-with gr.Blocks() as interface:
+# Gradio interface for paraphrasing and text correction
+with gr.Blocks() as paraphrase_interface:
     with gr.Row():
         with gr.Column():
             text_input = gr.Textbox(lines=5, label="Input Text")
-            detect_button = gr.Button("AI Detection")
             paraphrase_button = gr.Button("Paraphrase & Correct")
         with gr.Column():
-            output_label = gr.Textbox(label="Predicted Label 🎃")
-            output_prob = gr.Textbox(label="Probability (%)")
+            output_text = gr.Textbox(label="Paraphrased Text")
 
-    detect_button.click(detect_ai_generated, inputs=text_input, outputs=[output_label, output_prob])
-    paraphrase_button.click(paraphrase_and_correct, inputs=text_input, outputs=output_label)
+    paraphrase_button.click(paraphrase_and_correct, inputs=text_input, outputs=output_text)
 
-# Launch the Gradio app
-interface.launch(debug=False)
+# Launch the Gradio app for paraphrasing and text correction
+paraphrase_interface.launch(debug=False)