Used Gemini to generate the GUI so that I can focus on the actual Deep Q Network

After that I implemented a velocity variable that manipulates the rocket’s y coordinates

Finally, I implemented a reset function and a step function which changes the velocity of the rocket based on the AI’s action and it also returns required data for training (Y coordinate, reward, Done, Truncated, Extra Info)

To reduce the complexity of the problem, I am ignoring horizontal movement (x direction) and focusing only on the vertical movement while ignoring factors like air resistance.