main.py

import pygame

from cv import PoseEstimator
from menus import HomeMenu
from constants import *

pygame.init()
pygame.mixer.init()

screen = pygame.display.set_mode((WINDOW_WIDTH, WINDOW_HEIGHT), pygame.FULLSCREEN)
pygame.display.set_caption(f"WebcamWare {VERSION}")
pygame.display.gl_set_attribute(pygame.GL_MULTISAMPLEBUFFERS, 3)
pygame.display.set_icon(pygame.image.load('assets/img/menus/intergame/webcam.png'))
clock = pygame.time.Clock()

pygame.mixer.music.load('assets/music/background.mp3')
pygame.mixer.music.set_volume(0.5)
pygame.mixer.music.play(-1)

menu = HomeMenu(screen)

PoseEstimator().start_tracking()

while True:
    for event in pygame.event.get():
        new_menu = menu.handle_event(event)
        if new_menu:
            menu = new_menu

    new_menu = menu.draw()
    if new_menu:
        menu = new_menu

    pygame.display.update()
    clock.tick(60)