Update app.py

app.py

@@ -68,6 +68,20 @@ def force_first_letter_capital(text):
     capitalized_sentences = [sentence[0].capitalize() + sentence[1:] if sentence else "" for sentence in sentences]
     return ". ".join(capitalized_sentences)
 
+# Function to handle possessive 's and retain original meaning
+def handle_possessives(text):
+    doc = nlp(text)
+    corrected_text = []
+
+    for token in doc:
+        # If token is a possessive form (e.g., 'Earth's'), retain its original form
+        if token.text.endswith("'s") or token.text == "'s":
+            corrected_text.append(token.text)  # Keep it as is, even if a synonym is found
+        else:
+            corrected_text.append(token.text)
+
+    return ' '.join(corrected_text)
+
 # Function to correct tense errors in a sentence
 def correct_tense_errors(text):
     doc = nlp(text)
@@ -205,7 +219,7 @@ def rephrase_with_synonyms(text):
                     elif token.tag_ == "VBZ":  # Third-person singular present
                         synonym = synonym + 's'
                 elif token.pos_ == "NOUN" and token.tag_ == "NNS":  # Plural nouns
-                    synonym += 's' if not synonym.endswith('s') else ""
+                    synonym += 's' if not synonym ends with 's' else ""
                 rephrased_text.append(synonym)
             else:
                 rephrased_text.append(token.text)
@@ -214,10 +228,26 @@ def rephrase_with_synonyms(text):
 
     return ' '.join(rephrased_text)
 
-# Function to paraphrase and correct grammar with enhanced accuracy
+# Retain the structure of the input text (headings, paragraphs, line breaks)
+def retain_structure(text):
+    lines = text.split("\n")
+    formatted_lines = []
+    
+    for line in lines:
+        if line.strip().isupper():  # Heading if all caps
+            formatted_lines.append(f"# {line.strip()}")  # Treat it as a heading
+        else:
+            formatted_lines.append(line)  # Otherwise, it's a paragraph or normal text
+    
+    return "\n".join(formatted_lines)
+
+# Function to paraphrase and correct grammar with enhanced accuracy and retain structure
 def paraphrase_and_correct(text):
+    # Retain the structure (headings, paragraphs, line breaks)
+    structured_text = retain_structure(text)
+    
     # Remove meaningless or redundant words first
-    cleaned_text = remove_redundant_words(text)
+    cleaned_text = remove_redundant_words(structured_text)
     
     # Capitalize sentences and nouns
     paraphrased_text = capitalize_sentences_and_nouns(cleaned_text)
@@ -225,6 +255,9 @@ def paraphrase_and_correct(text):
     # Ensure first letter of each sentence is capitalized
     paraphrased_text = force_first_letter_capital(paraphrased_text)
     
+    # Handle possessives properly
+    paraphrased_text = handle_possessives(paraphrased_text)
+    
     # Apply grammatical corrections
     paraphrased_text = correct_article_errors(paraphrased_text)
     paraphrased_text = correct_singular_plural_errors(paraphrased_text)