Upload 6 files

Dockerfile

@@ -1,36 +1,19 @@
-# ─── Build stage ─────────────────────────────────────────────────────────────────
-FROM python:3.11-slim AS builder
-
-WORKDIR /build
-
-# Install dependencies into a prefix we'll copy to the final image
-COPY requirements.txt .
-RUN pip install --no-cache-dir --prefix=/install -r requirements.txt
-
-
-# ─── Runtime stage ────────────────────────────────────────────────────────────────
 FROM python:3.11-slim
 
-# Hugging Face Spaces expects the app on port 7860
 ENV PORT=7860
 ENV PYTHONUNBUFFERED=1
 ENV PYTHONDONTWRITEBYTECODE=1
 
 WORKDIR /app
 
-# Copy pre-installed packages from builder
-COPY --from=builder /install /usr/local
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
 
-# Copy application code
-COPY env.py      .
-COPY main.py     .
-COPY inference.py .
+COPY . .
 
-# HF Spaces: non-root user for safety
 RUN useradd -m -u 1000 appuser && chown -R appuser /app
 USER appuser
 
 EXPOSE 7860
 
-# Increase workers for concurrent evaluation runs
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860", "--workers", "1"]
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860", "--workers", "1"]

main.py

@@ -2,6 +2,7 @@
 API Gateway Defender — FastAPI Server
 =====================================
 Exposes the OpenEnv-compliant HTTP API for the environment.
+
 Endpoints
 ---------
   POST /reset       — Start a new episode
@@ -90,13 +91,13 @@ async def reset(
 ) -> Dict[str, Any]:
     """
     Start a new episode.
+
     Accepts ALL of these formats (validator may use any):
       - JSON body:    {"task_id": "easy"}
       - Query param:  POST /reset?task_id=easy
       - Empty body:   POST /reset  (defaults to "easy")
       - No body at all: POST /reset  (defaults to "easy")
     """
-    # Priority: JSON body > query param > default "easy"
     tid = (req.task_id if req else None) or task_id or "easy"
     try:
         obs: Observation = _env.reset(task_id=tid)
@@ -109,7 +110,9 @@ async def reset(
 def step(action: Action) -> Dict[str, Any]:
     """
     Submit one firewall rule.
+
     Returns StepResult: {observation, reward, done, info}
+
     Reward score: 0.0–1.0
       = detection_rate − (false_positive_rate × 5)
       = 0.0 if false positive rate > 10%

pyproject.toml

@@ -1,18 +1,18 @@
-[project]
-name = "api-gateway-defender"
-version = "1.0.0"
-description = "OpenEnv RL environment for API gateway defense"
-requires-python = ">=3.11"
-dependencies = [
-    "fastapi>=0.104.0",
-    "uvicorn[standard]>=0.24.0",
-    "pydantic>=2.0.0",
-    "openenv-core>=0.2.0",
-]
-
-[project.scripts]
-server = "main:app"
-
-[tool.openenv]
-env_id = "api-gateway-defender"
+[project]
+name = "api-gateway-defender"
+version = "1.0.0"
+description = "OpenEnv RL environment for API gateway defense"
+requires-python = ">=3.11"
+dependencies = [
+    "fastapi>=0.104.0",
+    "uvicorn[standard]>=0.24.0",
+    "pydantic>=2.0.0",
+    "openenv-core>=0.2.0",
+]
+
+[project.scripts]
+server = "main:app"
+
+[tool.openenv]
+env_id = "api-gateway-defender"
 entry_point = "main:app"

requirements.txt

@@ -1,5 +1,4 @@
-fastapi
-uvicorn
-pydantic
-requests
-openai
+fastapi>=0.104.0
+uvicorn[standard]>=0.24.0
+pydantic>=2.0.0
+openenv-core>=0.2.0