from __future__ import annotations

from fastapi import FastAPI
import uvicorn
from openenv.core.env_server import create_fastapi_app

from logging_utils import get_logger
from .environment import CareerAction, CareerEnvironment, CareerObservation

app: FastAPI = create_fastapi_app(
    CareerEnvironment,
    action_cls=CareerAction,
    observation_cls=CareerObservation,
)
logger = get_logger("server")


@app.get("/")
def root() -> dict[str, str]:
    logger.info("root endpoint hit.")
    return {"status": "ok"}


def main() -> None:
    logger.info("Starting uvicorn server.app:app on 0.0.0.0:8000")
    uvicorn.run("server.app:app", host="0.0.0.0", port=8000)


if __name__ == "__main__":
    main()