Update app.py

app.py

@@ -1,39 +1,48 @@
 import gradio as gr
 from transformers import pipeline
 import requests
+import spacy
 
-# Configure models
-vocab_model = pipeline("text-classification", model="distilbert-base-uncased")  # Free public model
-grammar_api = "https://api.languagetool.org/v2/check"  # Free grammar checker
+# Load models
+nlp = spacy.load("en_core_web_sm")  # For coherence analysis
+vocab_model = pipeline("text-classification", model="textattack/bert-base-uncased-CoLA")  # Better for essays
 
 def evaluate_ielts(essay):
-    # 1. Vocabulary Assessment
+    # Vocabulary Assessment
     vocab_result = vocab_model(essay)[0]
-    vocab_score = round(vocab_result['score'] * 9, 1)
+    vocab_score = round(vocab_result['score'] * 9 * 1.2, 1)  # Adjusted multiplier
     
-    # 2. Grammar Check
-    grammar_response = requests.post(grammar_api, data={"text": essay, "language": "en-GB"})
+    # Grammar Check (LanguageTool API)
+    grammar_response = requests.post(
+        "https://api.languagetool.org/v2/check",
+        data={"text": essay, "language": "en-GB"}
+    )
     grammar_errors = len(grammar_response.json()['matches'])
-    grammar_score = 9 - min(8, grammar_errors // 2)  # Deduct 1 point per 2 errors
+    grammar_score = max(1, 9 - (grammar_errors // 1.5))  # 1.5 errors = 1 point deduction
     
-    # 3. Coherence (Simplified)
-    coherence_score = 6  # Basic placeholder
+    # Coherence Analysis
+    doc = nlp(essay)
+    transition_words = ["however", "moreover", "therefore", "consequently"]
+    transitions = sum(1 for token in doc if token.text.lower() in transition_words)
+    coherence_score = min(9, transitions + 4)  # 4-9 scale
     
-    # Final Score
-    final_score = (vocab_score + grammar_score + coherence_score) / 3
+    # Overall Band (IELTS-style weighting)
+    overall = (vocab_score * 0.4) + (grammar_score * 0.3) + (coherence_score * 0.3)
     
     return {
-        "Vocabulary Band": vocab_score,
+        "Vocabulary Band": min(9, vocab_score),
         "Grammar Band": grammar_score,
         "Coherence Band": coherence_score,
-        "Overall Band": round(final_score, 1)
+        "Overall Band": round(overall, 1)
     }
 
-# Create Interface
 gr.Interface(
     fn=evaluate_ielts,
-    inputs=gr.Textbox(label="Paste Your Essay", lines=5),
+    inputs=gr.Textbox(label="Essay", lines=7),
     outputs=gr.JSON(),
-    title="IELTS Writing Evaluator",
-    examples=[["Education is essential for personal growth and societal development..."]]
+    examples=[[
+        "Education plays a vital role in societal development. " 
+        "Firstly, it equips individuals with critical thinking skills. " 
+        "Furthermore, educated citizens contribute to economic growth."
+    ]]
 ).launch()