Update server.py

server.py

@@ -3,16 +3,31 @@ from fastapi import FastAPI
 from pydantic import BaseModel
 from env import EmailSortingEnv
 
+# Create FastAPI app
 app = FastAPI(
     title="Email Sorting OpenEnv",
+    description="Real-world email sorting environment for RL agents",
     version="1.0.0"
 )
 
+# One global environment instance
 env = EmailSortingEnv()
 
+# ============================================
+# REQUEST MODELS
+# ============================================
+
 class StepRequest(BaseModel):
     action: str
 
+# ============================================
+# API ENDPOINTS
+# ============================================
+
+@app.get("/health")
+def health_check():
+    return {"status": "ok", "message": "Email Sorting Environment is running"}
+
 @app.get("/")
 def root():
     return {
@@ -22,10 +37,6 @@ def root():
         "endpoints": ["/reset", "/step", "/state", "/health"]
     }
 
-@app.get("/health")
-def health():
-    return {"status": "ok"}
-
 @app.post("/reset")
 def reset():
     state = env.reset()
@@ -34,13 +45,31 @@ def reset():
 @app.post("/step")
 def step(request: StepRequest):
     next_state, reward, done, info = env.step(request.action)
-    return {"status": "success", "state": next_state, "reward": reward, "done": done, "info": info}
+    return {
+        "status": "success",
+        "state": next_state,
+        "reward": reward,
+        "done": done,
+        "info": info
+    }
 
 @app.get("/state")
 def get_state():
     return {"status": "success", "state": env.state()}
 
+@app.get("/graders")
+def run_graders():
+    from graders import run_all_graders
+    results = run_all_graders()
+    return {"status": "success", "results": results}
+
+# ============================================
+# START SERVER
+# ============================================
+
 def main():
+    print("Starting Email Sorting Environment Server...")
+    print("Server running at http://localhost:7860")
     uvicorn.run(app, host="0.0.0.0", port=7860)
 
 if __name__ == "__main__":