from gradio.themes.base import Base

class CustomTheme(Base):

    def __init__(self):
        super().__init__()

        # neutral colors
        neutral_light = "#FDFBF6"
        neutral_dark_beige = "#F1E8D2"
        neutral_dark = "#232959"

        # primary, secondary, tertitary and CTA color
        red = "#B04F4D"
        light_green = "#AEC767"
        middle_green = "#749757"
        dark_green = "#426544"

        body_background_image = "url(https://huggingface.co/datasets/a-7/plantabot2/resolve/main/background-plantabot%201.jpg) center center/cover"

        super().set(
            body_background_fill=body_background_image,
            button_primary_background_fill=red,
            button_primary_text_color=neutral_light,
            button_secondary_background_fill=dark_green,
            button_secondary_text_color=neutral_light,
            button_secondary_border_color=neutral_light,
            color_accent_soft=neutral_dark_beige,
            border_color_accent_subdued=neutral_dark_beige,
            border_color_primary=neutral_light,
            block_title_background_fill=neutral_light,
            body_text_color=neutral_dark,
        )