Update main.py

main.py

@@ -1,19 +1,14 @@
-from fastapi import FastAPI, Request
+from fastapi import FastAPI, Request, Response
+from pydantic import BaseModel
 
-import subprocess, json
+import subprocess
 from regex import find_imports
 
-
-
-
 app = FastAPI()
 
 @app.post("/code")
-async def run_mojo_code(request:Request) -> dict:
-    data_as_str = await request.json()
-    print(data_as_str)
-    print(type(data_as_str))
-    data = data_as_str
+async def run_mojo_code(request:Request) -> Response:
+    data = await request.json()
     code = data["code"]
     filename = data["filename"]
     
@@ -24,6 +19,6 @@ async def run_mojo_code(request:Request) -> dict:
         with open(filename, "w") as f:
             f.write(code) 
             
-        return {"sucess":True, "output": subprocess.check_output(["mojo", filename]).decode("utf-8")}, 200
+        return Response(content={"sucess":True, "output": subprocess.check_output(["mojo", filename]).decode("utf-8")}, status_code=200)
     except:
-        return {"sucess":False}, 500
+        return Response(content={"sucess":False}, status_code=500)