Create app.py

app.py

@@ -0,0 +1,66 @@
+from fastapi import FastAPI, Form
+import subprocess
+import sys
+import tempfile
+import os
+import importlib.util
+
+CACHE_DIR = "/home/user/.cache_packages"
+
+# Ensure the cache dir exists
+os.makedirs(CACHE_DIR, exist_ok=True)
+
+# Add cache dir to Python path
+if CACHE_DIR not in sys.path:
+    sys.path.insert(0, CACHE_DIR)
+
+app = FastAPI()
+
+def install_package_if_needed(package_name: str):
+    """Install a package to CACHE_DIR only if not already installed."""
+    if importlib.util.find_spec(package_name) is None:
+        subprocess.check_call([
+            sys.executable, "-m", "pip", "install",
+            "--target", CACHE_DIR, package_name
+        ])
+
+@app.post("/run")
+def run_code(code: str = Form(...)):
+    try:
+        lines = code.strip().split("\n")
+        
+        # Separate pip installs from Python code
+        pip_commands = [line.strip() for line in lines if line.strip().startswith("pip install")]
+        python_code_lines = [line for line in lines if not line.strip().startswith("pip install")]
+
+        # Handle pip installs
+        for cmd in pip_commands:
+            parts = cmd.split()
+            for pkg in parts[2:]:  # Skip 'pip' and 'install'
+                try:
+                    install_package_if_needed(pkg)
+                except subprocess.CalledProcessError as e:
+                    return {"stdout": "", "stderr": f"Error installing: {pkg}\n{str(e)}"}
+
+        # Save Python code to temp file
+        python_code = "\n".join(python_code_lines)
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".py") as temp_file:
+            temp_file.write(python_code.encode())
+            temp_path = temp_file.name
+
+        # Run Python code
+        result = subprocess.run(
+            [sys.executable, temp_path],
+            capture_output=True,
+            text=True
+        )
+
+        os.unlink(temp_path)  # Clean up
+
+        return {
+            "stdout": result.stdout,
+            "stderr": result.stderr
+        }
+
+    except Exception as e:
+        return {"error": str(e)}