commit

app.py

@@ -0,0 +1,95 @@
+"""
+FastAPI server exposing the OpenEnv-compatible HTTP API.
+Deployed as a Hugging Face Space on port 7860.
+
+Endpoints
+─────────
+GET  /health          — liveness probe
+GET  /tasks           — list available tasks
+POST /reset           — start new episode  { "task_id": "easy_syntax" }
+POST /step            — take action        CodeReviewAction JSON body
+GET  /state           — current state snapshot
+"""
+import os
+from typing import Optional
+
+import uvicorn
+from fastapi import FastAPI, HTTPException, Query
+from fastapi.middleware.cors import CORSMiddleware
+
+from environment.env import CodeReviewEnv
+from environment.models import CodeReviewAction
+
+app = FastAPI(
+    title="CodeReview-Env",
+    description=(
+        "An OpenEnv reinforcement-learning environment where AI agents learn to "
+        "review code for syntax errors, logic bugs, and security vulnerabilities."
+    ),
+    version="1.0.0",
+)
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# Single global environment instance (one session per container)
+_env = CodeReviewEnv()
+
+
+# ── endpoints ────────────────────────────────────────────────────────────────
+
+@app.get("/health")
+async def health():
+    return {"status": "ok", "env": "CodeReview-Env", "version": "1.0.0"}
+
+
+@app.get("/tasks")
+async def list_tasks():
+    return {"tasks": _env.list_tasks()}
+
+
+@app.post("/reset")
+async def reset(task_id: Optional[str] = Query(default="easy_syntax")):
+    """
+    Reset the environment and start a new episode.
+    task_id: one of easy_syntax | medium_logic | hard_security
+    """
+    try:
+        result = _env.reset(task_id=task_id)
+        return result.model_dump()
+    except ValueError as e:
+        raise HTTPException(status_code=400, detail=str(e))
+
+
+@app.post("/step")
+async def step(action: CodeReviewAction):
+    """
+    Submit a code-review action.
+    Returns observation, reward (0.0–1.0), done flag, and grader feedback.
+    """
+    if not _env.is_initialized:
+        raise HTTPException(status_code=400, detail="Call POST /reset first.")
+    try:
+        result = _env.step(action)
+        return result.model_dump()
+    except RuntimeError as e:
+        raise HTTPException(status_code=400, detail=str(e))
+
+
+@app.get("/state")
+async def state():
+    """Return a snapshot of the current episode state."""
+    if not _env.is_initialized:
+        raise HTTPException(status_code=400, detail="Call POST /reset first.")
+    return _env.get_state().model_dump()
+
+
+# ── entrypoint ───────────────────────────────────────────────────────────────
+
+if __name__ == "__main__":
+    port = int(os.environ.get("PORT", 7860))
+    uvicorn.run("app:app", host="0.0.0.0", port=port, reload=False)