add python course and fix some bugs

app.py

@@ -135,9 +135,9 @@ def login_form():
         # Role selection at the top
         user_type = st.selectbox(
             "Please select your Role",
-            ["student", "faculty", "research_assistant", "analyst"]
+            ["Student", "Faculty", "Research Assistant", "Analyst"]
         )
-        
+        user_type = user_type.lower().replace(" ", "_")
         # Username/email and password stacked vertically
         username = st.text_input("Username or Email")
         password = st.text_input("Password", type="password")
@@ -180,7 +180,7 @@ def get_courses(username, user_type):
             # courses = list(courses_cursor1) + list(courses_cursor2)
             return list(courses)
     elif user_type == "faculty":
-        faculty = faculty_collection.find_one({"full_name": username})
+        faculty = faculty_collection.find_one({"$or": [{"full_name": username}, {"username": username}]})
         if faculty:
             course_ids = [
                 course["course_id"] for course in faculty.get("courses_taught", [])
@@ -224,7 +224,7 @@ def create_session(new_session, course_id):
         last_session_id = int(last_session_id[1:])
         new_session_id = last_session_id + 1
         new_session["session_id"] = "S" + str(new_session_id)
-        courses_collection2.update_one(
+        courses_collection.update_one(
             {"course_id": new_session["course_id"]},
             {"$push": {"sessions": new_session}},
         )
@@ -250,7 +250,7 @@ def create_session_form(course_id):
 
         new_session_id = None
         # Generate new session ID
-        course = courses_collection2.find_one({"course_id": course_id})
+        course = courses_collection.find_one({"course_id": course_id})
         if course and "sessions" in course and course["sessions"]:
             last_session_id = max(
                 int(session["session_id"][1:]) for session in course["sessions"]
@@ -281,7 +281,7 @@ def create_session_form(course_id):
                     "assignments": [],
                 },
             }
-            courses_collection2.update_one(
+            courses_collection.update_one(
                 {"course_id": course_id}, {"$push": {"sessions": new_session}}
             )
             st.success("Session created successfully!")
@@ -553,9 +553,9 @@ def register_page():
     # Select user type
     st.session_state.user_type = st.selectbox(
         "Please select your Role", 
-        ["student", "faculty", "research_assistant", "analyst"]
+        ["Student", "Faculty", "Research Assistant", "Analyst"]
     )
-    user_type = st.session_state.user_type
+    user_type = st.session_state.user_type.lower().replace(" ", "_")
 
     with st.form("register_form"):
         col1, col2 = st.columns(2)

session_page.py

@@ -3,7 +3,7 @@ import json
 import random
 import requests
 import streamlit as st
-from datetime import datetime
+from datetime import datetime, timedelta
 from youtube_transcript_api import YouTubeTranscriptApi
 from utils.helpers import display_progress_bar, create_notification, format_datetime
 from file_upload_vectorize import upload_resource, extract_text_from_file, create_vector_store, resources_collection, model, assignment_submit
@@ -23,6 +23,7 @@ from pre_class_analytics2 import NovaScholarAnalytics
 import openai
 from openai import OpenAI
 import google.generativeai as genai
+from google.generativeai import caching
 from goals2 import GoalAnalyzer
 from openai import OpenAI
 import asyncio
@@ -34,6 +35,7 @@ import streamlit.components.v1 as components
 from live_chat_feature import display_live_chat_interface 
 from code_playground import display_code_playground
 
+# Load environment variables
 load_dotenv()
 MONGO_URI = os.getenv('MONGO_URI')
 PERPLEXITY_API_KEY = os.getenv('PERPLEXITY_KEY')
@@ -47,6 +49,11 @@ assignment_evaluation_collection = db["assignment_evaluation"]
 subjective_tests_collection = db["subjective_tests"]
 synoptic_store_collection = db["synoptic_store"]
 
+
+# for implementing Context Caching: 
+# PROJECT_ID = "novascholar-446709"
+# vertexai.init(project=PROJECT_ID, location="us-west4")
+
 def get_current_user():
     if 'current_user' not in st.session_state:
         return None
@@ -334,7 +341,6 @@ def display_preclass_content(session, student_id, course_id):
                         "timestamp": datetime.utcnow()
                     }
                     st.session_state.messages.append(new_message)
-
                     # Update database
                     try:
                         chat_history_collection.update_one(
@@ -700,6 +706,19 @@ def upload_preclass_materials(session_id, course_id):
                     - URL: [{resource['url']}]({resource['url']})
                     """)
 
+    # Display pre-class materials
+    # Group resources by their types
+    grouped_resources = defaultdict(list)
+    materials = resources_collection.find({"session_id": session_id})
+    for material in materials:
+        grouped_resources[material['material_type']].append(material)
+
+    # Display grouped resources
+    for material_type, resources in grouped_resources.items():
+        st.markdown(f"##### {material_type.capitalize()} Resources")
+        for material in resources:
+            st.markdown(f"- **{material['file_name']}** ({material['file_type']})")
+
 def extract_external_content(url, content_type):
     """Extract content from external resources based on their type"""
     try: