Update app.py

app.py

@@ -40,16 +40,16 @@ def parse_questions(file_path):
     return questions
 
 # Function to select exam questions
-def select_exam_(exam_name):
+def select_exam_(exam_name, num_questions=2):
     questions = parse_questions(f'questions/{exam_name}.set')
-    #num_questions = len(questions)
-    num_questions = 120
+    num_questions = len(questions)
     print("num_questions", num_questions)
     selected_questions = questions[:int(num_questions)]
+    #return selected_questions
     cleaned_questions = [
         {'question': q['question'], 
-         'options': [o.replace('- [ ] ', '').replace('- [x] ', '') for o in q['options']], 
-         'correct': q['correct'].replace('- [ ] ', '').replace('- [x] ', '')} 
+        'options': [o.replace('- [ ] ', '').replace('- [x] ', '').replace('- [X] ', '') for o in q['options']], 
+        'correct': q['correct'].replace('- [x] ', '').replace('- [X] ', '') if q['correct'] is not None else ''}
         for q in selected_questions
     ]
     return cleaned_questions