Update app.py

app.py

@@ -1,48 +1,113 @@
 import streamlit as st
-from transformers import pipeline
+import requests
 
-# App Title
-st.title("🧑‍🎓 Advanced AI English Correction & Improvement")
+st.title("🚀 English Grammar & Spelling Correction Assistant")
 
-# Load Google's FLAN-T5-LARGE model (cached)
-@st.cache_resource
-def load_correction_model():
-    return pipeline("text2text-generation", model="google/flan-t5-large")
+API_KEY = "YOUR_SAPLING_API_KEY"
 
-correction_model = load_correction_model()
+# User input area
+user_text = st.text_area("Enter your sentence or paragraph:", height=200)
 
-# User input
-user_text = st.text_area("Enter your sentence, paragraph, or text here:", height=200)
+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"
+    }
 
-# Correction and analysis logic
-if st.button("Correct & Improve"):
-    if not user_text.strip():
-        st.warning("⚠️ Please enter text to analyze!")
+    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:
-        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 issues found. Excellent writing!")
+        return None, None
+
+if st.button("Correct & Explain"):
+    if user_text.strip() == "":
+        st.warning("⚠️ Please enter text to correct!")
+    else:
+        corrected_text, details = correct_text_with_sapling(user_text, API_KEY)
+
+        if corrected_text:
+            st.subheader("✅ Corrected Text:")
+            st.write(corrected_text)
 
-        # Learning recommendations
-        st.subheader("📚 Tips to Learn & Improve:")
-        st.markdown("""
-        - 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.
-        """)
+            grammar_issues = []
+            punctuation_issues = []
+            spelling_pluralization_issues = []
 
+            for detail in details:
+                issue_type = detail['type'].lower()
+
+                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)
+
+            # Display explanations
+            st.subheader("📝 Detailed Explanations:")
+
+            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']}")
+
+            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']}")
+
+            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 not (grammar_issues or punctuation_issues or spelling_pluralization_issues):
+                st.success("No issues found. Excellent job!")
+
+            # 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/)
+            """)
+
+        else:
+            st.error("Error with API connection. Check your API key or internet connection.")