from flask import request, redirect, url_for, flash, render_template, Blueprint, session
from flask_security import roles_required, roles_accepted

from .basic_forms import SessionDefaultsForm

basic_bp = Blueprint('basic_bp', __name__)


@basic_bp.route('/', methods=['GET', ])
def index():
    return render_template('index.html')


@basic_bp.route('/session_defaults', methods=['GET', 'POST'])
@roles_accepted('Super User', 'Tenant Admin')
def session_defaults():
    form = SessionDefaultsForm()

    if form.validate_on_submit():
        session['default_language'] = form.default_language.data
        session['default_embedding_model'] = form.default_embedding_model.data
        session['default_llm_model'] = form.default_llm_model.data

    return render_template('basic/session_defaults.html', form=form)