Update main.py

main.py

@@ -1,4 +1,4 @@
-from fastapi import FastAPI
+from fastapi import FastAPI, Request
 from pydantic import BaseModel
 
 import subprocess
@@ -10,7 +10,11 @@ from regex import find_imports
 app = FastAPI()
 
 @app.post("/code")
-def run_mojo_code(code: str, filename: str = '') -> dict:
+def run_mojo_code(request:Request) -> dict:
+    data = request.json()
+    code = data["code"]
+    filename = data["filename"]
+    
     try:
         imports = find_imports(code)
         for imported in imports: