Update main.py

main.py

@@ -1,25 +1,25 @@
-from flask import Flask, request, jsonify
+from fastapi import FastAPI
+from pydantic import BaseModel
+
 import subprocess
 from regex import find_imports
+class Item(BaseModel):
+    code: str
+    filename: str = ''
+
 
-app = Flask(__name__)
 
-@app.route("/")
-def index():
-    return "Hello World!"
+app = FastAPI()
 
-@app.route("/code", methods=['POST'])
-def run_mojo_code():
-    item = request.get_json()
+@app.post("/code")
+def run_mojo_code(item:Item) -> dict:
     try:
-        imports = find_imports(item['code'])
+        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']) 
+        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
+        return {"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)
+        return {"sucess":False}, 500