Update app.py

app.py

@@ -201,17 +201,28 @@ def get_student_profile():
 
 
 import re
-def calculate_score_and_grade(llm_response):
+import re
+
+def calculate_score_and_grades(llm_response):
     print(llm_response)
+    
     # Extract all the marks using a regular expression
     marks = re.findall(r'(\d+)/(\d+)', llm_response)
+    
+    if not marks:
+        # Handle the case where no marks are found
+        return 0, 0, 0, 'No grade'
 
     # Calculate total score and maximum possible score
     total_score = sum(int(mark[0]) for mark in marks)
     max_possible_score = sum(int(mark[1]) for mark in marks)
 
+    # Prevent division by zero
+    if max_possible_score == 0:
+        return total_score, max_possible_score, 0, 'Fail'
+
     # Calculate the percentage
-    percentage = (total_score / max_possible_score) * 100 if max_possible_score > 0 else 0
+    percentage = (total_score / max_possible_score) * 100
 
     # Determine the grade based on the percentage
     if percentage > 90:
@@ -234,7 +245,7 @@ def calculate_score_and_grade(llm_response):
 def assign_grade():
     data = request.json
     result = data['result']
-    total_score, max_possible_score, percentage, grade = calculate_score_and_grade(result)
+    total_score, max_possible_score, percentage, grade = calculate_score_and_grades(result)
     return jsonify({
         'total_score': total_score,
         'max_possible_score': max_possible_score,