Update app.py

app.py

@@ -1,113 +1,69 @@
 import streamlit as st
-import requests
+from transformers import pipeline
+from textblob import TextBlob
 
-st.title("🚀 English Grammar & Spelling Correction Assistant")
+st.title("📖 English Grammar, Spelling & Punctuation Assistant")
 
-API_KEY = "YOUR_SAPLING_API_KEY"
+# Load models once (cached for efficiency)
+@st.cache_resource
+def load_models():
+    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
 
-# User input area
-user_text = st.text_area("Enter your sentence or paragraph:", height=200)
+grammar_model, punctuation_model = load_models()
 
-def correct_text_with_sapling(text, api_key):
-    url = "https://api.sapling.ai/api/v1/edits"
-    headers = {"Content-Type": "application/json"}
-    data = {
-        "key": api_key,
-        "text": text,
-        "session_id": "streamlit-session"
-    }
+user_text = st.text_area("Enter text to correct:", height=200)
 
-    response = requests.post(url, json=data, headers=headers)
-
-    if response.status_code == 200:
-        corrections = response.json().get("edits", [])
-        corrected_text = text
-        offset = 0
-        details = []
-
-        # Apply corrections
-        for correction in corrections:
-            start = correction['start'] + offset
-            end = correction['end'] + offset
-            original = corrected_text[start:end]
-            replacement = correction['replacement']
-            
-            # Classify error type
-            error_type = correction.get('general_error_type', 'Unknown')
-
-            details.append({
-                "original": original,
-                "replacement": replacement,
-                "type": error_type,
-                "sentence": corrected_text[correction['sentence_start']:correction['sentence_end']]
-            })
-
-            corrected_text = corrected_text[:start] + replacement + corrected_text[end:]
-            offset += len(replacement) - len(original)
-
-        return corrected_text, details
-    else:
-        return None, None
-
-if st.button("Correct & Explain"):
+if st.button("Correct and Explain"):
     if user_text.strip() == "":
-        st.warning("⚠️ Please enter text to correct!")
+        st.warning("⚠️ Please enter text first.")
     else:
-        corrected_text, details = correct_text_with_sapling(user_text, API_KEY)
-
-        if corrected_text:
-            st.subheader("✅ Corrected Text:")
-            st.write(corrected_text)
-
-            grammar_issues = []
-            punctuation_issues = []
-            spelling_pluralization_issues = []
-
-            for detail in details:
-                issue_type = detail['type'].lower()
+        # Step 1: Grammar correction
+        grammar_corrected = grammar_model(user_text, max_length=512)[0]['generated_text']
 
-                if "punctuation" in issue_type:
-                    punctuation_issues.append(detail)
-                elif "spelling" in issue_type or "plural" in issue_type:
-                    spelling_pluralization_issues.append(detail)
-                else:
-                    grammar_issues.append(detail)
+        # Step 2: Punctuation correction
+        punctuation_corrected = punctuation_model(grammar_corrected, max_length=512)[0]['generated_text']
 
-            # Display explanations
-            st.subheader("📝 Detailed Explanations:")
+        # Step 3: Spelling and pluralization correction using TextBlob
+        blob = TextBlob(punctuation_corrected)
+        spelling_corrected = str(blob.correct())
 
-            if grammar_issues:
-                st.markdown("**📌 Grammar Issues:**")
-                for issue in grammar_issues:
-                    st.markdown(f"- `{issue['original']}` ➡️ `{issue['replacement']}`")
-                    st.markdown(f"  **Context:** {issue['sentence']}")
+        # Display final corrected text
+        st.subheader("✅ Fully Corrected Text:")
+        st.write(spelling_corrected)
 
-            if punctuation_issues:
-                st.markdown("**📌 Punctuation Issues:**")
-                for issue in punctuation_issues:
-                    st.markdown(f"- `{issue['original']}` ➡️ `{issue['replacement']}`")
-                    st.markdown(f"  **Context:** {issue['sentence']}")
+        # Provide clear explanations of corrections
+        st.subheader("📝 Detailed Explanations:")
 
-            if spelling_pluralization_issues:
-                st.markdown("**📌 Spelling/Pluralization Issues:**")
-                for issue in spelling_pluralization_issues:
-                    st.markdown(f"- `{issue['original']}` ➡️ `{issue['replacement']}`")
-                    st.markdown(f"  **Context:** {issue['sentence']}")
+        if user_text != spelling_corrected:
+            if user_text != grammar_corrected:
+                st.markdown("**📌 Grammar Corrections Applied:**")
+                st.write(grammar_corrected)
+                st.markdown("---")
 
-            if not (grammar_issues or punctuation_issues or spelling_pluralization_issues):
-                st.success("No issues found. Excellent job!")
+            if grammar_corrected != punctuation_corrected:
+                st.markdown("**📌 Punctuation Corrections Applied:**")
+                st.write(punctuation_corrected)
+                st.markdown("---")
 
-            # Learning Suggestions
-            st.subheader("📚 Learning Suggestions:")
-            st.markdown("""
-            - **Grammar**: Review sentence structure, verb tenses, subject-verb agreement.
-            - **Punctuation**: Focus on proper comma, question mark, and exclamation mark usage.
-            - **Spelling & Pluralization**: Pay attention to correct spelling and plural forms of nouns.
-            
-            Recommended resources:
-            - [EnglishGrammar.org](https://www.englishgrammar.org/)
-            - [Grammarly Blog](https://www.grammarly.com/blog/category/handbook/)
-            """)
+            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.error("Error with API connection. Check your API key or internet connection.")
+            st.success("🎉 Your text had no detectable errors!")
+
+        # Learning Recommendations
+        st.subheader("📚 Recommendations to Improve Your Writing:")
+        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.
+
+        **Recommended resources:**
+        - [Grammarly Handbook](https://www.grammarly.com/blog/category/handbook/)
+        - [English Grammar Lessons](https://www.englishgrammar.org/)
+        """)