File size: 1,511 Bytes
08b81aa f59a9d8 62f081e f59a9d8 08b81aa f59a9d8 08b81aa 24dd7bb f59a9d8 2182d10 f59a9d8 2182d10 62f081e f59a9d8 62f081e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 | from fastapi import FastAPI, HTTPException, Request
from pydantic import BaseModel
from server.environment import CodeReviewEnv
app = FastAPI(title="OpenEnv Code Review Space")
_env = None
class ActionRequest(BaseModel):
action: str
@app.get("/")
def ping():
return {"status": "ok", "message": "Hugging Face Space is running"}
@app.post("/reset")
async def reset(request: Request):
global _env
try:
body = await request.json()
except Exception:
body = {}
difficulty = (body or {}).get("difficulty", "medium")
_env = CodeReviewEnv(difficulty=difficulty)
obs = _env.reset()
return {"observation": obs, "status": "reset_successful"}
@app.post("/step")
def step(req: ActionRequest):
global _env
if _env is None:
raise HTTPException(status_code=400, detail="Environment not initialized. Call /reset first.")
obs, reward, done, error = _env.step(req.action)
return {
"observation": obs,
"reward": float(reward) if reward else 0.0,
"done": done,
"error": error
}
@app.get("/state")
def state():
global _env
if _env is None:
return {"status": "uninitialized"}
return {
"steps_taken": _env.steps_taken,
"rewards": _env.rewards,
"difficulty": _env.difficulty,
"correct_comments": _env.correct_comments
}
def main():
import uvicorn
uvicorn.run("server.app:app", host="0.0.0.0", port=7860)
if __name__ == "__main__":
main()
|