Switch light on on the press of a button. Turn it off automatically after 5s

energylight.py

# Energy light

from gpiozero import LED, Button
from time import sleep, time

from gpiozero.pins.pigpio import PiGPIOFactory
factory = PiGPIOFactory(host='raspberrypi')

led = LED(17, pin_factory=factory)
btn = Button(4, pin_factory=factory)

# Time when the light will switch off again
offtime = 0

while True:
    if btn.is_pressed:
        print("You pressed me")
        # Store time to switch light off
        offtime = time() + 5
    if time() < offtime:
        led.on()
    else:
        led.off()
    sleep(0.1)