Jaswin-27/MetaXSCALER_Hackathon

1from fastapi import FastAPI
2from env import LifeEnv
3from models import Action
4
5app = FastAPI()
6
7env = LifeEnv()
8
9# RESET
10@app.post("/reset")
11def reset():
12    state = env.reset()
13    return state
14
15# STEP
16@app.post("/step")
17def step(action: Action):
18    result = env.step(action)
19    return result
20
21# GET STATE
22@app.get("/state")
23def get_state():
24    return env.get_state()
25
26import uvicorn
27
28if __name__ == "__main__":
29    uvicorn.run(app, host="0.0.0.0", port=7860)