app.py

from flask import Flask, request, jsonify, render_template, session, redirect, url_for
import mysql.connector
from mysql.connector import Error
import pandas as pd
from database import create_connection

app = Flask(__name__)

app.secret_key = 'your_super_secret_key_12345'



@app.route('/')
def home():
  
    if 'user_id' in session:
        return redirect(url_for('dashboard'))
   
    return redirect(url_for('login_page'))
@app.route('/login')
def login_page():
    return render_template('login.html')

@app.route('/register')
def register_page():
    return render_template('register.html')
    
@app.route('/dashboard')
def dashboard():
    if 'user_id' not in session:
        return redirect(url_for('login_page'))
    return render_template('dashboard.html', user_name=session.get('user_name'))

@app.route('/logout')
def logout():
    session.clear()
    return redirect(url_for('login_page'))



@app.route('/api/login', methods=['POST'])
def login_api():
    data = request.get_json()
    conn = create_connection()
    if not conn: return jsonify({'error': 'Database connection failed'}), 500
    cursor = conn.cursor(dictionary=True)
    cursor.execute("SELECT * FROM Students WHERE Email = %s AND Password = %s", (data['email'], data['password']))
    user = cursor.fetchone()
    cursor.close()
    conn.close()
    if user:
        # login is successful, store the user's ID and name in the session.
        session['user_id'] = user['ID']
        session['user_name'] = user['Name']
        return jsonify({'message': 'Login successful'})
    else:
        return jsonify({'error': 'Invalid credentials'}), 401

@app.route('/api/register', methods=['POST'])
def register_api():
    data = request.get_json()
    conn = create_connection()
    if not conn: return jsonify({'error': 'Database connection failed'}), 500
    cursor = conn.cursor()
    try:
        cursor.execute("INSERT INTO Students (Name, Email, Password) VALUES (%s, %s, %s)",
                       (data['name'], data['email'], data['password']))
        conn.commit()
        return jsonify({'message': 'User registered successfully!'})
    except Error as e:
        return jsonify({'error': str(e)}), 400
    finally:
        cursor.close()
        conn.close()

@app.route('/api/subjects', methods=['GET', 'POST'])
def subjects_api():
    if 'user_id' not in session:
        return jsonify({'error': 'Not authenticated'}), 401

    student_id = session['user_id']
    conn = create_connection()
    if not conn: return jsonify({'error': 'Database connection failed'}), 500
    cursor = conn.cursor(dictionary=True)

    if request.method == 'POST':
        data = request.get_json()
        try:
            cursor.execute("INSERT INTO Subjects (Name, Difficulty_Level, Student_ID) VALUES (%s, %s, %s)",
                           (data['name'], data['difficulty'], student_id))
            conn.commit()
            return jsonify({'message': 'Subject added successfully!'})
        except Error as e:
            return jsonify({'error': str(e)}), 400
        finally:
            cursor.close()
            conn.close()
    
    if request.method == 'GET':
        try:
            cursor.execute("SELECT * FROM Subjects WHERE Student_ID = %s", (student_id,))
            subjects = cursor.fetchall()
            return jsonify(subjects)
        except Error as e:
            return jsonify({'error': str(e)}), 400
        finally:
            cursor.close()
            conn.close()

@app.route('/api/study-history', methods=['POST'])
def study_history_api():
    if 'user_id' not in session:
        return jsonify({'error': 'Not authenticated'}), 401

    student_id = session['user_id']
    data = request.get_json()
    conn = create_connection()
    if not conn: return jsonify({'error': 'Database connection failed'}), 500
    cursor = conn.cursor()
    try:
        cursor.execute("INSERT INTO Study_History (Student_ID, Subject_ID, Date, Hours_Studied) VALUES (%s, %s, %s, %s)",
                       (student_id, data['subject_id'], data['date'], data['hours']))
        conn.commit()
        return jsonify({'message': 'Study session logged successfully!'})
    except Error as e:
        return jsonify({'error': str(e)}), 400
    finally:
        cursor.close()
        conn.close()


