Update app.py

app.py

@@ -1,48 +1,45 @@
 import streamlit as st
-import language_tool_python
+from transformers import pipeline
 
-# App title
-st.title("English Writing Correction Assistant ✍️")
+# App Title
+st.title("📝 AI English Writing Assistant")
 
-# Initialize LanguageTool for English
-tool = language_tool_python.LanguageTool('en-US')
+# Initialize Hugging Face Grammar Correction pipeline
+@st.cache_resource
+def load_model():
+    return pipeline("text2text-generation", model="vennify/t5-base-grammar-correction")
 
-# Text input from user
-user_text = st.text_area("Enter your text here:", height=200)
+model = load_model()
 
-# Analyze button
-if st.button("Correct and Analyze"):
+# User input
+user_text = st.text_area("Enter your text to correct:", height=200)
+
+if st.button("Correct Text"):
     if user_text.strip() == "":
-        st.warning("Please enter some text to analyze.")
+        st.warning("Please enter text for correction.")
     else:
-        # Get grammar corrections
-        matches = tool.check(user_text)
-        corrected_text = language_tool_python.utils.correct(user_text, matches)
+        # Perform correction
+        result = model(user_text, max_length=512, clean_up_tokenization_spaces=True)
+        corrected_text = result[0]['generated_text']
 
-        # Display Corrected Text
+        # Display corrected text
         st.subheader("✅ Corrected Text:")
         st.write(corrected_text)
 
-        # Detailed feedback on issues
-        st.subheader("📝 Improvement Suggestions:")
-        if matches:
-            for idx, match in enumerate(matches, start=1):
-                st.markdown(f"**{idx}. Issue:** {match.ruleDescription}")
-                st.markdown(f"**Context:** ...{match.context}...")
-                if match.replacements:
-                    st.markdown(f"**Suggested Corrections:** {', '.join(match.replacements)}")
-                else:
-                    st.markdown("**Suggested Corrections:** None provided.")
-                st.markdown("---")
+        # Simple explanation and suggestions (basic demo)
+        st.subheader("💡 Improvement Suggestions:")
+        if user_text != corrected_text:
+            st.info("Review differences carefully to learn grammar usage.")
+            st.markdown("- Compare your original sentence to the corrected sentence.")
+            st.markdown("- Note the punctuation, tense, or grammar corrections.")
+            st.markdown("- Practice similar sentence constructions.")
         else:
-            st.success("No grammatical or stylistic issues found!")
+            st.success("No corrections needed. Great job!")
 
-        # Learning suggestions
         st.subheader("📚 Learning Recommendations:")
-        if matches:
-            st.info("Consider reviewing:")
-            suggestions = set(match.ruleIssueType for match in matches if match.ruleIssueType)
-            for s in suggestions:
-                st.markdown(f"- {s.replace('_', ' ').title()}")
-        else:
-            st.write("Your writing looks great! Keep it up! 🎉")
+        st.markdown("""
+        - Review grammar rules related to sentence structure, verb tense consistency, and punctuation.
+        - Practice regularly with corrected examples.
+        - Utilize grammar learning resources online ([Grammarly](https://www.grammarly.com/blog/category/handbook/), [EnglishGrammar.org](https://www.englishgrammar.org/)).
+        """)
+