server/app.py

# SPDX-License-Identifier: BSD-3-Clause

"""FastAPI application for the watercolour environment."""

try:
    from openenv.core.env_server import create_app

    from ..models import WatercolourAction, WatercolourObservation
    from .watercolour_environment import WatercolourEnvironment
except ImportError:
    from openenv.core.env_server import create_app

    from models import WatercolourAction, WatercolourObservation
    from server.watercolour_environment import WatercolourEnvironment

# The class is passed rather than an instance so each session gets its own
# environment and its own sampled subject.
app = create_app(
    WatercolourEnvironment,
    WatercolourAction,
    WatercolourObservation,
    env_name="watercolour_env",
)


def main():
    import uvicorn

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


if __name__ == "__main__":
    main()