cet / app.py
KunalThakare279's picture
Update app.py
4504e21 verified
from flask import Flask, render_template, request, jsonify
from model import get_colleges, get_courses_for_zone # Import the model function
import pandas as pd
import pickle
app = Flask(__name__)
zones = ['Amravati', 'Pune', 'Aurangabad', 'Mumbai & Thane', 'Konkan', 'Nagpur', 'Nashik']
seat_levels = ['Home', 'State', 'Other Than Home']
categories = ['OPEN', 'SC', 'ST', 'VJ', 'NT1', 'NT2', 'NT3', 'OBC', 'DEFOPEN',
'TFW', 'DEFROBC', 'EW', 'PWDOPEN', 'PWDRSC', 'DEFRSC', 'PWDROBC',
'M', 'DEFOBC', 'DEFRNT1', 'DEFRNT2', 'ORPHA', 'PWDRVJ', 'PWDOBC',
'PWDRNT1', 'DEFRNT3', 'DEFRVJ', 'DEFSC', 'PWDRNT2', 'PWDSC',
'PWDRNT3', 'PWDRST', 'DEFRST', 'PWDROB']
@app.route('/', methods=['GET', 'POST'])
def index():
colleges_data = None
if request.method == 'POST':
# Get data from the form
marks = float(request.form['marks'])
zone = request.form['zone']
course = request.form['course']
category = request.form['category']
gender = request.form['gender']
seat_level = request.form['seat_level']
round = 1
print(marks, zone, course, category, gender, seat_level, round)
# Call the get_colleges function to predict the list of colleges
colleges_data = get_colleges(marks, zone, course, category, gender, seat_level, round)
return render_template('index.html', colleges_data=colleges_data, zones = zones, categories = categories, seat_levels = seat_levels)
@app.route('/get_courses', methods=['POST'])
def get_courses():
zone = request.json['zone']
courses = get_courses_for_zone(zone)
courses = courses.tolist()
return jsonify(courses)
if __name__ == '__main__':
app.run(debug=True)