@app.route('/register', methods=['GET', 'POST'])
def register():
    if request.method == 'POST':
        reg_username = request.form.get['username']
        reg_password = request.form.get['password']
        # Check if username exists in database
        query = "SELECT username FROM users WHERE username = :username"
        if db.execute(query, {'username': reg_username}).first():
            flash('Username already exists')
            return render_template('register.html')
        else:
            new_user = User(reg_username, reg_password)
            db.session.add(new_user)
            db.session.commit()
            session['username']=reg_username
            return redirect(url_for('login'))
    return render_template('register.html')

@app.route('/register', methods=['GET', 'POST'])
def register():
    if request.method == 'POST':
        reg_username = request.form['username']
        reg_password = request.form['password']
        # Check if username exists in database
        query = "SELECT username FROM users WHERE username = :username"
        if db.execute(query, {'username': reg_username}).first():
            flash('Username already exists')
            return render_template('register.html')
        else:
            new_user = User(reg_username, reg_password)
            db.session.add(new_user)
            db.session.commit()
            session['username']=reg_username
            return redirect(url_for('login'))
    return render_template('register.html')