Update app.py

app.py

@@ -9,10 +9,10 @@ from nltk.tokenize import sent_tokenize
 # Download punkt for sentence tokenization
 nltk.download("punkt")
 
-# Load the grammar tool
-grammar_tool = language_tool_python.LanguageToolPublicAPI('en-US')
+# Use local LanguageTool server (runs via setup.sh)
+grammar_tool = language_tool_python.LanguageTool('en-US')
 
-# Load models
+# Load summarizer and hate-speech detection model
 summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-6-6")
 toxicity_classifier = pipeline("text-classification", model="Hate-speech-CNERG/bert-base-uncased-hatexplain")
 
@@ -25,16 +25,20 @@ def extract_text(input_type, text_input, url_input):
     return text_input
 
 def check_grammar(text):
-    matches = grammar_tool.check(text)
-    return [
-        {
-            "text": match.context,
-            "error": match.message,
-            "suggestions": match.replacements,
-            "offset": match.offset,
-            "length": match.errorLength
-        } for match in matches
-    ]
+    try:
+        matches = grammar_tool.check(text)
+        return [
+            {
+                "text": match.context,
+                "error": match.message,
+                "suggestions": match.replacements,
+                "offset": match.offset,
+                "length": match.errorLength
+            } for match in matches
+        ]
+    except Exception as e:
+        print(f"[Grammar Check Error] {e}")
+        return [{"text": "", "error": "Grammar check skipped (server error)", "suggestions": [], "offset": 0, "length": 0}]
 
 def detect_sensitive_content(text):
     sentences = sent_tokenize(text)
@@ -72,7 +76,6 @@ def highlight_text(text, grammar_issues, sensitive_issues):
         highlighted = highlighted[:start] + span + highlighted[end:]
         offset_adjust += len(span) - len(error_text)
 
-    sentences = sent_tokenize(text)
     for issue in sensitive_issues:
         sentence = issue['sentence']
         highlighted = highlighted.replace(sentence, f"<span style='background-color: red'>{sentence}</span>")
@@ -116,9 +119,10 @@ def apply_changes(text, suggestions, approved_indices):
             continue
     return text
 
+# Gradio Interface
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("## 🧠 AI Blog Reviewer")
-    gr.Markdown("Analyze blog text or URL for grammar issues and sensitive content (bias, toxicity, etc.).")
+    gr.Markdown("## 🧠 AI Blog Reviewer with Grammar & Bias Detection")
+    gr.Markdown("Enter blog content or a URL. Detect grammar issues and sensitive (toxic, biased) content.")
 
     input_type = gr.Radio(["Text", "URL"], label="Input Type", value="Text")
     text_input = gr.Textbox(label="Blog Text", lines=10, visible=True)