from fastapi import FastAPI
from env import FortisL7Env

app = FastAPI()

env = FortisL7Env()

@app.post("/reset")
def reset():
    obs, info = env.reset()
    return {"observation": obs.tolist(), "info": info}

@app.post("/step")
def step(action: int):
    obs, reward, terminated, truncated, info = env.step(action)
    return {
        "observation": obs.tolist(),
        "reward": reward,
        "done": terminated or truncated,
        "info": info
    }

@app.get("/state")
def state():
    return env.state()