Update app.py

app.py

@@ -14,18 +14,27 @@ def load_model():
 
 tokenizer, model = load_model()
 
-# Step 1: Fix typos using pyspellchecker
+# Step 0: Preprocess the input
+def preprocess_input(text):
+    # Remove special characters like '#' from the end
+    cleaned = re.sub(r'[^\w\s]$', '', text.strip())
+
+    # Ensure sentence ends with a period if not already
+    if not cleaned.endswith('.'):
+        cleaned += '.'
+
+    return cleaned
+
+# Step 1: Spelling correction
 def correct_spelling(text):
     spell = SpellChecker()
     words = re.findall(r'\b\w+\b|\S', text)
     corrected_words = []
 
     for word in words:
-        # Remove non-alphanumeric characters for spellcheck
         clean_word = re.sub(r'[^\w\s]', '', word)
         if clean_word.isalpha():
             corrected_word = spell.correction(clean_word.lower()) or clean_word
-            # Restore punctuation
             trailing = ''.join(re.findall(r'[^\w\s]', word))
             corrected_words.append(corrected_word + trailing)
         else:
@@ -33,7 +42,7 @@ def correct_spelling(text):
 
     return ' '.join(corrected_words)
 
-# Step 2: Grammar Correction using the model
+# Step 2: Grammar correction
 def correct_grammar(text):
     input_text = "gec: " + text
     input_ids = tokenizer.encode(input_text, return_tensors="pt", max_length=512, truncation=True)
@@ -41,31 +50,38 @@ def correct_grammar(text):
     corrected = tokenizer.decode(outputs[0], skip_special_tokens=True)
     return corrected
 
-# UI
+# Streamlit UI
 st.set_page_config(page_title="Grammar & Spelling Assistant", page_icon="🧠")
 st.title("🧠 Grammar & Spelling Correction Assistant")
-st.write("Fixes your typos and grammar without changing your meaning.")
+st.write("Fixes grammar and spelling errors without changing your original meaning.")
 
-user_input = st.text_area("✍️ Enter your sentence below:", height=150)
+user_input = st.text_area("✍️ Enter your sentence:", height=150)
 
 if st.button("Correct & Explain"):
     if not user_input.strip():
         st.warning("Please enter a sentence.")
     else:
-        # Step 1: Spelling correction
-        spelling_fixed = correct_spelling(user_input)
+        # Step 0: Preprocess
+        preprocessed = preprocess_input(user_input)
+
+        # Step 1: Spell check
+        spelling_fixed = correct_spelling(preprocessed)
 
         # Step 2: Grammar correction
         final_output = correct_grammar(spelling_fixed)
 
+        # Output
         st.markdown("### ✅ Final Correction:")
         st.success(final_output)
 
-        st.markdown("### 🔍 Explanation of Changes:")
+        st.markdown("### 🔍 Explanation:")
         st.info(f"""
 **Original Sentence:**  
 {user_input}
 
+**After Preprocessing (remove #, enforce period):**  
+{preprocessed}
+
 **After Spelling Correction:**  
 {spelling_fixed}
 
@@ -73,7 +89,8 @@ if st.button("Correct & Explain"):
 {final_output}
 
 **Explanation:**  
-- Typos like `ober` → `over`, `dogz#` → `dogs` were corrected.  
-- Grammar and punctuation (like capitalization) were fixed.  
-- Your original message and word choices were preserved.
+- Special characters like `#` were removed  
+- Misspelled words like `ober` → `over`, `dogz` → `dogs` were fixed  
+- Grammar (capitalization, punctuation) was corrected  
+- No unwanted words like `#5` were added  
 """)