Spaces:
Running
Running
| """FastAPI app: the OpenEnv API (/reset, /step, /state, /ws, /schema, task routes) plus the play page at /. | |
| uvicorn scott_wu_math.server.app:app --port 8012 | |
| """ | |
| from __future__ import annotations | |
| import os | |
| from pathlib import Path | |
| from fastapi.responses import FileResponse, RedirectResponse | |
| from fastapi.staticfiles import StaticFiles | |
| from openenv.core.env_server.http_server import create_app | |
| from ..models import CardAction, CardObservation, CardState | |
| from .environment import ScottWuMathEnvironment | |
| WEB = Path(__file__).resolve().parent / "web" | |
| README = Path(__file__).resolve().parents[1] / "README.md" | |
| # The play page replaces OpenEnv's generic Gradio playground. | |
| os.environ["ENABLE_WEB_INTERFACE"] = "false" | |
| app = create_app( | |
| ScottWuMathEnvironment, | |
| CardAction, | |
| CardObservation, | |
| env_name="scott_wu_math", | |
| max_concurrent_envs=int(os.getenv("MAX_CONCURRENT_ENVS", "256")), | |
| state_cls=CardState, | |
| ) | |
| app.mount("/static", StaticFiles(directory=WEB), name="static") | |
| def play_page() -> FileResponse: | |
| return FileResponse(WEB / "index.html", headers={"Cache-Control": "no-cache"}) | |
| def readme() -> FileResponse: | |
| """The environment card, rendered by /static/card.html.""" | |
| return FileResponse(README, media_type="text/markdown; charset=utf-8", headers={"Cache-Control": "no-cache"}) | |
| def web_alias() -> RedirectResponse: | |
| return RedirectResponse("/") | |
| def main(host: str = "0.0.0.0", port: int | None = None) -> None: | |
| import uvicorn | |
| uvicorn.run(app, host=host, port=port or int(os.getenv("PORT", "8000"))) | |
| if __name__ == "__main__": | |
| main() | |