Update app.py

app.py

@@ -67,7 +67,7 @@ def get_or_create_env(agent_id: str):
 
 
 # ── FastAPI App ───────────────────────────────────────────────────────────────
-fastapi_app = FastAPI(
+app = FastAPI(
     title="OpenEnv: Split-Brain Collapse",
     description=(
         "An OpenEnv-compliant multi-agent RL environment simulating a "
@@ -83,7 +83,7 @@ fastapi_app = FastAPI(
 # CORE RL API  (openenv validate + external agents)
 # ══════════════════════════════════════════════════════════════════════════════
 
-@fastapi_app.get("/health")
+@app.get("/health")
 def health():
     """Health check — returns 200 OK."""
     return {
@@ -95,7 +95,7 @@ def health():
     }
 
 
-@fastapi_app.post("/reset")
+@app.post("/reset")
 async def reset(request: Request):
     """
     Reset the environment for a given task.
@@ -120,7 +120,7 @@ async def reset(request: Request):
     return JSONResponse(content=obs.model_dump())
 
 
-@fastapi_app.post("/step")
+@app.post("/step")
 async def step(request: Request):
     """Execute one raw action in the environment."""
     if active_env is None:
@@ -131,7 +131,7 @@ async def step(request: Request):
     return JSONResponse(content=result.model_dump())
 
 
-@fastapi_app.get("/state")
+@app.get("/state")
 def state():
     """Return the current observation without advancing the episode."""
     if active_env is None:
@@ -139,7 +139,7 @@ def state():
     return JSONResponse(content=active_env.state().model_dump())
 
 
-@fastapi_app.get("/tasks")
+@app.get("/tasks")
 def list_tasks(agent_id: str = "split_brain"):
     """List all available tasks with difficulty metadata for a specific agent."""
     if agent_id not in AGENT_REGISTRY:
@@ -151,7 +151,7 @@ def list_tasks(agent_id: str = "split_brain"):
 # AGENT REGISTRY API  (frontend discovers agents + tasks)
 # ══════════════════════════════════════════════════════════════════════════════
 
-@fastapi_app.get("/agents")
+@app.get("/agents")
 def list_agents():
     """
     Return all registered agents with their tasks.
@@ -179,7 +179,7 @@ class AgentStepRequest(BaseModel):
     task:     Optional[str] = Field(None, description="Task hint (set via /reset)")
 
 
-@fastapi_app.post("/agent/step")
+@app.post("/agent/step")
 def agent_step(body: AgentStepRequest):
     """
     Run ONE LLM-powered step on the active Split-Brain environment.
@@ -242,9 +242,9 @@ def agent_step(body: AgentStepRequest):
 # STATIC FILES — serves static/index.html at /
 # Mount AFTER all API routes so API routes take priority.
 # ══════════════════════════════════════════════════════════════════════════════
-fastapi_app.mount("/", StaticFiles(directory="static", html=True), name="static")
+app.mount("/", StaticFiles(directory="static", html=True), name="static")
 
 
 if __name__ == "__main__":
     # Change "app:app" (implicit) to "app:fastapi_app"
-    uvicorn.run("app:fastapi_app", host="0.0.0.0", port=7860, log_level="info")
+    uvicorn.run(app, host="0.0.0.0", port=7860, log_level="info")