Create inference.py

inference.py

@@ -0,0 +1,26 @@
+from fastapi import FastAPI
+from env import DeliveryEnv
+import random
+
+app = FastAPI()
+env = DeliveryEnv()
+
+ACTIONS = ["up", "down", "left", "right", "pickup", "drop"]
+
+@app.get("/")
+def home():
+    return {"message": "Environment running"}
+
+@app.get("/reset")
+def reset():
+    return env.reset()
+
+@app.get("/step")
+def step():
+    action = random.choice(ACTIONS)
+    state, reward, done, _ = env.step(action)
+    return {
+        "state": state,
+        "reward": reward,
+        "done": done
+    }