server/app.py

"""FastAPI app for the DiskPanic environment server."""

from openenv.core.env_server.http_server import create_app

try:
    from ..models import DiskPanicAction, DiskPanicObservation
    from .environment import DiskPanicEnvironment
except ImportError:
    from models import DiskPanicAction, DiskPanicObservation
    from server.environment import DiskPanicEnvironment


app = create_app(
    DiskPanicEnvironment,
    DiskPanicAction,
    DiskPanicObservation,
    env_name="disk_panic",
    max_concurrent_envs=1,
)


def main() -> None:
    """Entry point for `uv run server` / `python -m server.app`."""
    import argparse
    import uvicorn

    parser = argparse.ArgumentParser()
    parser.add_argument("--host", type=str, default="0.0.0.0")
    parser.add_argument("--port", type=int, default=8000)
    args = parser.parse_args()
    uvicorn.run(app, host=args.host, port=args.port)


if __name__ == "__main__":
    main()