Update app.py

app.py

@@ -49,6 +49,29 @@ def get_synonyms_nltk(word, pos):
         return [lemma.name() for lemma in lemmas]
     return []
 
+# Function to check and correct tenses and verbs using spaCy
+def check_tense_and_correct(text):
+    doc = nlp(text)
+    corrected_text = []
+    
+    for token in doc:
+        # Checking for verbs and their tense
+        if token.pos_ == 'VERB':
+            tense = token.tag_  # Get the specific tense tag (e.g., VBZ, VBD, VBG, etc.)
+            
+            if tense == 'VBZ':  # 3rd person singular present
+                corrected_text.append(token.lemma_)  # Replace with base form (example: goes -> go)
+            elif tense == 'VBD':  # Past tense
+                corrected_text.append(token.text)  # Keep past tense as is
+            elif tense == 'VBG':  # Gerund/Present participle
+                corrected_text.append(token.text)  # Keep it unchanged for now
+            else:
+                corrected_text.append(token.text)  # For other cases, append the word as is
+        else:
+            corrected_text.append(token.text)
+    
+    return ' '.join(corrected_text)
+
 # Function to capitalize the first letter of sentences and proper nouns
 def capitalize_sentences_and_nouns(text):
     doc = nlp(text)
@@ -105,15 +128,18 @@ def correct_grammar(text):
     result = parser.parse(text)
     return result['result']
 
-# Combined function: Paraphrase -> Capitalization -> Grammar Correction
+# Combined function: Paraphrase -> Tense Check -> Capitalization -> Grammar Correction
 def paraphrase_and_correct(text):
     # Step 1: Paraphrase the text
     paraphrased_text = paraphrase_with_spacy_nltk(text)
     
-    # Step 2: Capitalize sentences and proper nouns
-    capitalized_text = capitalize_sentences_and_nouns(paraphrased_text)
+    # Step 2: Check tense and verbs, and attempt correction
+    tense_checked_text = check_tense_and_correct(paraphrased_text)
+    
+    # Step 3: Capitalize sentences and proper nouns
+    capitalized_text = capitalize_sentences_and_nouns(tense_checked_text)
     
-    # Step 3: Correct grammar
+    # Step 4: Correct grammar
     final_text = correct_grammar(capitalized_text)
     
     return final_text