Update app.py

app.py

@@ -151,15 +151,26 @@ def answer_matching(text):
     return letters[0].upper() if len(letters) == 1 else random.choice(['A', 'B', 'C', 'D'])
 
 
-def evaluate_uploaded_json(user_file, model_name):
+def evaluate_uploaded_json(
+    user_file: str,
+    model_name: str,
+    multi_choice_file: str = "multi_choice.json",
+):
+    
     print(f"Model Name: {model_name}")
     print(f"Uploaded File: {user_file}")
 
     id_answer_mapping = load_id_answer_mapping()
 
+    with open(multi_choice_file, "r", encoding="utf-8") as f:
+        mc_data = json.load(f)
+    id_class_mapping = {q["id"]: q["class"] for q in mc_data}
+
     with open(user_file, "r", encoding="utf-8") as f:
         user_data = json.load(f)
 
+
+    
     correct = 0
     total = 0
 
@@ -170,7 +181,7 @@ def evaluate_uploaded_json(user_file, model_name):
         question_id = item["id"]
         raw_user_answer = item.get("model_answer", "")
         user_answer = answer_matching(raw_user_answer)
-        question_class = item.get("class", "Unknown")
+        question_class = id_class_mapping.get(question_id, "Unknown")
 
         class_total[question_class] += 1
         total += 1