BPLabs/route.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import os
from flask import Flask, url_for, render_template, jsonify, request, make_response, g
from flask_socketio import emit
import pdb
import csv
import io
import re
import base64
import shutil

import webview
import webbrowser
import app
import time
from threading import Thread, Event
import numpy as np
import random
from pysndfile import sndio
from scipy.optimize import minimize

from app import generate_matrix_stimulus
from matrix_test.filesystem import globDir, organiseWavs, prepareOutDir
from matrix_test_thread import MatTestThread
from pathops import dir_must_exist
from participant import Participant, find_participants, gen_participant_num

from config import server, socketio, participants


'''
General routing
'''
@server.route("/")
def landing():
    """
    Render index.html
    """
    return render_template("index.html")

@server.route('/participant_home')
def participant_homepage():
    title = "Welcome"
    paragraph = [
        "Please wait while the experimenter sets up the test..."
    ]

    try:
        return render_template("participant_home.html", title = title, paragraph=paragraph)
    except Exception as e:
        return str(e)


@server.route('/home')
def homepage():
    title = "Welcome"
    paragraph = [
        "This is the clinician view. Use the dropdown menus to access controls "
        "and feedback for the various tests available.",
        "This web app was developed for the generation of stimulus and "
        "running of experiments for the PhD project \"Predicting speech "
        "in noise performance using evoked responses\"."
    ]

    try:
        return render_template("home.html", title = title, paragraph=paragraph)
    except Exception as e:
        return str(e)


@server.route("/choose/path")
def choose_path():
    """
    Invokes a folder selection dialog
    """
    dirs = webview.create_file_dialog(webview.FOLDER_DIALOG)
    if dirs and len(dirs) > 0:
        directory = dirs[0]
        if isinstance(directory, bytes):
            directory = directory.decode("utf-8")

        response = {"status": "ok", "directory": directory}
    else:
        response = {"status": "cancel"}

    return jsonify(response)


@server.route("/fullscreen")
def fullscreen():
    webview.toggle_fullscreen()
    return jsonify({})

'''
Participant routing
'''
@server.route('/participant/manage')
def manage_participant_page():
    # Find all pre-existing participants
    participants = find_participants()
    return render_template("manage_participants.html", part_keys=participants.keys())

@server.route('/participant/create')
def create_participant_page():
    # Find all pre-existing participants
    participants = find_participants()
    part_num = gen_participant_num(participants)
    return render_template("create_participant.html", num=part_num)

@server.route('/participant/create/submit', methods=["POST"])
def create_participant_submit():
    data = request.form
    key = "participant_{}".format(data['number'])
    participants[key] = Participant(participant_dir="./participant_data/{}".format(key), **data)
    participants[key].save("info")
    return render_template("manage_participants.html", part_keys = participants.keys())

'''
EEG routing
'''
@server.route('/eeg')
def eeg_setup():
    participants = find_participants()
    return render_template("eeg_setup.html", part_keys=participants.keys())

'''
Matrix behavioral test routing
'''
@server.route('/matrix_test')
def matrix_test_setup():
    participants = find_participants()
    return render_template("matrix_test_setup.html", part_keys=participants.keys())

@server.route('/matrix_test/run')
def run_matrix_test():
    return render_template("mat_test_run.html")

@server.route('/matrix_test/complete')
def mat_end():
    return render_template("mat_test_end.html")

@server.route('/matrix_test/clinician/control')
def clinician_control_mat():
    return render_template("mat_test_clinician_view.html")

@server.route('/matrix_test/clinician/complete')
def clinician_mat_end():
    return render_template("mat_test_clinician_end.html")

@server.route('/matrix_test/stimulus_generation')
def matDecStim():
    return render_template("matrix_decode_stim.html")

'''
Click stimulus routing
'''
@server.route('/click_stim')
def clickStim():
    return render_template("click_stim.html")


'''
/da/ stimulus routing
'''
@server.route('/da_stim')
def daStim():
    return render_template("da_stim.html")