Update main.py

main.py

@@ -1,25 +1,25 @@
-from fastapi import FastAPI
-from pydantic import BaseModel
-
+from flask import Flask, request, jsonify
 import subprocess
 from regex import find_imports
-class Item(BaseModel):
-    code: str
-    filename: str = ''
-
 
+app = Flask(__name__)
 
-app = FastAPI()
+@app.route("/")
+def index():
+    return "Hello World!"
 
-@app.post("/code")
-def run_mojo_code(item:Item):
+@app.route("/code", methods=['POST'])
+def run_mojo_code():
+    item = request.get_json()
     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 {"sucess":True, "output": subprocess.check_output(["mojo", item.filename]).decode("utf-8")}, 200
+        return jsonify({"sucess":True, "output": subprocess.check_output(["mojo", item['filename']]).decode("utf-8")}), 200
     except:
-        return {"sucess":False}, 500
+        return jsonify({"sucess":False}), 500
+
+app.run(port=7860, debug=True)