from fastapi import FastAPI
from pydantic import BaseModel

from guess_env.server.guess_environment import GuessEnvironment
from guess_env.models import GuessAction

app = FastAPI()
env = GuessEnvironment(max_guesses=5)

class StepRequest(BaseModel):
    action: GuessAction

@app.post("/reset")
def reset():
    return env.reset()

@app.post("/step")
def step(request: StepRequest):
    obs = env.step(request.action)
    return obs