Update app.py

app.py

@@ -2,68 +2,57 @@ import streamlit as st
 from transformers import pipeline
 from textblob import TextBlob
 
-st.title("📖 English Grammar, Spelling & Punctuation Assistant")
+st.title("📖 AI Grammar & Spelling Correction")
 
-# Load models once (cached for efficiency)
+# Load the grammar correction model
 @st.cache_resource
-def load_models():
+def load_model():
     grammar_model = pipeline('text2text-generation', model='prithivida/grammar_error_correcter_v1')
-    punctuation_model = pipeline('text2text-generation', model='oliverguhr/fullstop-punctuation-multilang-large')
-    return grammar_model, punctuation_model
+    return grammar_model
 
-grammar_model, punctuation_model = load_models()
+grammar_model = load_model()
 
-user_text = st.text_area("Enter text to correct:", height=200)
+user_text = st.text_area("Enter text for correction:", height=200)
 
-if st.button("Correct and Explain"):
-    if user_text.strip() == "":
-        st.warning("⚠️ Please enter text first.")
+if st.button("Correct & Explain"):
+    if not user_text.strip():
+        st.warning("⚠️ Please enter some text.")
     else:
-        # Step 1: Grammar correction
-        grammar_corrected = grammar_model(user_text, max_length=512)[0]['generated_text']
-
-        # Step 2: Punctuation correction
-        punctuation_corrected = punctuation_model(grammar_corrected, max_length=512)[0]['generated_text']
-
-        # Step 3: Spelling and pluralization correction using TextBlob
-        blob = TextBlob(punctuation_corrected)
-        spelling_corrected = str(blob.correct())
-
-        # Display final corrected text
-        st.subheader("✅ Fully Corrected Text:")
-        st.write(spelling_corrected)
-
-        # Provide clear explanations of corrections
-        st.subheader("📝 Detailed Explanations:")
-
-        if user_text != spelling_corrected:
-            if user_text != grammar_corrected:
-                st.markdown("**📌 Grammar Corrections Applied:**")
-                st.write(grammar_corrected)
-                st.markdown("---")
-
-            if grammar_corrected != punctuation_corrected:
-                st.markdown("**📌 Punctuation Corrections Applied:**")
-                st.write(punctuation_corrected)
-                st.markdown("---")
-
-            if punctuation_corrected != spelling_corrected:
-                st.markdown("**📌 Spelling & Pluralization Corrections Applied:**")
-                st.write(spelling_corrected)
-                st.markdown("---")
-
-            st.info("Review each step carefully to understand the corrections made.")
-        else:
-            st.success("🎉 Your text had no detectable errors!")
-
-        # Learning Recommendations
-        st.subheader("📚 Recommendations to Improve Your Writing:")
+        # Grammar & punctuation correction
+        corrected_output = grammar_model(user_text, max_length=512)
+        corrected_text = corrected_output[0]['generated_text']
+
+        # Spelling & pluralization correction using TextBlob
+        blob = TextBlob(corrected_text)
+        final_text = str(blob.correct())
+
+        # Display corrected text
+        st.subheader("✅ Corrected Text:")
+        st.write(final_text)
+
+        # Explanation of corrections clearly
+        st.subheader("📝 Explanation of Corrections:")
+        
+        if user_text != corrected_text:
+            st.markdown("**Grammar & Punctuation Corrections Applied:**")
+            st.markdown(f"- `{corrected_text}`")
+            st.markdown("---")
+        
+        if corrected_text != final_text:
+            st.markdown("**Spelling & Pluralization Corrections Applied:**")
+            st.markdown(f"- `{final_text}`")
+            st.markdown("---")
+
+        if user_text == final_text:
+            st.success("No corrections needed, your text was already correct!")
+
+        st.subheader("📚 Recommendations to Improve:")
         st.markdown("""
-        - **Grammar**: Review rules related to verb tenses, articles, and sentence structures.
-        - **Punctuation**: Focus on correct usage of commas, periods, question marks, and exclamation points.
-        - **Spelling & Pluralization**: Regularly check spelling and plural forms, especially for irregular words.
+        - Regularly practice grammar, punctuation, and spelling.
+        - Review differences carefully to understand corrections.
+        - Use tools like Grammarly for continuous learning.
 
-        **Recommended resources:**
+        **Useful Links:**
         - [Grammarly Handbook](https://www.grammarly.com/blog/category/handbook/)
-        - [English Grammar Lessons](https://www.englishgrammar.org/)
-        """)
+        - [EnglishGrammar.org](https://www.englishgrammar.org/)
+        """)