Update app.py

app.py

@@ -263,6 +263,113 @@ def transcript_display(transcript_dict):
     return "\n".join([f"- {course}" for course in transcript_dict["courses"]] +
                      [f"Grade Level: {transcript_dict['grade_level']}", f"GPA: {transcript_dict['gpa']}"])
 
+# ========== AI TEACHING ASSISTANT ==========
+def load_profile():
+    files = [f for f in os.listdir("student_profiles") if f.endswith('.json')]
+    if files:
+        with open(os.path.join("student_profiles", files[0]), "r") as f:
+            return json.load(f)
+    return {}
+
+def generate_response(message, history):
+    profile = load_profile()
+    if not profile:
+        return "Please complete and save your profile first using the previous tabs."
+    
+    # Get profile data
+    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", "")
+    
+    # Common responses
+    greetings = ["hi", "hello", "hey"]
+    study_help = ["study", "learn", "prepare", "exam"]
+    grade_help = ["grade", "gpa", "score"]
+    interest_help = ["interest", "hobby", "passion"]
+    
+    if any(greet in message.lower() for greet in greetings):
+        return f"Hello {profile.get('name', 'there')}! How can I help you today?"
+    
+    elif any(word in message.lower() for word in study_help):
+        if "Visual" in learning_style:
+            response = ("Based on your visual learning style, I recommend:\n"
+                       "- Creating mind maps or diagrams\n"
+                       "- Using color-coded notes\n"
+                       "- Watching educational videos\n"
+                       "- Drawing concepts to understand them better")
+        elif "Auditory" in learning_style:
+            response = ("Based on your auditory learning style, I recommend:\n"
+                       "- Recording lectures and listening to them\n"
+                       "- Participating in study groups\n"
+                       "- Explaining concepts out loud\n"
+                       "- Using mnemonic devices with rhymes")
+        elif "Reading/Writing" in learning_style:
+            response = ("Based on your reading/writing learning style, I recommend:\n"
+                       "- Writing detailed notes\n"
+                       "- Creating summaries in your own words\n"
+                       "- Reading textbooks and articles\n"
+                       "- Making flashcards with written explanations")
+        elif "Kinesthetic" in learning_style:
+            response = ("Based on your kinesthetic learning style, I recommend:\n"
+                       "- Hands-on practice\n"
+                       "- Creating physical models\n"
+                       "- Taking frequent movement breaks\n"
+                       "- Using real-world applications of concepts")
+        else:
+            response = ("Here are some general study tips:\n"
+                       "- Break study sessions into 25-minute chunks\n"
+                       "- Review material regularly\n"
+                       "- Teach concepts to someone else\n"
+                       "- Practice with past exams or problems")
+        
+        # Add grade-specific advice
+        if "9" in grade_level or "Freshman" in grade_level:
+            response += "\n\nAs a freshman, focus on building good study habits early."
+        elif "10" in grade_level or "Sophomore" in grade_level:
+            response += "\n\nAs a sophomore, start exploring subjects that interest you."
+        elif "11" in grade_level or "Junior" in grade_level:
+            response += "\n\nAs a junior, focus on preparing for college entrance exams."
+        elif "12" in grade_level or "Senior" in grade_level:
+            response += "\n\nAs a senior, balance college applications with maintaining your grades."
+        
+        return response
+    
+    elif any(word in message.lower() for word in grade_help):
+        return (f"Your GPA information:\n"
+               f"- Unweighted: {gpa.get('unweighted', 'N/A')}\n"
+               f"- Weighted: {gpa.get('weighted', 'N/A')}\n\n"
+               "To improve your grades:\n"
+               "- Set specific, measurable goals\n"
+               "- Meet with teachers to understand expectations\n"
+               "- Develop a consistent study schedule\n"
+               "- Focus on your weakest areas first")
+    
+    elif any(word in message.lower() for word in interest_help):
+        return (f"I see you're interested in: {interests}\n\n"
+               "You might want to:\n"
+               "- Find clubs or activities related to these interests\n"
+               "- Explore career paths that align with them\n"
+               "- Connect these interests to your school subjects")
+    
+    elif "help" in message.lower():
+        return ("I can help with:\n"
+               "- Study tips based on your learning style\n"
+               "- GPA and grade information\n"
+               "- Academic advice tailored to your grade level\n"
+               "- Connecting your interests to learning\n\n"
+               "Try asking about:\n"
+               "- How to study for your classes\n"
+               "- Ways to improve your grades\n"
+               "- How to connect your interests to school")
+    
+    else:
+        return ("I'm your personalized teaching assistant. I can help with:\n"
+               "- Study strategies based on your learning style\n"
+               "- Understanding your academic performance\n"
+               "- Finding ways to connect school with your interests\n\n"
+               "Try asking me about study tips or your academic progress!")
+
 # ========== GRADIO INTERFACE ==========
 with gr.Blocks() as app:
     with gr.Tab("Step 1: Upload Transcript"):
@@ -326,4 +433,17 @@ with gr.Blocks() as app:
                             transcript_data, learning_output],
                      outputs=output_summary)
 
+    # AI Teaching Assistant Tab
+    with gr.Tab("🤖 AI Teaching Assistant"):
+        gr.Markdown("## Your Personalized Learning Assistant")
+        chatbot = gr.ChatInterface(
+            fn=generate_response,
+            examples=[
+                "How should I study for my next test?",
+                "What's my GPA information?",
+                "How can I improve my grades?",
+                "How can I connect my interests to school?"
+            ]
+        )
+
 app.launch()