pyledriver/webInterface.py

import logging
from auxilary import async
from flask import Flask, render_template, Response, Blueprint, redirect, url_for
from flask_wtf import FlaskForm
from wtforms.fields import StringField, SubmitField
from wtforms.validators import InputRequired

logger = logging.getLogger(__name__)

# gag the flask logger unless it has something useful to say
werkzeug = logging.getLogger('werkzeug')
werkzeug.setLevel(logging.ERROR)

class TTSForm(FlaskForm):
	tts = StringField(validators=[InputRequired()])
	submitTTS = SubmitField('Speak')

# TODO: fix random connection fails (might be an nginx thing)

@async(daemon=True)
def initWebInterface(stateMachine):
	siteRoot = Blueprint('siteRoot', __name__, static_folder='static', static_url_path='')

	@siteRoot.route('/', methods=['GET', 'POST'])
	@siteRoot.route('/index', methods=['GET', 'POST'])
	def index():
		ttsForm = TTSForm()
		
		if ttsForm.validate_on_submit() and ttsForm.submitTTS.data:
			stateMachine.soundLib.speak(ttsForm.tts.data)
			#~ ttsQueue.put_nowait(ttsForm.tts.data)
			return redirect(url_for('siteRoot.index'))

		return render_template(
			'index.html',
			ttsForm=ttsForm,
			state=stateMachine.currentState
		)

	app = Flask(__name__)
	app.secret_key = '3276d68dac56985bea352325125641ff'
	app.register_blueprint(siteRoot, url_prefix='/pyledriver')

	# TODO: not sure exactly how threaded=True works, intended to enable
	# multiple connections. May want to use something more robust w/ camera
	# see here: https://blog.miguelgrinberg.com/post/video-streaming-with-flask
	logger.info('Starting web interface')
	app.run(debug=False, threaded=True)