Update env.py

env.py

@@ -1,7 +1,8 @@
 import asyncio
-from typing import Optional
+from typing import Optional, List
 from types import SimpleNamespace
-from openenv.core.env_server import Environment, serve
+from fastapi import FastAPI 
+from openenv.core.env_server import Environment
 from models import MyEnvV4Observation, MyEnvV4Action
 
 class MyEnvV4Env(Environment):
@@ -82,9 +83,22 @@ class MyEnvV4Env(Environment):
 
 
 
-# 1. Create your environment instance
+# Create the environment instance
 my_env = MyEnvV4Env()
 
-# 2. Wrap it in an EnvServer to create the FastAPI app
-# This is what Docker/Uvicorn is looking for!
-app = serve(my_env)
+# defining the FastAPI app and routes
+app = FastAPI()
+
+@app.get("/reset")
+async def reset():
+    result = await my_env.reset()
+    return {"observation": result.observation, "reward": result.reward, "done": result.done}
+
+@app.post("/step")
+async def step(action: MyEnvV4Action):
+    result = await my_env.step(action)
+    return {"observation": result.observation, "reward": result.reward, "done": result.done}
+
+@app.get("/state")
+async def state():
+    return await my_env.state()