Update app.py

app.py

@@ -25,14 +25,14 @@ word_vectors = api.load("glove-wiki-gigaword-50")
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 # Load AI Detector model and tokenizer from Hugging Face (DistilBERT)
-tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english")
-model = AutoModelForSequenceClassification.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english").to(device)
+tokenizer_ai = AutoTokenizer.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english")
+model_ai = AutoModelForSequenceClassification.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english").to(device)
 
 # AI detection function using DistilBERT
 def detect_ai_generated(text):
-    inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=512).to(device)
+    inputs = tokenizer_ai(text, return_tensors="pt", truncation=True, max_length=512).to(device)
     with torch.no_grad():
-        outputs = model(**inputs)
+        outputs = model_ai(**inputs)
         probabilities = torch.softmax(outputs.logits, dim=1)
     ai_probability = probabilities[0][1].item()  # Probability of being AI-generated
     return f"AI-Generated Content Probability: {ai_probability:.2f}%"
@@ -96,6 +96,16 @@ def paraphrase_with_spacy_nltk(text):
     
     return corrected_text
 
+# Combined function: Paraphrase -> Capitalization
+def paraphrase_and_correct(text):
+    # Step 1: Paraphrase the text
+    paraphrased_text = paraphrase_with_spacy_nltk(text)
+    
+    # Step 2: Capitalize sentences and proper nouns
+    final_text = capitalize_sentences_and_nouns(paraphrased_text)
+    
+    return final_text
+
 # Gradio interface definition
 with gr.Blocks() as interface:
     with gr.Row():
@@ -107,7 +117,7 @@ with gr.Blocks() as interface:
             output_text = gr.Textbox(label="Output")
 
     detect_button.click(detect_ai_generated, inputs=text_input, outputs=output_text)
-    paraphrase_button.click(paraphrase_with_spacy_nltk, inputs=text_input, outputs=output_text)
+    paraphrase_button.click(paraphrase_and_correct, inputs=text_input, outputs=output_text)
 
 # Launch the Gradio app
 interface.launch(debug=False)