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 PharmaDDI Environment. | |
| Endpoints: | |
| - POST /reset: Reset the environment with a new patient scenario | |
| - POST /step: Submit drug interaction analysis for grading | |
| - GET /state: Get current environment state | |
| - GET /schema: Get action/observation schemas | |
| - WS /ws: WebSocket endpoint for persistent sessions | |
| Usage: | |
| uvicorn server.app:app --reload --host 0.0.0.0 --port 8000 | |
| """ | |
| try: | |
| from openenv.core.env_server.http_server import create_app | |
| except Exception as e: | |
| raise ImportError( | |
| "openenv is required. Install with: uv sync" | |
| ) from e | |
| try: | |
| from models import PharmaDDIAction, PharmaDDIObservation | |
| from .PharmaDDIEnv_environment import PharmaDDIEnvironment | |
| except ModuleNotFoundError: | |
| from models import PharmaDDIAction, PharmaDDIObservation | |
| from server.PharmaDDIEnv_environment import PharmaDDIEnvironment | |
| # Initialize a single global instance of the environment | |
| # This ensures state is preserved across stateless HTTP calls | |
| global_env = PharmaDDIEnvironment() | |
| # Create the app with web interface and README integration | |
| app = create_app( | |
| lambda: global_env, | |
| PharmaDDIAction, | |
| PharmaDDIObservation, | |
| env_name="PharmaDDIEnv", | |
| max_concurrent_envs=1, | |
| ) | |
| def main(): | |
| """Entry point for the environment server.""" | |
| import uvicorn | |
| uvicorn.run("server.app:app", host="0.0.0.0", port=8000) | |
| if __name__ == "__main__": | |
| main() | |