from openenv.core.env_server import create_app
from .environment import AccountantEng
from .schemas import Action, AccountantState

# This creates the API server
# It automatically handles the WebSocket connections for you!
app = create_app(
    env=AccountantEng,
    action_cls=Action,
    observation_cls=AccountantState,
    env_name="mini-accountant",
)

if __name__ == "__main__":
    import uvicorn
    # This runs the server locally on port 8000
    uvicorn.run(app, host="0.0.0.0", port=7860)