Update app.py

app.py

@@ -7,76 +7,75 @@ from pydantic import BaseModel, Field
 
 # Defines the structure of the request body
 class ExecutionRequest(BaseModel):
-    package_name: str = Field(..., example="numpy", description="The name of the pip package to install.")
-    script_content: str = Field(..., example="import numpy as np; print(np.array([1, 2, 3]))", description="The Python script content to execute.")
+    package_name: str = Field(
+        ...,
+        example="numpy requests beautifulsoup4",
+        description="One or more pip packages to install (space or comma separated)."
+    )
+    script_content: str = Field(
+        ...,
+        example="import numpy as np; print(np.array([1, 2, 3]))",
+        description="The Python script content to execute."
+    )
 
 app = FastAPI()
 
 @app.post("/execute/")
 def install_and_execute(request: ExecutionRequest):
     """
-    Installs a package to a temporary directory and executes a Python script.
-
-    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 temporary directory and its contents are automatically removed after the
-    request is complete.
+    Installs one or more packages into a temporary directory and executes a Python script.
+    Supports multiple packages separated by spaces or commas.
     """
-    # Use a temporary directory that will be automatically cleaned up
     with tempfile.TemporaryDirectory() as temp_dir:
         pip_output = ""
         script_output = ""
         error = ""
 
         try:
-            # 1. Install the package to the temporary directory
+            # 1. Create the install directory
             install_path = os.path.join(temp_dir, "packages")
             os.makedirs(install_path)
 
+            # 2. Split package names by space or comma
+            packages = [pkg.strip() for pkg in request.package_name.replace(",", " ").split() if pkg.strip()]
+
+            # 3. Install the packages to the temporary directory
             pip_output = subprocess.check_output(
                 [
                     sys.executable,
                     "-m",
                     "pip",
                     "install",
-                    request.package_name,
+                    *packages,  # multiple packages supported
                     "--target",
-                    install_path, # Install here
+                    install_path,
                 ],
                 stderr=subprocess.STDOUT,
                 text=True
             )
 
-            # 2. Create a temporary file for the user's script
+            # 4. Create a temporary script file
             script_path = os.path.join(temp_dir, "script.py")
             with open(script_path, 'w') as temp_script:
                 temp_script.write(request.script_content)
 
-            # 3. Prepare the environment for the script execution
-            #    This ensures the script can find the package in `install_path`
+            # 5. Prepare the environment with custom PYTHONPATH
             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
+            # 6. Execute the Python script
             script_output = subprocess.check_output(
                 [sys.executable, script_path],
                 stderr=subprocess.STDOUT,
                 text=True,
-                env=exec_env # Use the modified environment
+                env=exec_env
             )
 
         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
-        }
+        }