fix: unified Dockerfile with entrypoint for server/training mode

- Single Dockerfile handles both UI server and GRPO training
- OPENGRID_MODE=training runs run_training.py
- OPENGRID_MODE=server (default) runs uvicorn
- Training Space env var OPENGRID_MODE=training already set via HF API

Dockerfile

@@ -1,5 +1,8 @@
 # Hugging Face Docker Space — OpenGrid
 # Docs: https://huggingface.co/docs/hub/spaces-sdks-docker
+#
+# This Dockerfile serves both the UI Space and the Training Space.
+# Set OPENGRID_MODE=training to run GRPO training instead of the server.
 
 FROM python:3.10-slim
 
@@ -14,19 +17,27 @@ ENV PATH="/home/user/.local/bin:$PATH"
 
 WORKDIR /app
 
-# Install dependencies
+# Install dependencies (both server and training)
 COPY --chown=user requirements.txt .
-RUN pip install --no-cache-dir --upgrade -r requirements.txt
+COPY --chown=user requirements-training.txt .
+RUN pip install --no-cache-dir --upgrade -r requirements.txt \
+    && pip install --no-cache-dir --upgrade -r requirements-training.txt
 
 # Copy application code
 COPY --chown=user . /app
 
+# Make entrypoint executable
+RUN chmod +x entrypoint.sh
+
+# Default to server mode (override with OPENGRID_MODE=training)
+ENV OPENGRID_MODE=server
+
 # Expose HF Spaces default port
 EXPOSE 7860
 
-# Healthcheck
+# Healthcheck (only applies in server mode)
 HEALTHCHECK --interval=30s --timeout=5s --start-period=15s \
     CMD python -c "import httpx; httpx.get('http://localhost:7860/health').raise_for_status()" || exit 1
 
-# Run the server
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+# Entrypoint switches between server and training
+CMD ["./entrypoint.sh"]

entrypoint.sh

@@ -0,0 +1,22 @@
+#!/bin/bash
+# OpenGrid entrypoint — switches between UI server and GRPO training
+# based on the OPENGRID_MODE environment variable.
+#
+# OPENGRID_MODE=training  → runs GRPO training pipeline
+# OPENGRID_MODE=server    → runs the FastAPI UI server (default)
+
+set -e
+
+MODE="${OPENGRID_MODE:-server}"
+
+if [ "$MODE" = "training" ]; then
+    echo "========================================"
+    echo "  OpenGrid — GRPO Training Mode"
+    echo "========================================"
+    exec python run_training.py
+else
+    echo "========================================"
+    echo "  OpenGrid — Control Room Server"
+    echo "========================================"
+    exec uvicorn app:app --host 0.0.0.0 --port 7860
+fi