Update main.py

main.py

@@ -3,23 +3,21 @@ from pydantic import BaseModel
 
 import subprocess
 from regex import find_imports
-class Item(BaseModel):
-    code: str
-    filename: str = ''
+
 
 
 
 app = FastAPI()
 
 @app.post("/code")
-def run_mojo_code(item:Item) -> dict:
+def run_mojo_code(code: str, filename: str = '') -> dict:
     try:
-        imports = find_imports(item.code)
+        imports = find_imports(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(filename, "w") as f:
+            f.write(code) 
             
-        return {"sucess":True, "output": subprocess.check_output(["mojo", item.filename]).decode("utf-8")}, 200
+        return {"sucess":True, "output": subprocess.check_output(["mojo", filename]).decode("utf-8")}, 200
     except:
         return {"sucess":False}, 500