Update app.py

app.py

@@ -110,9 +110,19 @@ def detect_grammar_errors(text):
     outputs = grammar_model.generate(inputs, max_length=512, num_beams=4, early_stopping=True)
     grammar_analysis = grammar_tokenizer.decode(outputs[0], skip_special_tokens=True)
     
-    # Count the number of errors based on specific indicators (customize based on analysis)
-    error_count = grammar_analysis.count('error')  # Use your own criteria
-    return error_count
+    # Compare the corrected sentence with the original one
+    corrected_sentence = grammar_analysis
+    
+    # Simple heuristic to count changes as errors (e.g., difference in word count)
+    # You can also use edit distance or other sophisticated techniques to count errors
+    original_words = text.split()
+    corrected_words = corrected_sentence.split()
+    
+    # Calculate a rough estimate of grammar errors based on differences in word count
+    error_count = abs(len(original_words) - len(corrected_words))
+    
+    return error_count, corrected_sentence
+
 
 def penalize_for_grammar(student_answer):
     grammar_errors = detect_grammar_errors(student_answer)