"""FastAPI app for the SUPERBEAM environment.

    uvicorn server.app:app --host 0.0.0.0 --port 8000     (from the env directory)
"""

import os

from openenv.core.env_server import create_app

try:
    from ..models import SuperBeamAction, SuperBeamObservation, SuperBeamState
    from .superbeam_environment import SuperBeamEnvironment
except ImportError as e:
    if "relative import" not in str(e) and "no known parent package" not in str(e):
        raise
    from models import SuperBeamAction, SuperBeamObservation, SuperBeamState
    from server.superbeam_environment import SuperBeamEnvironment

app = create_app(
    SuperBeamEnvironment,
    SuperBeamAction,
    SuperBeamObservation,
    env_name="superbeam_env",
    state_cls=SuperBeamState,
    max_concurrent_envs=int(os.getenv("MAX_CONCURRENT_ENVS", "16")),
)


def main():
    import uvicorn

    uvicorn.run(app, host="0.0.0.0", port=int(os.getenv("PORT", "8000")))


if __name__ == "__main__":
    main()