from time import sleep
import pygame
import pygame_menu
from pygame_menu import themes
import pygame_menu.events

pygame.init()
surface = pygame.display.set_mode((600, 400))

def set_difficulty(value, difficulty):
    print(value)
    print(difficulty)

def start_the_game():
    pass

def level_menu():
    level.enable()
    mainmenu.disable()

mainmenu = pygame_menu.Menu('Welcome', 600, 400, theme=themes.THEME_SOLARIZED)
mainmenu.add.text_input('Name:', default='username', maxchar=20)
mainmenu.add.button('Play', start_the_game)
mainmenu.add.button('Levels', level_menu)
mainmenu.add.button('Quit', pygame_menu.events.EXIT)

level = pygame_menu.Menu('Select a Difficulty', 600, 400, theme=themes.THEME_BLUE)
level.add.selector('Difficulty :', [('Hard', 1), ('Easy', 2)], onchange=set_difficulty)

UPDATE_LOADING = pygame.USEREVENT + 1
pygame.time.set_timer(UPDATE_LOADING, 1000)  # Trigger every second

while True:
    events = pygame.event.get()
    for event in events:
        if event.type == UPDATE_LOADING:
            progress = loading.get_widget("1")
            progress.set_value(progress.get_value() + 1)
            if progress.get_value() == 100:
                pygame.time.set_timer(UPDATE_LOADING, 0)
        if event.type == pygame.QUIT:
            exit()

    if mainmenu.is_enabled():
        mainmenu.update(events)
        mainmenu.draw(surface)
        selected_widget = mainmenu.get_current().get_selected_widget()
        if selected_widget:
            arrow.draw(surface, selected_widget)

    pygame.display.update()