from flask import Flask, render_template
from flask_executor import Executor
from flask_socketio import SocketIO, emit
from flask_cors import cross_origin, CORS
from main import run
from gevent import monkey

monkey.patch_all(ssl=False)
app = Flask(__name__)
app.config['SECRET_KEY'] = 'secret!'
socketio = SocketIO(app, cors_allowed_origins="*", async_mode='gevent', logger=True)
cors = CORS(app)

executor = Executor(app)

executor.init_app(app)
app.config['EXECUTOR_MAX_WORKERS'] = 10

@app.route('/')
def index():
    return render_template('index.html')


@socketio.on('message')
def handle_message(data):
    question = data['question']
    print("question: " + question)

    if executor.futures:
        emit('response', {'response': 'Server is busy, please try again later'})
        return

    try:
        future = executor.submit(run, question)
        response = future.result()
        emit('response', {'response': response})
    except Exception as e:
        traceback.print_exc()
        emit('response', {'response': 'Server is busy. Please try again later.'})

if __name__ == '__main__':
    socketio.run(app, host="0.0.0.0", port=7860)