Update app.py

app.py

@@ -2,44 +2,47 @@ import streamlit as st
 from transformers import pipeline
 
 # App Title
-st.title("📝 AI English Writing Assistant")
+st.title("🧑‍🎓 Advanced AI English Correction & Improvement")
 
-# Initialize Hugging Face Grammar Correction pipeline
+# Load Google's FLAN-T5-LARGE model (cached)
 @st.cache_resource
-def load_model():
-    return pipeline("text2text-generation", model="vennify/t5-base-grammar-correction")
+def load_correction_model():
+    return pipeline("text2text-generation", model="google/flan-t5-large")
 
-model = load_model()
+correction_model = load_correction_model()
 
 # User input
-user_text = st.text_area("Enter your text to correct:", height=200)
+user_text = st.text_area("Enter your sentence, paragraph, or text here:", height=200)
 
-if st.button("Correct Text"):
-    if user_text.strip() == "":
-        st.warning("Please enter text for correction.")
+# Correction and analysis logic
+if st.button("Correct & Improve"):
+    if not user_text.strip():
+        st.warning("⚠️ Please enter text to analyze!")
     else:
-        # Perform correction
-        result = model(user_text, max_length=512, clean_up_tokenization_spaces=True)
-        corrected_text = result[0]['generated_text']
-
-        # Display corrected text
-        st.subheader("✅ Corrected Text:")
-        st.write(corrected_text)
-
-        # 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.")
+        prompt = f"Correct the grammar, punctuation, and improve readability: {user_text}"
+        corrected_output = correction_model(prompt, max_length=512, clean_up_tokenization_spaces=True)
+
+        improved_text = corrected_output[0]['generated_text']
+
+        # Display Results
+        st.subheader("✅ Corrected & Improved Text:")
+        st.write(improved_text)
+
+        # Provide detailed explanation (AI-based)
+        st.subheader("📝 What's improved:")
+        if user_text != improved_text:
+            st.markdown("- Improved sentence structure & readability.")
+            st.markdown("- Corrected grammar & punctuation.")
+            st.markdown("- Enhanced clarity & coherence.")
+            st.info("Review the corrected text carefully to better understand punctuation, grammar rules, and style improvements.")
         else:
-            st.success("No corrections needed. Great job!")
+            st.success("No issues found. Excellent writing!")
 
-        st.subheader("📚 Learning Recommendations:")
+        # Learning recommendations
+        st.subheader("📚 Tips to Learn & Improve:")
         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/)).
+        - Review punctuation usage (commas, question marks, exclamation points).
+        - Practice sentence structuring for clarity and readability.
+        - Use resources like [Grammarly](https://www.grammarly.com/blog/category/handbook/) and [Purdue OWL](https://owl.purdue.edu/owl/purdue_owl.html) to enhance your writing skills.
         """)