from flask import Blueprint, render_template
from website import db, Users
from flask import Blueprint, render_template, request, redirect, session
from localStoragePy import localStoragePy
from website.loginDecorator import login_required
from website.Map import validate_page_access
import logging

Diagnosis = Blueprint('Diagnosis', __name__)
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

##### ADD/ADHD BRANCHES #####


@Diagnosis.route('AD_V1')
@login_required
def AD_V1():
    redirect_check = validate_page_access(session.get('page'), ['/Green', '/Pink', '/AD_V3','/AD_V1'])
    if redirect_check:
        return redirect_check
    session['page'] = '/AD_V1'
    logger.info(f"Accessed page: {session.get('page')}")
    return render_template("Step1/Diagnosis/ADHD/AD_V1.html")



@Diagnosis.route('AD_V3')
@login_required
def AD_V3():
    redirect_check = validate_page_access(session.get('page'), ['/DX_V1', '/AD_V1','/AD_V3'])
    if redirect_check:
        return redirect_check
    session['page'] = '/AD_V3'
    logger.info(f"Accessed page: {session.get('page')}")
    return render_template("Step1/Diagnosis/ADHD/AD_V3.html")

##### DYSLEXIA BRANCHES #####

@Diagnosis.route('Dyslexie')
@login_required
def Dyslexie():
    redirect_check = validate_page_access(session.get('page'), ['/DX_V1', '/Bolded','/Dyslexie'])
    if redirect_check:
        return redirect_check
    session['page'] = '/Dyslexie'
    logger.info(f"Accessed page: {session.get('page')}")
    return render_template("Step1/Diagnosis/Dyslexia/Dyslexie.html")


@Diagnosis.route('DX_V1')
@login_required
def DX_V1():
    redirect_check = validate_page_access(session.get('page'), ['/Dyslexie', '/AD_V3','/DX_V1'])
    if redirect_check:
        return redirect_check
    session['page'] = '/DX_V1'
    logger.info(f"Accessed page: {session.get('page')}")
    return render_template("Step1/Diagnosis/Dyslexia/DX_V1.html")