Update app.py

app.py

@@ -1022,7 +1022,7 @@ class ProfileManager:
                     transcript: Dict, learning_style: str, 
                     movie: str, movie_reason: str, show: str, show_reason: str, 
                     book: str, book_reason: str, character: str, character_reason: str, 
-                    blog: str = "") -> str:
+                    blog: str) -> str:
         """Save student profile with validation."""
         try:
             # Validate required fields
@@ -1151,19 +1151,19 @@ class ProfileManager:
         """Generate markdown summary of the profile."""
         transcript = data.get("transcript", {})
         favorites = data.get("favorites", {})
-        learning_style = data.get("learning_style", "Not assessed")
         
-        # Extract just the learning style name (remove any markdown formatting)
-        learning_style_name = learning_style.split("##")[0].strip()
-        if "Your primary learning style is" in learning_style_name:
-            learning_style_name = learning_style_name.split("Your primary learning style is")[1].strip()
-            learning_style_name = learning_style_name.split("\n")[0].strip()
+        # Extract just the learning style name
+        learning_style = data.get("learning_style", "")
+        if "Your primary learning style is:" in learning_style:
+            style_match = re.search(r"Your primary learning style is: \*\*(.*?)\*\*", learning_style)
+            if style_match:
+                learning_style = style_match.group(1)
         
         markdown = f"""## Student Profile: {data['name']}
 ### Basic Information
 - **Age:** {data['age']}
 - **Interests:** {data.get('interests', 'Not specified')}
-- **Learning Style:** {learning_style_name}
+- **Learning Style:** {learning_style}
 ### Academic Information
 {self._format_transcript(transcript)}
 ### Favorites
@@ -1182,52 +1182,13 @@ class ProfileManager:
     
     def _format_transcript(self, transcript: Dict) -> str:
         """Format transcript data for display."""
-        if not transcript:
+        if not transcript or "courses" not in transcript:
             return "_No transcript information available_"
         
-        display = ""
+        display = "#### Course History\n"
+        courses_by_grade = transcript["courses"]
         
-        # Handle different transcript formats
-        if "student_info" in transcript:
-            # Structured transcript format
-            student = transcript.get("student_info", {})
-            display += f"**Name:** {student.get('name', 'Unknown')}\n"
-            display += f"**Student ID:** {student.get('id', 'Unknown')}\n"
-            display += f"**Current Grade:** {student.get('current_grade', 'Unknown')}\n"
-            display += f"**Graduation Year:** {student.get('graduation_year', 'Unknown')}\n"
-            
-            if 'unweighted_gpa' in student and 'weighted_gpa' in student:
-                display += f"**Unweighted GPA:** {student['unweighted_gpa']}\n"
-                display += f"**Weighted GPA:** {student['weighted_gpa']}\n"
-            elif 'gpa' in student:
-                display += f"**GPA:** {student['gpa']}\n"
-            
-            if 'total_credits' in student:
-                display += f"**Total Credits Earned:** {student['total_credits']}\n"
-            if 'community_service_hours' in student:
-                display += f"**Community Service Hours:** {student['community_service_hours']}\n"
-            
-            # Course History
-            if "course_history" in transcript:
-                display += "\n## Course History\n"
-                courses_by_year = defaultdict(list)
-                for course in transcript["course_history"]:
-                    year_key = course.get("school_year", course.get("completion_date", "Unknown"))
-                    courses_by_year[year_key].append(course)
-                
-                for year in sorted(courses_by_year.keys()):
-                    display += f"\n### {year}\n"
-                    for course in courses_by_year[year]:
-                        display += f"- **{course.get('course_code', '')} {course.get('description', 'Unnamed course')}**\n"
-                        display += f"  Subject: {course.get('subject', 'N/A')} | "
-                        display += f"Grade: {course.get('grade', 'N/A')} | "
-                        display += f"Credits: {course.get('credits', 'N/A')}\n"
-        
-        elif "courses" in transcript:
-            # Simple course format
-            display += "\n## Course History\n"
-            courses_by_grade = transcript["courses"]
-            
+        if isinstance(courses_by_grade, dict):
             for grade in sorted(courses_by_grade.keys(), key=lambda x: int(x) if x.isdigit() else x):
                 display += f"\n**Grade {grade}**\n"
                 for course in courses_by_grade[grade]:
@@ -1237,14 +1198,259 @@ class ProfileManager:
                     if 'credits' in course:
                         display += f" | Credits: {course['credits']}"
                     display += f" | Year: {course.get('year', 'N/A')}\n"
+        
+        if 'gpa' in transcript:
+            gpa = transcript['gpa']
+            display += "\n**GPA**\n"
+            display += f"- Unweighted: {gpa.get('unweighted', 'N/A')}\n"
+            display += f"- Weighted: {gpa.get('weighted', 'N/A')}\n"
+        
+        return display
+
+# Initialize profile manager
+profile_manager = ProfileManager()
+
+# ========== AI TEACHING ASSISTANT ==========
+class TeachingAssistant:
+    def __init__(self):
+        self.context_history = []
+        self.max_context_length = 5  # Keep last 5 exchanges for context
+    
+    async def generate_response(self, message: str, history: List[List[Union[str, None]]], session_token: str) -> str:
+        """Generate personalized response based on student profile and context."""
+        try:
+            # Load profile with session token
+            profile = profile_manager.load_profile(session_token=session_token)
+            if not profile:
+                return "Please complete and save your profile first using the previous tabs."
+            
+            # Update context history
+            self._update_context(message, history)
+            
+            # Extract profile information
+            name = profile.get("name", "there")
+            learning_style = profile.get("learning_style", "")
+            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", {})
+            favorites = profile.get("favorites", {})
+            
+            # Process message with context
+            response = await self._process_message(message, profile)
+            
+            # Add follow-up suggestions
+            if "study" in message.lower() or "learn" in message.lower():
+                response += "\n\nWould you like me to suggest a study schedule based on your courses?"
+            elif "course" in message.lower() or "class" in message.lower():
+                response += "\n\nWould you like help finding resources for any of these courses?"
             
-            if 'gpa' in transcript:
-                gpa = transcript['gpa']
-                display += "\n**GPA**\n"
-                display += f"- Unweighted: {gpa.get('unweighted', 'N/A')}\n"
-                display += f"- Weighted: {gpa.get('weighted', 'N/A')}\n"
+            return response
         
-        return display if display else "_No transcript information available_"
+        except Exception as e:
+            logging.error(f"Error generating response: {str(e)}")
+            return "I encountered an error processing your request. Please try again."
+    
+    def _update_context(self, message: str, history: List[List[Union[str, None]]]) -> None:
+        """Maintain conversation context."""
+        self.context_history.append({"role": "user", "content": message})
+        if history:
+            for h in history[-self.max_context_length:]:
+                if h[0]:  # User message
+                    self.context_history.append({"role": "user", "content": h[0]})
+                if h[1]:  # Assistant message
+                    self.context_history.append({"role": "assistant", "content": h[1]})
+        
+        # Trim to maintain max context length
+        self.context_history = self.context_history[-(self.max_context_length*2):]
+    
+    async def _process_message(self, message: str, profile: Dict) -> str:
+        """Process user message with profile context."""
+        message_lower = message.lower()
+        
+        # Greetings
+        if any(greet in message_lower for greet in ["hi", "hello", "hey", "greetings"]):
+            return f"Hello {profile.get('name', 'there')}! How can I help you with your learning today?"
+        
+        # Study help
+        study_words = ["study", "learn", "prepare", "exam", "test", "homework"]
+        if any(word in message_lower for word in study_words):
+            return self._generate_study_advice(profile)
+        
+        # Grade help
+        grade_words = ["grade", "gpa", "score", "marks", "results"]
+        if any(word in message_lower for word in grade_words):
+            return self._generate_grade_advice(profile)
+        
+        # Interest help
+        interest_words = ["interest", "hobby", "passion", "extracurricular"]
+        if any(word in message_lower for word in interest_words):
+            return self._generate_interest_advice(profile)
+        
+        # Course help
+        course_words = ["courses", "classes", "transcript", "schedule", "subject"]
+        if any(word in message_lower for word in course_words):
+            return self._generate_course_advice(profile)
+        
+        # Favorites
+        favorite_words = ["movie", "show", "book", "character", "favorite"]
+        if any(word in message_lower for word in favorite_words):
+            return self._generate_favorites_response(profile)
+        
+        # General help
+        if "help" in message_lower:
+            return self._generate_help_response()
+        
+        # Default response
+        return ("I'm your personalized teaching assistant. I can help with study tips, "
+                "grade information, course advice, and more. Try asking about how to "
+                "study effectively or about your course history.")
+    
+    def _generate_study_advice(self, profile: Dict) -> str:
+        """Generate study advice based on learning style."""
+        learning_style = profile.get("learning_style", "")
+        response = ""
+        
+        if "Visual" in learning_style:
+            response = ("Based on your visual learning style, I recommend:\n"
+                       "- Creating colorful mind maps or diagrams\n"
+                       "- Using highlighters to color-code your notes\n"
+                       "- Watching educational videos on the topics\n"
+                       "- Creating flashcards with images\n\n")
+        elif "Auditory" in learning_style:
+            response = ("Based on your auditory learning style, I recommend:\n"
+                       "- Recording your notes and listening to them\n"
+                       "- Participating in study groups to discuss concepts\n"
+                       "- Explaining the material out loud to yourself\n"
+                       "- Finding podcasts or audio lectures on the topics\n\n")
+        elif "Reading/Writing" in learning_style:
+            response = ("Based on your reading/writing learning style, I recommend:\n"
+                       "- Writing detailed summaries in your own words\n"
+                       "- Creating organized outlines of the material\n"
+                       "- Reading additional textbooks or articles\n"
+                       "- Rewriting your notes to reinforce learning\n\n")
+        elif "Kinesthetic" in learning_style:
+            response = ("Based on your kinesthetic learning style, I recommend:\n"
+                       "- Creating physical models or demonstrations\n"
+                       "- Using hands-on activities to learn concepts\n"
+                       "- Taking frequent movement breaks while studying\n"
+                       "- Associating information with physical actions\n\n")
+        else:
+            response = ("Here are some general study tips:\n"
+                       "- Use the Pomodoro technique (25 min study, 5 min break)\n"
+                       "- Space out your study sessions over time\n"
+                       "- Test yourself with practice questions\n"
+                       "- Teach the material to someone else\n\n")
+        
+        # Add time management advice
+        response += ("**Time Management Tips**:\n"
+                    "- Create a study schedule and stick to it\n"
+                    "- Prioritize difficult subjects when you're most alert\n"
+                    "- Break large tasks into smaller, manageable chunks\n"
+                    "- Set specific goals for each study session")
+        
+        return response
+    
+    def _generate_grade_advice(self, profile: Dict) -> str:
+        """Generate response about grades and GPA."""
+        gpa = profile.get("transcript", {}).get("gpa", {})
+        courses = profile.get("transcript", {}).get("courses", {})
+        
+        response = (f"Your GPA information:\n"
+                   f"- Unweighted: {gpa.get('unweighted', 'N/A')}\n"
+                   f"- Weighted: {gpa.get('weighted', 'N/A')}\n\n")
+        
+        # Identify any failing grades
+        weak_subjects = []
+        for grade_level, course_list in courses.items():
+            for course in course_list:
+                if course.get('grade', '').upper() in ['D', 'F']:
+                    weak_subjects.append(f"{course.get('code', '')} {course.get('name', 'Unknown course')}")
+        
+        if weak_subjects:
+            response += ("**Areas for Improvement**:\n"
+                        f"You might want to focus on these subjects: {', '.join(weak_subjects)}\n\n")
+        
+        response += ("**Grade Improvement Strategies**:\n"
+                    "- Meet with your teachers to discuss your performance\n"
+                    "- Identify specific areas where you lost points\n"
+                    "- Create a targeted study plan for weak areas\n"
+                    "- Practice with past exams or sample questions")
+        
+        return response
+    
+    def _generate_interest_advice(self, profile: Dict) -> str:
+        """Generate response based on student interests."""
+        interests = profile.get("interests", "")
+        response = f"I see you're interested in: {interests}\n\n"
+        
+        response += ("**Suggestions**:\n"
+                    "- Look for clubs or extracurricular activities related to these interests\n"
+                    "- Explore career paths that align with these interests\n"
+                    "- Find online communities or forums about these topics\n"
+                    "- Consider projects or independent study in these areas")
+        
+        return response
+    
+    def _generate_course_advice(self, profile: Dict) -> str:
+        """Generate response about courses."""
+        courses = profile.get("transcript", {}).get("courses", {})
+        grade_level = profile.get("transcript", {}).get("grade_level", "unknown")
+        
+        response = "Here's a summary of your courses:\n"
+        for grade in sorted(courses.keys(), key=lambda x: int(x) if x.isdigit() else x):
+            response += f"\n**Grade {grade}**:\n"
+            for course in courses[grade]:
+                response += f"- {course.get('code', '')} {course.get('name', 'Unnamed course')}"
+                if 'grade' in course:
+                    response += f" (Grade: {course['grade']})"
+                response += "\n"
+        
+        response += f"\nAs a grade {grade_level} student, you might want to:\n"
+        if grade_level in ["9", "10"]:
+            response += ("- Focus on building strong foundational skills\n"
+                        "- Explore different subjects to find your interests\n"
+                        "- Start thinking about college/career requirements")
+        elif grade_level in ["11", "12"]:
+            response += ("- Focus on courses relevant to your college/career goals\n"
+                        "- Consider taking AP or advanced courses if available\n"
+                        "- Ensure you're meeting graduation requirements")
+        
+        return response
+    
+    def _generate_favorites_response(self, profile: Dict) -> str:
+        """Generate response about favorite items."""
+        favorites = profile.get("favorites", {})
+        response = "I see you enjoy:\n"
+        
+        if favorites.get('movie'):
+            response += f"- Movie: {favorites['movie']} ({favorites.get('movie_reason', 'no reason provided')})\n"
+        if favorites.get('show'):
+            response += f"- TV Show: {favorites['show']} ({favorites.get('show_reason', 'no reason provided')})\n"
+        if favorites.get('book'):
+            response += f"- Book: {favorites['book']} ({favorites.get('book_reason', 'no reason provided')})\n"
+        if favorites.get('character'):
+            response += f"- Character: {favorites['character']} ({favorites.get('character_reason', 'no reason provided')})\n"
+        
+        response += "\nThese preferences suggest you might enjoy:\n"
+        response += "- Similar books/movies in the same genre\n"
+        response += "- Creative projects related to these stories\n"
+        response += "- Analyzing themes or characters in your schoolwork"
+        
+        return response
+    
+    def _generate_help_response(self) -> str:
+        """Generate help response with available commands."""
+        return ("""I can help with:
+- **Study tips**: "How should I study for math?"
+- **Grade information**: "What's my GPA?"
+- **Course advice**: "Show me my course history"
+- **Interest suggestions**: "What clubs match my interests?"
+- **General advice**: "How can I improve my grades?"
+Try asking about any of these topics!""")
+
+# Initialize teaching assistant
+teaching_assistant = TeachingAssistant()
 
 # ========== GRADIO INTERFACE ==========
 def create_interface():
@@ -1274,7 +1480,6 @@ def create_interface():
         .quiz-question { margin-bottom: 15px; padding: 15px; background: #f5f5f5; border-radius: 5px; }
         .quiz-results { margin-top: 20px; padding: 20px; background: #e8f5e9; border-radius: 8px; }
         .error-message { color: #d32f2f; background-color: #ffebee; padding: 10px; border-radius: 4px; margin: 10px 0; }
-        .blog-textbox { margin-top: 15px; }
         
         /* Dark mode support */
         .dark .tab-content { background-color: #2d2d2d !important; border-color: #444 !important; }
@@ -1368,8 +1573,6 @@ def create_interface():
                         error_msg = f"❌ Error: {str(e)}"
                         if "PDF" in str(e):
                             error_msg += "\n\nTIPS FOR PDF FILES:\n1. Try opening and re-saving the PDF\n2. Ensure it's not password protected\n3. Try converting to an image"
-                        elif "image" in str(e).lower():
-                            error_msg += "\n\nTIPS FOR IMAGE FILES:\n1. Ensure the image is clear and well-lit\n2. Try cropping to just the transcript area\n3. Avoid blurry or low-resolution images"
                         return (
                             error_msg,
                             None,
@@ -1481,14 +1684,6 @@ def create_interface():
                                 placeholder="e.g., Science, Music, Sports, Art..."
                             )
                         
-                        # Add the blog textbox here
-                        blog = gr.Textbox(
-                            label="Personal Blog (Optional)",
-                            placeholder="Share your thoughts, goals, or anything you'd like to remember...",
-                            lines=4,
-                            elem_classes="blog-textbox"
-                        )
-                        
                         save_personal_btn = gr.Button("Save Information", variant="primary")
                         save_confirmation = gr.HTML(visible=False)
                     
@@ -1503,8 +1698,16 @@ def create_interface():
                             book_reason = gr.Textbox(label="Why do you like it?", lines=2)
                             character = gr.Textbox(label="Favorite Character (from any story)")
                             character_reason = gr.Textbox(label="Why do you like them?", lines=2)
+                        
+                        # Added blog section
+                        with gr.Accordion("Personal Blog (Optional)", open=False):
+                            blog = gr.Textbox(
+                                label="Share your thoughts",
+                                placeholder="Write something about yourself, your goals, or anything you'd like to share...",
+                                lines=5
+                            )
                 
-                def save_personal_info(name, age, interests, current_tab_status, blog=""):
+                def save_personal_info(name, age, interests, current_tab_status):
                     try:
                         name = validate_name(name)
                         age = validate_age(age)
@@ -1530,7 +1733,7 @@ def create_interface():
                 
                 save_personal_btn.click(
                     fn=save_personal_info,
-                    inputs=[name, age, interests, tab_completed, blog],
+                    inputs=[name, age, interests, tab_completed],
                     outputs=[tab_completed, step3, step4, save_confirmation, nav_message]
                 )
 
@@ -1633,12 +1836,12 @@ def create_interface():
                 )
                 
                 clear_btn.click(
-                    fn=lambda: [gr.update(value="") for _ in range(14)],  # Updated to include blog
+                    fn=lambda: [gr.update(value="") for _ in range(12)],
                     outputs=[
                         name, age, interests, 
                         movie, movie_reason, show, show_reason,
                         book, book_reason, character, character_reason,
-                        blog, output_summary
+                        output_summary
                     ]
                 )