Update app.py

app.py

@@ -14,7 +14,7 @@ except Exception as e:
     print(f"Could not load NER model: {e}")
     ner_pipeline = None
 
-# ========== TRANSCRIPT PARSING FUNCTIONS ==========
+# ========== IMPROVED TRANSCRIPT PARSING ==========
 def extract_gpa(text, gpa_type):
     pattern = rf'{gpa_type}\s*([\d\.]+)'
     match = re.search(pattern, text)
@@ -42,6 +42,8 @@ def extract_courses_from_table(text):
         course_name = course_name.strip()
         if 'DE:' in course_name:
             course_name = course_name.replace('DE:', 'Dual Enrollment:')
+        if 'AP' in course_name:
+            course_name = course_name.replace('AP', 'AP ')
         
         course_info = {
             'name': f"{course_code} {course_name}",
@@ -76,19 +78,23 @@ def parse_transcript(file):
         # Extract all courses with grades and year taken
         courses_by_grade = extract_courses_from_table(text)
         
-        # Prepare output text
-        output_text = f"Grade Level: {grade_level}\n"
+        # Prepare detailed output
+        output_text = f"Student Transcript Summary\n{'='*40}\n"
+        output_text += f"Current Grade Level: {grade_level}\n"
         output_text += f"Weighted GPA: {gpa_data['weighted']}\n"
         output_text += f"Unweighted GPA: {gpa_data['unweighted']}\n\n"
-        output_text += "Course History:\n"
+        output_text += "Course History:\n{'='*40}\n"
         
-        for grade, courses in sorted(courses_by_grade.items()):
-            output_text += f"\nGrade {grade}:\n"
-            for course in courses:
+        # Sort grades numerically (09, 10, 11, 12)
+        for grade in sorted(courses_by_grade.keys(), key=int):
+            output_text += f"\nGrade {grade}:\n{'-'*30}\n"
+            for course in courses_by_grade[grade]:
                 output_text += f"- {course['name']}"
-                if 'grade' in course:
+                if 'grade' in course and course['grade']:
                     output_text += f" (Grade: {course['grade']})"
-                output_text += "\n"
+                if 'credits' in course:
+                    output_text += f" | Credits: {course['credits']}"
+                output_text += f" | Year: {course['year']}\n"
         
         return output_text, {
             "gpa": gpa_data,
@@ -268,18 +274,20 @@ def transcript_display(transcript_dict):
     if not transcript_dict or "courses" not in transcript_dict:
         return "No course information available"
     
-    display = "### Course History\n\n"
+    display = "### Detailed Course History\n"
     courses_by_grade = transcript_dict["courses"]
     
     if isinstance(courses_by_grade, dict):
-        for grade, courses in sorted(courses_by_grade.items()):
-            display += f"**Grade {grade}**\n"
-            for course in courses:
+        # Sort grades numerically
+        for grade in sorted(courses_by_grade.keys(), key=int):
+            display += f"\n**Grade {grade}**\n"
+            for course in courses_by_grade[grade]:
                 display += f"- {course['name']}"
-                if 'grade' in course:
+                if 'grade' in course and course['grade']:
                     display += f" (Grade: {course['grade']})"
-                display += "\n"
-            display += "\n"
+                if 'credits' in course:
+                    display += f" | Credits: {course['credits']}"
+                display += f" | Year: {course['year']}\n"
     
     if 'gpa' in transcript_dict:
         gpa = transcript_dict['gpa']
@@ -309,12 +317,14 @@ def generate_response(message, history):
     grade_level = profile.get("transcript", {}).get("grade_level", "unknown")
     gpa = profile.get("transcript", {}).get("gpa", {})
     interests = profile.get("interests", "")
+    courses = profile.get("transcript", {}).get("courses", {})
     
     # Common responses
     greetings = ["hi", "hello", "hey"]
     study_help = ["study", "learn", "prepare", "exam"]
     grade_help = ["grade", "gpa", "score"]
     interest_help = ["interest", "hobby", "passion"]
+    course_help = ["courses", "classes", "transcript", "schedule"]
     
     if any(greet in message.lower() for greet in greetings):
         return f"Hello {profile.get('name', 'there')}! How can I help you today?"
@@ -363,10 +373,22 @@ def generate_response(message, history):
                "- Find clubs or activities related to these interests\n"
                "- Explore career paths that align with them")
     
+    elif any(word in message.lower() for word in course_help):
+        response = "Here's a summary of your courses:\n"
+        for grade in sorted(courses.keys(), key=int):
+            response += f"\nGrade {grade}:\n"
+            for course in courses[grade]:
+                response += f"- {course['name']}"
+                if 'grade' in course:
+                    response += f" (Grade: {course['grade']})"
+                response += "\n"
+        return response
+    
     elif "help" in message.lower():
         return ("I can help with:\n"
                "- Study tips based on your learning style\n"
                "- GPA and grade information\n"
+               "- Course history and schedules\n"
                "- General academic advice\n\n"
                "Try asking about study strategies or your grades!")
     
@@ -380,7 +402,7 @@ with gr.Blocks() as app:
     with gr.Tab("Step 1: Upload Transcript"):
         gr.Markdown("### Upload your transcript (PDF recommended for best results)")
         transcript_file = gr.File(label="Transcript file", file_types=[".pdf"])
-        transcript_output = gr.Textbox(label="Transcript Results", lines=10)
+        transcript_output = gr.Textbox(label="Transcript Results", lines=20)
         transcript_data = gr.State()
         transcript_file.change(
             fn=parse_transcript,
@@ -394,7 +416,7 @@ with gr.Blocks() as app:
         for i, (question, options) in enumerate(zip(learning_style_questions, learning_style_options)):
             quiz_components.append(gr.Radio(options, label=f"{i+1}. {question}"))
         
-        learning_output = gr.Textbox(label="Your Learning Style", lines=10)
+        learning_output = gr.Textbox(label="Your Learning Style", lines=15)
         gr.Button("Submit Quiz").click(
             fn=learning_style_quiz,
             inputs=quiz_components,
@@ -435,11 +457,11 @@ with gr.Blocks() as app:
             examples=[
                 "How should I study for my next test?",
                 "What's my GPA information?",
-                "Help me with study strategies",
+                "Show me my course history",
                 "How can I improve my grades?"
             ]
         )
 
 if __name__ == "__main__":
     app.launch()
-    
+