server.py: Add logic to keep screen on