import flet as ft


class Theme:
    def __init__(self, page: ft.Page) -> None:
        self.page = page

    def change(self, theme: ft.ThemeMode = None):
        if theme:
            self.page.theme_mode = theme
        self.page.update()