import pygame
import sys

pygame.init()
screen = pygame.display.set_mode((800, 800))
clock = pygame.time.Clock()

white = (255, 255, 255)
black = (0, 0, 0)
current_time = 0
button_press_time = 0

running = True

while running:
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            running = False
        if event.type == pygame.KEYDOWN:
            button_press_time = pygame.time.get_ticks()
            screen.fill(white)

    current_time = pygame.time.get_ticks()
    
    if current_time - button_press_time > 2000:
        screen.fill(black)

    print(f"current time: {current_time} button press time: {button_press_time}")
    pygame.display.flip()
    clock.tick(60)



pygame.quit()
sys.exit()