Upload app.py

app.py

@@ -388,16 +388,21 @@ def classify_solution_stream(question: str, solution: str):
             yield "Models not loaded", "", render(log)
             return
         log[-1] = "✅ Models loaded."
+    
+    verdicts_mapping = {"correct": "Correct.", "conceptual_error": "Conceptual error.", "computational_error": "Computational error."}
 
     try:
         # ---------- Stage 1: Conceptual ----------
         log.append("⏳ **Stage 1: Conceptual check**")
         yield "⏳ Working…", "Starting initial conceptual check…", render(log)
-
         conceptual = run_conceptual_check(question, solution, classifier_model, classifier_tokenizer)
         pred = conceptual["prediction"]
         conf = conceptual["probabilities"][pred]
-        log[-1] = f"✅ **Stage 1: Conceptual check** — prediction: **{pred}** (p={conf:.2%})"
+        if pred == "flawed":   
+            log[-1] = f"🟥 **Stage 1: Conceptual check** — (Complete) — prediction: **{pred}** (p={conf:.2%})"
+        elif pred == "correct":
+            log[-1] = f"✅ **Stage 1: Conceptual check** — (Complete) — prediction: **{pred}** (p={conf:.2%})"
+            
         yield "⏳ Working…", f"Stage 1: {pred} (p={conf:.2%}). Now checking calculations…", render(log)
 
         # ---------- Stage 2: Computational ----------
@@ -411,7 +416,7 @@ def classify_solution_stream(question: str, solution: str):
             # mark stage 2 as failed
             line_txt = computational["line_text"]
             corr = computational["correct_calc"]
-            log[-1] = f"🟥 **Stage 2: Computational check** — error on line “{line_txt}” (correct: `{corr}`)"
+            log[-1] = f"🟥 **Stage 2: Computational check** — (Completed; error found) — — error on line “{line_txt}” (correct: `{corr}`)"
             classification = "computational_error"
             explanation = (
                 "A calculation error was found.\n"
@@ -419,7 +424,7 @@ def classify_solution_stream(question: str, solution: str):
                 f"The correct calculation should be: {corr}"
             )
         else:
-            log[-1] = "✅ **Stage 2: Computational check** — no arithmetic issues found."
+            log[-1] = "✅ **Stage 2: Computational check** — (Complete) — no arithmetic issues found."
             if pred == "correct":
                 classification = "correct"
                 explanation = "All calculations are correct and the overall logic appears to be sound."
@@ -429,7 +434,7 @@ def classify_solution_stream(question: str, solution: str):
                     "All calculations are correct, but there appears to be a conceptual error "
                     "in the logic or setup of the solution."
                 )
-
+        classification = verdicts_mapping[classification]
         # final yield updates both result fields + the complete checklist
         yield classification, explanation, render(log)
 
@@ -476,7 +481,7 @@ with gr.Blocks(title="Math Solution Classifier", theme=gr.themes.Soft()) as app:
         ["John has three apples and Mary has seven, how many apples do they have together?",
              "They have 7 + 3 = 11 apples."],
         ["A tank holds 60 liters of fuel. A generator uses fuel at a rate of 5 liters per hour. After running for 9 hours, how many liters are still in the tank?",
-"The generator uses 5 L/h × 9 h = 45 L of fuel in 9 hours. The, there remain 60 L + 45 L = 105 L in the tank. Final answer: 105 L"],
+"The generator uses 5 L/h × 9 h = 45 L of fuel in 9 hours.\n Then, there remain 60 L + 45 L = 105 L in the tank.\n Final answer: 105 L"],
             ["What is 15% of 200?",
              "15% = 15/100 = 0.15\n0.15 × 200 = 30"],
         ],