Update Dockerfile

Dockerfile

@@ -1,49 +1,35 @@
-# ── OpenEnv: Split-Brain Collapse — GPU Training Space ────────────────────────
-# This Dockerfile runs the GRPO curriculum training script.
-# Deploy as a separate HF Space with GPU hardware (A10G Small recommended).
-#
-# Env variables to set in Space Secrets:
-#   HF_TOKEN     — your HF write token (to push adapter + plots)
-#   HF_USERNAME  — your HF username (default: soonvalley04)
-# ──────────────────────────────────────────────────────────────────────────────
+# ── OpenEnv: Distributed Data Cluster Triage ──────────────────────────────
+# Serves:
+#   • OpenEnv HTTP API:  POST /reset, POST /step, GET /state, GET /health
+#   • HTML Frontend:     GET  /  (interactive dashboard)
+# Port: 7860 (required for Hugging Face Spaces)
+# ──────────────────────────────────────────────────────────────────────────
 
-FROM python:3.10-slim
+FROM python:3.10.14-slim-bookworm
 
 WORKDIR /app
 
-# System dependencies
-RUN apt-get update && apt-get install -y --no-install-recommends \
-    git curl gcc g++ build-essential \
-    && apt-get clean && rm -rf /var/lib/apt/lists/*
-
-# Install PyTorch from the official CUDA 12.1 wheel index (safest for HF Spaces A10G)
-# We do not pin versions so pip can cleanly resolve PyTorch 2.6.0 + Torchvision 0.21.0
-# Note: Use cu128 or cu130 as cu121 is outdated for Torch 2.10+
-RUN pip install --no-cache-dir --upgrade pip && \
-    pip install --no-cache-dir \
-    "torch==2.10.0" \
-    "torchvision==0.25.0" \
-    "torchaudio==2.10.0" \
-    --index-url https://download.pytorch.org/whl/cu128
-
-# ── 2. Install Unsloth + Training Stack (Correct Dependencies) ──────────────
-RUN pip install --no-cache-dir \
-    "unsloth @ git+https://github.com/unslothai/unsloth.git" \
-    "unsloth_zoo" \
-    "trl" \
-    "datasets" \
-    "matplotlib" \
-    "pydantic" \
-    "networkx" \
-    "python-dotenv" \
-    "huggingface_hub" \
-    "openai" \
-    "bitsandbytes" \
-    "xformers" \
-    "torchao>=0.16.0"
-
-# Copy repo source
+# Apply latest security patches
+RUN apt-get update && apt-get upgrade -y \
+    && apt-get install -y --no-install-recommends curl \
+    && apt-get clean \
+    && rm -rf /var/lib/apt/lists/*
+
+# Install Python dependencies first (Docker cache layer)
+COPY requirements.txt .
+RUN pip install --no-cache-dir --upgrade pip \
+    && pip install --no-cache-dir -r requirements.txt
+
+# Copy environment source
 COPY . .
 
+# Health check — confirms the FastAPI server is live
+HEALTHCHECK --interval=30s --timeout=10s --start-period=15s --retries=3 \
+    CMD curl -f http://localhost:7860/health || exit 1
+
+# Expose Hugging Face Spaces port
+EXPOSE 7860
+
 # Run the FastAPI server (app:app = the `app` FastAPI instance in app.py)
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860", "--workers", "1"]
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860", "--workers", "1"]
+