@app.route('/api/assignments', methods=['GET', 'POST'])
def assignments_api():
    if 'user_id' not in session:
        return jsonify({'error': 'Not authenticated'}), 401

    student_id = session['user_id']
    conn = create_connection()
    if not conn: return jsonify({'error': 'Database connection failed'}), 500
    cursor = conn.cursor(dictionary=True)

    if request.method == 'POST':
        data = request.get_json()
        try:
            cursor.execute("INSERT INTO Assignments (Subject_ID, Due_Date, Status, Student_ID) VALUES (%s, %s, %s, %s)",
                           (data['subject_id'], data['due_date'], 'Pending', student_id))
            conn.commit()
            return jsonify({'message': 'Assignment added successfully!'})
        except Error as e:
            return jsonify({'error': str(e)}), 400
        finally:
            cursor.close()
            conn.close()

    if request.method == 'GET':
        query = """
            SELECT s.Name as Subject_Name, a.Due_Date, a.Status
            FROM Assignments a
            JOIN Subjects s ON a.Subject_ID = s.Subject_ID
            WHERE a.Student_ID = %s ORDER BY a.Due_Date ASC
        """
        try:
            cursor.execute(query, (student_id,))
            assignments = cursor.fetchall()
            for assignment in assignments:
                assignment['Due_Date'] = assignment['Due_Date'].strftime('%Y-%m-%d')
            return jsonify(assignments)
        except Error as e:
            return jsonify({'error': str(e)}), 400
        finally:
            cursor.close()
            conn.close()


@app.route('/api/recommendations')
def recommendations_api():
    if 'user_id' not in session:
        return jsonify({'error': 'Not authenticated'}), 401
        
    student_id = session['user_id']
    conn = create_connection()
    if not conn: return jsonify({'error': 'Database connection failed'}), 500
    
    query = "SELECT Hours_Studied FROM Study_History WHERE Student_ID = %s"
    try:
        df = pd.read_sql(query, conn, params=(student_id,))
        if len(df) < 2:
            return jsonify({'recommendations': 'Log at least two study sessions to get a personalized plan.'})

        baseline_hours = df['Hours_Studied'].mean()
        multipliers = {'Hard': 1.5, 'Medium': 1.0, 'Easy': 0.75}
        recommendations = {
            'Easy': f"{baseline_hours * multipliers['Easy']:.2f} hours",
            'Medium': f"{baseline_hours * multipliers['Medium']:.2f} hours",
            'Hard': f"{baseline_hours * multipliers['Hard']:.2f} hours"
        }
        return jsonify({'recommendations': recommendations})
    except Exception as e:
        return jsonify({'error': str(e)}), 500
    finally:
        if conn and conn.is_connected():
            conn.close()


@app.route('/api/chart-data')
def chart_data_api():
    if 'user_id' not in session:
        return jsonify({'error': 'Not authenticated'}), 401

    student_id = session['user_id']
    conn = create_connection()
    if not conn: return jsonify({'error': 'Database connection failed'}), 500

    query = """
        SELECT s.Name, SUM(sh.Hours_Studied) as Total_Hours
        FROM Study_History sh JOIN Subjects s ON sh.Subject_ID = s.Subject_ID
        WHERE sh.Student_ID = %s
        GROUP BY s.Name ORDER BY Total_Hours DESC
    """
    try:
        df = pd.read_sql(query, conn, params=(student_id,))
        if df.empty:
            return jsonify({'error': 'No study data available to create charts.'}), 404
        chart_data = {'labels': df['Name'].tolist(), 'data': df['Total_Hours'].tolist()}
        return jsonify(chart_data)
    except Error as e:
        return jsonify({'error': str(e)}), 500
    finally:
        if conn and conn.is_connected():
            conn.close()

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