commit

app.py

@@ -1,95 +1,15 @@
-"""
-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=["*"],
+from openenv.core.env_server.http_server import create_app
+from environment.models import CodeReviewAction, CodeReviewObservation
+from environment.env import CodeReviewEnvironment
+
+app = create_app(
+    CodeReviewEnvironment,
+    CodeReviewAction,
+    CodeReviewObservation,
+    env_name="code_review_env",
+    max_concurrent_envs=4,
 )
 
-# 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)
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)