Spaces:
Sleeping
Sleeping
| # Copyright (c) Meta Platforms, Inc. and affiliates. | |
| # All rights reserved. | |
| # | |
| # This source code is licensed under the BSD-style license found in the | |
| # LICENSE file in the root directory of this source tree. | |
| """FastAPI application for the Store OpenEnv simulation.""" | |
| try: | |
| from openenv.core.env_server.http_server import create_app | |
| except Exception as e: # pragma: no cover | |
| raise ImportError( | |
| "openenv is required for the web interface. Install dependencies with '\n uv sync\n'" | |
| ) from e | |
| try: | |
| from ..models import StoreAction, StoreObservation | |
| from .store_environment import StoreEnvironment | |
| except ImportError: | |
| # Top-level ``server`` when PYTHONPATH is the env root (e.g. uvicorn server.app:app) | |
| # raises ImportError for ``..models``, not ModuleNotFoundError. | |
| from models import StoreAction, StoreObservation | |
| from server.store_environment import StoreEnvironment | |
| app = create_app( | |
| StoreEnvironment, | |
| StoreAction, | |
| StoreObservation, | |
| env_name="store", | |
| max_concurrent_envs=4, | |
| ) | |
| def main(host: str = "0.0.0.0", port: int = 8000): | |
| """ | |
| Entry point for direct execution via ``uv run`` or ``python -m``. | |
| """ | |
| import uvicorn | |
| uvicorn.run(app, host=host, port=port) | |
| if __name__ == "__main__": | |
| import argparse | |
| parser = argparse.ArgumentParser() | |
| parser.add_argument("--port", type=int, default=8000) | |
| args = parser.parse_args() | |
| if args.port == 8000: | |
| main() | |
| else: | |
| main(port=args.port) | |