feat: optimize Space — dockerignore, dotenv secrets, multi-worker uvicorn

.dockerignore

@@ -0,0 +1,14 @@
+venv/
+__pycache__/
+*.pyc
+*.pyo
+.env
+.git/
+.gitignore
+.gitattributes
+CLAUDE.md
+assets/
+training/
+eval/
+requirements.txt
+data/facts.json

.env.example

@@ -0,0 +1,15 @@
+# Copy to .env and fill in values for local development
+# In HF Spaces, set these as Space secrets (Settings → Variables and secrets)
+
+# Number of corrupt docs per episode (1-4). Leave blank for random per episode.
+DIFFICULTY=
+
+# Max concurrent environment sessions
+MAX_CONCURRENT_ENVS=64
+
+# WandB (only needed for training, not the server)
+WANDB_API_KEY=
+
+# HuggingFace (only needed for training, not the server)
+HF_TOKEN=
+HF_HUB_MODEL_ID=

Dockerfile

@@ -1,8 +1,18 @@
 FROM python:3.11-slim
+
 WORKDIR /app
+
+# Install deps first (cached layer — only invalidated when requirements change)
 COPY requirements-server.txt .
 RUN pip install --no-cache-dir -r requirements-server.txt
+
+# Copy source (excludes venv/, training/, assets/, .git/ via .dockerignore)
 COPY . .
-RUN python -c "from data.loader import build_fact_database; build_fact_database()" || true
+
+# Pre-generate facts.json at build time; falls back gracefully if network is down
+RUN python -m data.loader || echo "facts.json generation skipped — will use fallback facts"
+
 EXPOSE 7860
-CMD ["uvicorn", "environment.server:app", "--host", "0.0.0.0", "--port", "7860"]
+
+# 2 workers: handles concurrent sessions without threading issues
+CMD ["uvicorn", "environment.server:app", "--host", "0.0.0.0", "--port", "7860", "--workers", "2"]

environment/server.py

@@ -1,16 +1,24 @@
+import os
+from dotenv import load_dotenv
 from openenv.core import create_app
 import uvicorn
 
+load_dotenv()
+
 from environment.actions import ContextCorruptionAction, EpisodeObservation
 from environment.env import ContextCorruptionEnv
 
+_difficulty_env = os.getenv("DIFFICULTY")
+_difficulty = int(_difficulty_env) if _difficulty_env else None
+_max_sessions = int(os.getenv("MAX_CONCURRENT_ENVS", "64"))
+
 app = create_app(
-    env=lambda: ContextCorruptionEnv(),
+    env=lambda: ContextCorruptionEnv(difficulty=_difficulty),
     action_cls=ContextCorruptionAction,
     observation_cls=EpisodeObservation,
     env_name="ContextCorruption-Env",
-    max_concurrent_envs=64,
+    max_concurrent_envs=_max_sessions,
 )
 
 if __name__ == "__main__":
-    uvicorn.run("environment.server:app", host="0.0.0.0", port=8000, reload=False)
+    uvicorn.run("environment.server:app", host="0.0.0.0", port=7860, reload=False)