server/app.py

from __future__ import annotations

import os

USE_OPENENV_HELPER = os.environ.get("ENTERPRISE_FINANCE_USE_OPENENV_HELPER", "").lower() in {
    "1",
    "true",
    "yes",
}

if USE_OPENENV_HELPER:
    try:
        from openenv.core.env_server.http_server import create_app
    except ImportError:
        from enterprise_finance_env._compat import create_app
else:
    from enterprise_finance_env._compat import create_app

try:
    from enterprise_finance_env.models import EnterpriseFinanceActionPayload, EnterpriseFinanceObservation
    from enterprise_finance_env.server.enterprise_env import EnterpriseFinanceEnv
except ImportError:
    from models import EnterpriseFinanceActionPayload, EnterpriseFinanceObservation
    from server.enterprise_env import EnterpriseFinanceEnv


TASK = os.environ.get("ENTERPRISE_FINANCE_TASK", "easy").lower()


def _env_factory() -> EnterpriseFinanceEnv:
    return EnterpriseFinanceEnv(difficulty=TASK)


app = create_app(
    _env_factory,
    EnterpriseFinanceActionPayload,
    EnterpriseFinanceObservation,
    env_name="enterprise_finance_env",
)


def main() -> None:
    import uvicorn

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


if __name__ == "__main__":
    main()