Update app.py

app.py

@@ -94,9 +94,31 @@ def auto_detect(file_path):
     m = re.match(r"spm_(\d{4})_(\w+)\.pdf", fname, re.IGNORECASE)
     if m:
         year, subject = m.groups()
-        return year, subject.capitalize()
+        return year, subject
     return None, None
 
+def normalize_subject(subject):
+    """Normalize subject names to match dropdown choices."""
+    valid_subjects = ["BM", "English", "Math", "History", "Science", "MoralStudies"]
+    if not subject:
+        return "BM"
+    subject = subject.strip().lower()
+    mapping = {
+        "bm": "BM",
+        "bahasa": "BM",
+        "bahasamelayu": "BM",
+        "b.m": "BM",
+        "english": "English",
+        "math": "Math",
+        "mathematics": "Math",
+        "history": "History",
+        "sejarah": "History",
+        "science": "Science",
+        "moral": "MoralStudies",
+        "moralstudies": "MoralStudies",
+    }
+    return mapping.get(subject, "BM") if subject in mapping else subject.capitalize()
+
 def process_pdf(file, subject, year):
     raw = ocr_agent.extract_from_pdf(file)
     cleaned = ocr_agent.clean_text(raw)
@@ -115,7 +137,8 @@ def prefill_subject_year(file):
     if not file:
         return "BM", "2018"
     year, subject = auto_detect(file)
-    return subject if subject else "BM", year if year else "2018"
+    subject = normalize_subject(subject)
+    return subject, year if year else "2018"
 
 # ----------------- Gradio UI -----------------
 with gr.Blocks() as demo:
@@ -189,3 +212,4 @@ if __name__ == "__main__":
 
 
 
+