from flask import Flask, request, jsonify
import subprocess
from regex import find_imports

app = Flask(__name__)

@app.route("/")
def index():
    return "Hello World!"

@app.route("/code", methods=['POST'])
def run_mojo_code():
    item = request.get_json()
    try:
        imports = find_imports(item['code'])
        for imported in imports:
            subprocess.call(["python3", "-m", "pip", "install", imported], shell=True)
        with open(item['filename'], "w") as f:
            f.write(item['code']) 
            
        return jsonify({"sucess":True, "output": subprocess.check_output(["mojo", item['filename']]).decode("utf-8")}), 200
    except:
        return jsonify({"sucess":False}), 500

app.run(port=7860, debug=True)