Spaces:
Sleeping
Sleeping
File size: 1,848 Bytes
2043afa | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 | """FastAPI app factory for PolypharmacyEnv."""
from __future__ import annotations
from pathlib import Path
from dotenv import load_dotenv
from fastapi import HTTPException
from fastapi.middleware.cors import CORSMiddleware
from fastapi.staticfiles import StaticFiles
from openenv.core.env_server.http_server import create_app
from starlette.responses import FileResponse
from ..env_core import PolypharmacyEnv
from ..models import PolypharmacyAction, PolypharmacyObservation
from .routes.agent import router as agent_router
load_dotenv()
class SPAStaticFiles(StaticFiles):
"""Serve SPA index for unknown frontend routes."""
async def get_response(self, path: str, scope):
response = await super().get_response(path, scope)
if response.status_code != 404:
return response
index_path = Path(self.directory) / "index.html"
if index_path.exists():
return FileResponse(index_path)
raise HTTPException(status_code=404, detail="Not Found")
def create_polypharmacy_app():
app = create_app(
PolypharmacyEnv,
PolypharmacyAction,
PolypharmacyObservation,
env_name="polypharmacy_env",
)
app.add_middleware(
CORSMiddleware,
allow_origins=[
"http://localhost:5173",
"http://127.0.0.1:5173",
],
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
app.include_router(agent_router)
# In Docker Space deployment, serve built frontend from same container.
project_root = Path(__file__).resolve().parents[4]
frontend_dist = project_root / "frontend" / "dist"
if frontend_dist.exists():
app.mount("/", SPAStaticFiles(directory=frontend_dist, html=True), name="frontend")
return app
app = create_polypharmacy_app()
|