Update app.py

app.py

@@ -136,12 +136,12 @@ def replace_with_synonym(token):
     
     if synonyms:
         synonym = synonyms[0]
-        if token.tag_ == "VBG":  # Present participle
-            synonym += 'ing'
+        if token.tag_ == "VBG":  # Present participle (e.g., running)
+            synonym = synonym + 'ing'
         elif token.tag_ == "VBD" or token.tag_ == "VBN":  # Past tense or past participle
-            synonym += 'ed'
+            synonym = synonym + 'ed'
         elif token.tag_ == "VBZ":  # Third-person singular present
-            synonym += 's'
+            synonym = synonym + 's'
         return synonym
     return token.text
 
@@ -175,15 +175,18 @@ def correct_spelling(text):
     corrected_words = []
     for word in words:
         corrected_word = spell.correction(word)
-        corrected_words.append(corrected_word if corrected_word else word)
+        corrected_words.append(corrected_word if corrected_word else word)  # Keep original if correction is None
     return ' '.join(corrected_words)
 
 # Function to correct punctuation issues
 def correct_punctuation(text):
-    # Fix spacing before punctuation
-    text = re.sub(r'\s([?.!,";:])', r'\1', text)
-    # Add missing commas in compound sentences
-    text = re.sub(r'(\w+)\s+and\s+(\w+)', r'\1, and \2', text)
+    text = re.sub(r'\s+([?.!,";:])', r'\1', text)  # Remove space before punctuation
+    text = re.sub(r'([?.!,";:])\s+', r'\1 ', text)  # Ensure a single space after punctuation
+    return text
+
+# Function to ensure correct handling of possessive forms
+def handle_possessives(text):
+    text = re.sub(r"\b(\w+)'s\b", r"\1's", text)  # Preserve possessive forms
     return text
 
 # Function to rephrase text and replace words with their synonyms while maintaining form
@@ -207,12 +210,12 @@ def rephrase_with_synonyms(text):
             if synonyms:
                 synonym = synonyms[0]  # Just using the first synonym for simplicity
                 if token.pos_ == "VERB":
-                    if token.tag_ == "VBG":  # Present participle
-                        synonym += 'ing'
+                    if token.tag_ == "VBG":  # Present participle (e.g., running)
+                        synonym = synonym + 'ing'
                     elif token.tag_ == "VBD" or token.tag_ == "VBN":  # Past tense or past participle
-                        synonym += 'ed'
+                        synonym = synonym + 'ed'
                     elif token.tag_ == "VBZ":  # Third-person singular present
-                        synonym += 's'
+                        synonym = synonym + 's'
                 elif token.pos_ == "NOUN" and token.tag_ == "NNS":  # Plural nouns
                     synonym += 's' if not synonym.endswith('s') else ""
                 rephrased_text.append(synonym)
@@ -241,6 +244,7 @@ def paraphrase_and_correct(text):
     # Correct spelling and punctuation
     paraphrased_text = correct_spelling(paraphrased_text)
     paraphrased_text = correct_punctuation(paraphrased_text)
+    paraphrased_text = handle_possessives(paraphrased_text)  # Handle possessives
 
     # Rephrase with synonyms
     paraphrased_text = rephrase_with_synonyms(paraphrased_text)
@@ -250,13 +254,21 @@ def paraphrase_and_correct(text):
 
     return final_text
 
-# Gradio interface
+# Gradio Interface
+def process_text(input_text):
+    ai_label, ai_score = predict_en(input_text)
+    corrected_text = paraphrase_and_correct(input_text)
+    return ai_label, ai_score, corrected_text
+
+# Create Gradio interface
 iface = gr.Interface(
-    fn=paraphrase_and_correct,
-    inputs=gr.Textbox(label="Input Text", lines=5, placeholder="Enter text here..."),
-    outputs=gr.Textbox(label="Output Text", lines=5),
-    title="Text Paraphraser and Grammar Corrector",
-    description="This tool paraphrases the input text, corrects grammar, improves punctuation, and enhances overall clarity."
+    fn=process_text,
+    inputs="text",
+    outputs=["text", "number", "text"],
+    title="AI Content Detection and Grammar Correction",
+    description="Enter text to detect AI-generated content and correct grammar."
 )
 
-iface.launch()
+# Launch the Gradio app
+if __name__ == "__main__":
+    iface.launch()