import traceback
from core.env_server import create_fastapi_app
from server.environment import GridWorldEnv
from models import GridAction, GridObservation


def env_factory():
    try:
        return GridWorldEnv()
    except Exception:
        traceback.print_exc()
        raise


app = create_fastapi_app(
    env_factory,
    action_cls=GridAction,
    observation_cls=GridObservation,
)

if __name__ == "__main__":
    import uvicorn
    uvicorn.run(app, host="0.0.0.0", port=8000)