Update inference.py

pull request to correct routing

inference.py

@@ -1,14 +1,88 @@
 import os
+from typing import Any, Optional
 
 import uvicorn
+from fastapi import FastAPI
+from fastapi.middleware.cors import CORSMiddleware
+from pydantic import BaseModel
 
-from server.app import app, main
+from env.environment import CareerEnv
 
 
-def main() -> None:
-    port = int(os.environ.get("PORT", "7860"))
-    uvicorn.run(app, host="0.0.0.0", port=port)
+app = FastAPI()
+
+# Add CORS middleware for cross-origin requests
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+env = CareerEnv()
+
+
+class ResetRequest(BaseModel):
+    user_skills: Optional[list[str]] = None
+
+
+class StepRequest(BaseModel):
+    action: dict[str, Any]
+
+
+@app.get("/")
+def healthcheck() -> dict[str, str]:
+    return {"status": "ok"}
+
+
+@app.post("/reset")
+def reset(payload: ResetRequest) -> dict[str, Any]:
+    user_skills = payload.user_skills
+    env.reset(user_skills=user_skills)
+    return {
+        "observation": {"message": "env reset"},
+        "reward": 0,
+        "done": False,
+    }
+
+
+@app.post("/step")
+def step(payload: StepRequest) -> dict[str, Any]:
+    action = payload.action
+    observation, reward, done, info = env.step(action)
+    return {
+        "observation": observation,
+        "reward": reward,
+        "done": done,
+        "info": info,
+    }
+
+
+# Keep legacy endpoints for backward compatibility
+@app.post("/openenv/reset")
+def reset_legacy(payload: ResetRequest) -> dict[str, Any]:
+    user_skills = payload.user_skills
+    env.reset(user_skills=user_skills)
+    return {
+        "observation": {"message": "env reset"},
+        "reward": 0,
+        "done": False,
+    }
+
+
+@app.post("/openenv/step")
+def step_legacy(payload: StepRequest) -> dict[str, Any]:
+    action = payload.action
+    observation, reward, done, info = env.step(action)
+    return {
+        "observation": observation,
+        "reward": reward,
+        "done": done,
+        "info": info,
+    }
 
 
 if __name__ == "__main__":
-    main()
+    port = int(os.environ.get("PORT", "7860"))
+    uvicorn.run(app, host="0.0.0.0", port=port)