Nihar-776/Hackathon-MetaPytorch

1from fastapi import FastAPI2import uvicorn3 4app = FastAPI()5 6# Simple state7state_data = {"position": 0}8 9@app.get("/")10def home():11    return {"message": "OpenEnv running"}12 13@app.get("/reset")14def reset():15    global state_data16    state_data = {"position": 0}17    return state_data18 19@app.post("/step")20def step(action: dict):21    global state_data22 23    move = action.get("action", "")24 25    if move == "right":26        state_data["position"] += 127 28    reward = 1.0 if state_data["position"] >= 5 else 0.029    done = state_data["position"] >= 530 31    return {32        "state": state_data,33        "reward": reward,34        "done": done35    }36 37@app.get("/state")38def state():39    return state_data40 41def main():42    uvicorn.run(app, host="0.0.0.0", port=7860)43 44if __name__ == '__main__':45    main()