Чтобы эта картинка держалась до тех пор, пока не будет нажата клавиша

import pygame
import keyboard
import time
from pygame.locals import *
pygame.init()
game_display = pygame.display.set_mode((1000, 800))
pygame.display.set_caption('1')
def event_handler():
    for event in pygame.event.get():
        if (event.type == QUIT):
            pygame.quit()
            quit()
game_display.fill('white')
img = pygame.image.load('emd.jpg')
game_display.blit(img, (0, 0))
while True:
    event_handler()
    pygame.display.update()

def event_handler():
    for event in pygame.event.get():
        if (event.type == QUIT):
            pygame.quit()
            quit()
        if (event.type == KEYDOWN):
            if event.key == K_SPACE:
                vanish = True

while True:
    event_handler()
    if vanish:
        game_display.fill('black')
    pygame.display.update()

import pygame
import keyboard
import time
from pygame.locals import *
 
 
def main():
    pygame.init()
    game_display = pygame.display.set_mode((1000, 800))
    pygame.display.set_caption('1')
    game_display.fill(pygame.Color('white'))
 
    img = pygame.image.load('emd.jpg')
    game_display.blit(img, (0, 0))
 
    vanish = False
 
    running = True
    while running:
        for event in pygame.event.get():
            if event.type == QUIT:
                running = False
            if event.type == KEYDOWN:
                if event.key == K_SPACE:
                    vanish = True
        
        if vanish:
            game_display.fill(pygame.Color('black'))
 
        pygame.display.flip()
    
    pygame.quit()
        
 
if __name__ == '__main__':
    main()