Upload 3 files

Dockerfile

@@ -0,0 +1,9 @@
+FROM ubuntu:latest
+
+RUN apt-get update
+RUN apt-get install -y curl && apt-get install -y python3 && apt-get install -y python3-pip
+RUN curl https://get.modular.com | \ MODULAR_AUTH=mut_e87f7861fb9a4d4aa311afb0491b0398 \ sh -
+RUN modular install mojo
+
+COPY . .
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

main.py

@@ -0,0 +1,19 @@
+from fastapi import FastAPI
+import subprocess, importlib
+from tools.regex import find_imports
+
+
+app = FastAPI()
+
+@app.post("/code")
+def run_mojo_code(code:str, filename:str) -> dict[str]:
+    try:
+        imports = find_imports(code)
+        for imported in imports:
+            subprocess.call(["python3", "-m", "pip", "install", imported], shell=True)
+        with open(filename, "w") as f:
+            f.write(code) 
+            
+        return {"sucess":True, "output": subprocess.check_output(["mojo", filename]).decode("utf-8")}, 200
+    except:
+        return {"sucess":False}, 500

regex.py

@@ -0,0 +1,19 @@
+import re
+
+def find_imports(code):
+    pattern = r'Python\.import_module\("([^"]*)"\)'
+    matches = re.findall(pattern, code)
+    return matches
+
+if __name__ == "__main__":
+    code = '''
+    from python import Python
+    Python.import_module("numpy")
+    Python.import_module("pandas")
+    Python.import_module("matplotlib")
+
+    def main():
+        print("hello world")
+    '''
+
+    print(find_imports(code))  # Output: ['numpy', 'pandas', 'matplotlib']