Update app.py

app.py

@@ -15,51 +15,68 @@ app = FastAPI()
 @app.post("/execute/")
 def install_and_execute(request: ExecutionRequest):
     """
-    Installs a specified pip package and then executes a provided Python script.
+    Installs a package to a temporary directory and executes a Python script.
 
-    This endpoint first uses pip to install the package named in the `package_name`
-    field. It then writes the `script_content` to a temporary file and
-    executes it using the same Python interpreter that is running the FastAPI
-    application.
+    This endpoint addresses permission errors in restricted environments like
+    Hugging Face Spaces by:
+    1. Creating a temporary directory for the request.
+    2. Installing the specified pip package into this temporary directory using `pip install --target`.
+    3. Setting the PYTHONPATH environment variable to include this directory.
+    4. Executing the provided script, which can now import the installed package.
 
-    The output from both the pip installation and the script execution is
-    captured and returned in the JSON response. Any errors during the process
-    are also captured and returned.
+    The temporary directory and its contents are automatically removed after the
+    request is complete.
     """
-    pip_output = ""
-    script_output = ""
-    error = ""
+    # Use a temporary directory that will be automatically cleaned up
+    with tempfile.TemporaryDirectory() as temp_dir:
+        pip_output = ""
+        script_output = ""
+        error = ""
 
-    try:
-        # Execute the pip install command
-        pip_output = subprocess.check_output(
-            [sys.executable, "-m", "pip", "install", request.package_name],
-            stderr=subprocess.STDOUT,
-            text=True
-        )
+        try:
+            # 1. Install the package to the temporary directory
+            install_path = os.path.join(temp_dir, "packages")
+            os.makedirs(install_path)
+
+            pip_output = subprocess.check_output(
+                [
+                    sys.executable,
+                    "-m",
+                    "pip",
+                    "install",
+                    request.package_name,
+                    "--target",
+                    install_path, # Install here
+                ],
+                stderr=subprocess.STDOUT,
+                text=True
+            )
 
-        # Create a temporary file with a .py extension to save the script content
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".py", mode='w') as temp_script:
-            temp_script.write(request.script_content)
-            temp_script_path = temp_script.name
+            # 2. Create a temporary file for the user's script
+            script_path = os.path.join(temp_dir, "script.py")
+            with open(script_path, 'w') as temp_script:
+                temp_script.write(request.script_content)
 
-        try:
-            # Execute the Python script
+            # 3. Prepare the environment for the script execution
+            #    This ensures the script can find the package in `install_path`
+            exec_env = os.environ.copy()
+            # Add the install path to the Python Path
+            exec_env["PYTHONPATH"] = install_path + os.pathsep + exec_env.get("PYTHONPATH", "")
+
+            # 4. Execute the Python script using the modified environment
             script_output = subprocess.check_output(
-                [sys.executable, temp_script_path],
+                [sys.executable, script_path],
                 stderr=subprocess.STDOUT,
-                text=True
+                text=True,
+                env=exec_env # Use the modified environment
             )
-        finally:
-            # Clean up the temporary file
-            os.remove(temp_script_path)
 
-    except subprocess.CalledProcessError as e:
-        # If any command fails, capture the output as an error
-        error = e.output
+        except subprocess.CalledProcessError as e:
+            # If any command fails, capture its output as an error
+            error = e.output
 
-    return {
-        "pip_install_output": pip_output,
-        "script_execution_output": script_output,
-        "error_output": error
-    }
+        return {
+            "pip_install_output": pip_output,
+            "script_execution_output": script_output,
+            "error_output": error
+        }