# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
#
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

"""FastAPI application for the Chess Environment."""

import os

from openenv.core.env_server import create_app

from ..models import ChessAction, ChessObservation, ChessState
from .chess_environment import ChessEnvironment

# Create the FastAPI app
# Pass the class (factory) instead of an instance for WebSocket session support
app = create_app(
    ChessEnvironment,
    ChessAction,
    ChessObservation,
    env_name="chess_env",
    state_cls=ChessState,
    max_concurrent_envs=int(os.getenv("MAX_CONCURRENT_ENVS", "8")),
)


def main():
    import uvicorn

    uvicorn.run(app, host="0.0.0.0", port=8000)


if __name__ == "__main__":
    main()