Update app.py

app.py

@@ -76,7 +76,8 @@ class ProfessionalEssayGrader:
         self.grammar_model = pipeline(
             "text2text-generation",
             model="pszemraj/flan-t5-large-grammar-synthesis",
-            device=-1
+            device=-1,
+            use_fast=False
         )
         self.coherence_model = pipeline(
             "sentiment-analysis",
@@ -96,8 +97,29 @@ class ProfessionalEssayGrader:
         self.paraphrase_model = pipeline(
             "text2text-generation",
             model="Vamsi/T5_Paraphrase_Paws",
+            device=-1,
+            tokenizer_kwargs={"use_fast": False}
+        )
+        # Add advanced readability model
+        self.readability_model = pipeline(
+            "text-classification",
+            model="nlptown/bert-base-multilingual-uncased-sentiment",
             device=-1
         )
+        # Add advanced grammar correction model
+        self.grammar_correction_model = pipeline(
+            "text2text-generation",
+            model="prithivida/grammar_error_correcter_v1",
+            device=-1,
+            use_fast=False
+        )
+        # Add advanced feedback generator
+        self.feedback_model = pipeline(
+            "text2text-generation",
+            model="google/flan-t5-large",
+            device=-1,
+            use_fast=False
+        )
         # Grammar checker fallback: if Java is missing, use basic regex-based grammar checks
         try:
             self.grammar_checker = language_tool_python.LanguageTool('en-US')
@@ -126,6 +148,16 @@ class ProfessionalEssayGrader:
         # Paraphrase (for feedback)
         paraphrase = self.paraphrase_model(f"paraphrase: {essay_text[:256]}")[0]['generated_text'] if len(essay_text) > 50 else "Essay too short for paraphrase."
 
+        # Readability analysis
+        readability_score = self.readability_model(essay_text)[0]['label']
+
+        # Grammar correction
+        grammar_correction = self.grammar_correction_model(essay_text)[0]['generated_text']
+
+        # Advanced feedback generation
+        feedback_prompt = f"Give extremely detailed Ontario curriculum feedback for this essay: {essay_text}"
+        advanced_feedback = self.feedback_model(feedback_prompt)[0]['generated_text']
+
         # Weighted overall score
         overall_score = self._calculate_enhanced_score(stats, readability, coherence, grammar, structure, content, style)
         overall_score = max(0, min(100, overall_score))
@@ -135,7 +167,7 @@ class ProfessionalEssayGrader:
 
         # Generate feedback
         feedback = self._generate_ontario_feedback(
-            overall_score, rubric, stats, readability, coherence, grammar, structure, content, style, topic_results, summary, paraphrase
+            overall_score, rubric, stats, readability, coherence, grammar, structure, content, style, topic_results, summary, paraphrase, readability_score, grammar_correction, advanced_feedback
         )
 
         return {
@@ -145,6 +177,9 @@ class ProfessionalEssayGrader:
             "topic_detection": topic_results,
             "summary": summary,
             "paraphrase": paraphrase,
+            "readability_score": readability_score,
+            "grammar_correction": grammar_correction,
+            "advanced_feedback": advanced_feedback,
             "detailed_breakdown": {
                 "enhanced_statistics": stats,
                 "readability_analysis": readability,
@@ -167,7 +202,7 @@ class ProfessionalEssayGrader:
                 }
         return ONTARIO_RUBRIC["R"]
     
-    def _generate_ontario_feedback(self, score, rubric, stats, readability, coherence, grammar, structure, content, style, topic_results, summary, paraphrase) -> Dict:
+    def _generate_ontario_feedback(self, score, rubric, stats, readability, coherence, grammar, structure, content, style, topic_results, summary, paraphrase, readability_score, grammar_correction, advanced_feedback) -> Dict:
         """Generate extremely detailed and specific feedback aligned with Ontario curriculum"""
         
         feedback = []
@@ -179,6 +214,9 @@ class ProfessionalEssayGrader:
         feedback.append(f"Summary: {summary}")
         feedback.append(f"Paraphrase: {paraphrase}")
         feedback.append(f"Topic Detection: {topic_results}")
+        feedback.append(f"Readability Score: {readability_score}")
+        feedback.append(f"Grammar Correction: {grammar_correction}")
+        feedback.append(f"Advanced Feedback: {advanced_feedback}")
         feedback.append(f"Statistics: {stats}")
         feedback.append(f"Readability: {readability}")
         feedback.append(f"Coherence: {coherence}")
@@ -191,6 +229,7 @@ class ProfessionalEssayGrader:
         feedback.append("- Use topic analysis to strengthen essay focus.")
         feedback.append("- Improve grammar and coherence as suggested.")
         feedback.append("- Expand vocabulary and sentence variety.")
+        feedback.append("- Apply advanced feedback for Ontario curriculum excellence.")
         
         return feedback