Update app.py

app.py

@@ -291,11 +291,17 @@ def parse_transcript(file):
         if 'courses' in parsed_data:
             courses_by_grade = parsed_data['courses']
             
-            # Sort grades numerically (09, 10, 11, 12) or use original order
-            try:
-                grades_sorted = sorted(courses_by_grade.keys(), key=int)
-            except:
-                grades_sorted = sorted(courses_by_grade.keys())
+            # Improved grade sorting that handles both numeric and text grades
+            def grade_sort_key(grade):
+                try:
+                    # Extract numeric part from strings like "9th Grade" or "Grade 9"
+                    num = int(re.search(r'\d+', grade).group())
+                    return num
+                except (AttributeError, ValueError):
+                    # For non-numeric grades like "All", sort them last
+                    return float('inf')
+            
+            grades_sorted = sorted(courses_by_grade.keys(), key=grade_sort_key)
             
             for grade in grades_sorted:
                 output_text += f"\nGrade {grade}:\n{'-'*30}\n"
@@ -313,6 +319,46 @@ def parse_transcript(file):
     else:
         return "Unsupported file format (PDF only for transcript parsing)", None
 
+def transcript_display(transcript_dict):
+    if not transcript_dict or "courses" not in transcript_dict:
+        return "No course information available"
+    
+    display = "### Detailed Course History\n"
+    courses_by_grade = transcript_dict["courses"]
+    
+    if isinstance(courses_by_grade, dict):
+        # Improved grade sorting that handles both numeric and text grades
+        def grade_sort_key(grade):
+            try:
+                # Extract numeric part from strings like "9th Grade" or "Grade 9"
+                num = int(re.search(r'\d+', grade).group())
+                return num
+            except (AttributeError, ValueError):
+                # For non-numeric grades like "All", sort them last
+                return float('inf')
+        
+        grades_sorted = sorted(courses_by_grade.keys(), key=grade_sort_key)
+        
+        for grade in grades_sorted:
+            display += f"\n**Grade {grade}**\n"
+            for course in courses_by_grade[grade]:
+                display += f"- {course.get('name', 'Unnamed Course')}"
+                if 'grade' in course and course['grade']:
+                    display += f" (Grade: {course['grade']})"
+                if 'credits' in course:
+                    display += f" | Credits: {course['credits']}"
+                if 'school_year' in course:
+                    display += f" | Year: {course['school_year']}"
+                display += "\n"
+    
+    if 'gpa' in transcript_dict:
+        gpa = transcript_dict['gpa']
+        display += "\n**GPA Information**\n"
+        display += f"- Unweighted: {gpa.get('unweighted', 'N/A')}\n"
+        display += f"- Weighted: {gpa.get('weighted', 'N/A')}\n"
+    
+    return display
+
 # ========== LEARNING STYLE QUIZ ==========
 learning_style_questions = [
     "When you study for a test, you prefer to:",
@@ -479,40 +525,6 @@ def save_profile(name, age, interests, transcript, learning_style,
 """
     return markdown_summary
 
-def transcript_display(transcript_dict):
-    if not transcript_dict or "courses" not in transcript_dict:
-        return "No course information available"
-    
-    display = "### Detailed Course History\n"
-    courses_by_grade = transcript_dict["courses"]
-    
-    if isinstance(courses_by_grade, dict):
-        # Sort grades numerically
-        try:
-            grades_sorted = sorted(courses_by_grade.keys(), key=int)
-        except:
-            grades_sorted = sorted(courses_by_grade.keys())
-            
-        for grade in grades_sorted:
-            display += f"\n**Grade {grade}**\n"
-            for course in courses_by_grade[grade]:
-                display += f"- {course.get('name', 'Unnamed Course')}"
-                if 'grade' in course and course['grade']:
-                    display += f" (Grade: {course['grade']})"
-                if 'credits' in course:
-                    display += f" | Credits: {course['credits']}"
-                if 'school_year' in course:
-                    display += f" | Year: {course['school_year']}"
-                display += "\n"
-    
-    if 'gpa' in transcript_dict:
-        gpa = transcript_dict['gpa']
-        display += "\n**GPA Information**\n"
-        display += f"- Unweighted: {gpa.get('unweighted', 'N/A')}\n"
-        display += f"- Weighted: {gpa.get('weighted', 'N/A')}\n"
-    
-    return display
-
 # ========== AI TEACHING ASSISTANT ==========
 def load_profile():
     if not os.path.exists("student_profiles"):
@@ -592,11 +604,16 @@ def generate_response(message, history):
     elif any(word in message.lower() for word in course_help):
         response = "Here's a summary of your courses:\n"
         if isinstance(courses, dict):
-            try:
-                grades_sorted = sorted(courses.keys(), key=int)
-            except:
-                grades_sorted = sorted(courses.keys())
-                
+            # Use the same grade sorting logic as in transcript display
+            def grade_sort_key(grade):
+                try:
+                    num = int(re.search(r'\d+', grade).group())
+                    return num
+                except (AttributeError, ValueError):
+                    return float('inf')
+            
+            grades_sorted = sorted(courses.keys(), key=grade_sort_key)
+            
             for grade in grades_sorted:
                 response += f"\nGrade {grade}:\n"
                 for course in courses[grade]:
@@ -687,5 +704,4 @@ with gr.Blocks() as app:
         )
 
 if __name__ == "__main__":
-    app.launch()
-     
+    app.launch()