app.py

# render_template -  api uses to generate html 
# request - object we need for forms 
from data.modify_data import clean_course_info
from flask import Flask, jsonify, request, redirect, render_template, flash
import pandas as pd 
import pickle
from data.get_recs import recommend_courses
from data.encryption import decrypt_file
from filters import get_filters
import traceback
from flask_limiter import Limiter
from flask_limiter.util import get_remote_address

app = Flask(__name__)

# set to correct semester
current_semester = '2024;FA'

# set current semester in app.py
term = current_semester.replace(';', '') # use correct term (2023FA for fall 2023)

# limits the number of requests by a single user per day
limiter = Limiter(
    get_remote_address,
    app=app,
    default_limits=["50 per day"],
    storage_uri="memory://",
)

@app.errorhandler(429)
def ratelimit_handler(e):
    error = "Too many requests. Please try again soon."
    return render_template('index.html',  error = error) 
  
# decrypt_files
for file_name in [f'data/encrypted_vectors_{term}_courses.pkl']:
    decrypt_file(file_name)
    
vector_courses = pd.read_pickle(f"data/decrypted_vectors_{term}_courses.pkl")

# modify course data (note: could be done before encryption)
clean_course_info(vector_courses)

@app.route('/')
def index():
    return render_template('index.html', current_semester = current_semester) 


# load course data
aspc_courses_ids = pd.read_csv('data/courses-2024.csv')

@app.route('/rec',methods=['POST'])
@limiter.limit("50 per day") # can modify this if needed
def getvalue():
	try:
		query = request.form['search']

		course_area = request.form['department']

		selected_days = request.form.getlist('days')  
        
		campus_list = request.form.getlist('campus')    

		df = recommend_courses(query, courses_df = vector_courses,
									 course_area= course_area,
									 campus_list= campus_list, selected_days=selected_days)
        
		# add link to rec course reviews df
		df['link'] = df.apply(lambda x: get_course_review_link(x['CourseCode']), axis=1)
		filters = get_filters(course_area, campus_list, selected_days)
		return render_template('result.html', tables = df, query = query, current_semester = current_semester, is_empty = df.empty, filters=filters)
        
	except Exception as e:
		traceback.print_exc()  # Print the error traceback
		error = "We are temporarily unable to process your request. Please contact product@aspc.pomona.edu."
		return render_template('index.html',  error = error)


def get_course_review_link(code):
      course_id = aspc_courses_ids.loc[(aspc_courses_ids['Code'] == code[:-3])]['Id']
      if len(course_id) == 0:  return None
      return f"https://pomonastudents.org/courses/{course_id.values[0]}"

if __name__ == '__main__':
    app.run(debug=True)