import gradio as gr
from definitions import SIZES_PER_CATEGORY


def update_main_categories(main):
    if isinstance(SIZES_PER_CATEGORY[main], list):
        return (
            main,
            "",
            "",
            gr.Dropdown(list(SIZES_PER_CATEGORY[main]), label="Size", interactive=True),
        )
    return (
        main,
        "",
        "",
        gr.Dropdown(
            list(SIZES_PER_CATEGORY[main].keys()),
            label="Size",
            interactive=False,
        ),
    )


def update_sub_categories(main, sub):
    if isinstance(SIZES_PER_CATEGORY[main][sub], list):
        return (
            main,
            sub,
            "",
            gr.Dropdown(
                list(SIZES_PER_CATEGORY[main][sub]), label="Size", interactive=True
            ),
        )
    return (
        main,
        sub,
        "",
        gr.Dropdown(
            list(SIZES_PER_CATEGORY[main][sub].keys()),
            label="Size",
            interactive=False,
        ),
    )


def update_sub_sub_categories(main, sub, sub_sub):
    if isinstance(SIZES_PER_CATEGORY[main][sub][sub_sub], list):
        return (
            main,
            sub,
            sub_sub,
            gr.Dropdown(
                list(SIZES_PER_CATEGORY[main][sub][sub_sub]),
                label="Size",
                interactive=True,
            ),
        )
    return (
        main,
        sub,
        sub_sub,
        gr.Dropdown(
            list(SIZES_PER_CATEGORY[main][sub][sub_sub].keys()),
            label="Size",
            interactive=False,
        ),
    )


def submit_form(
    title,
    description,
    condition,
    main_category,
    sub_category,
    sub_sub_category,
    size,
    color,
    hashtags,
    designer_names,
    followerno,
    user_score,
):
    print(
        f"Title: {title}, Description: {description}, Condition: {condition}, hashtags: {hashtags}, designer: {designer_names}, Main Category: {main_category}, Sub:{sub_category}, sub_sub:{sub_sub_category},  Size: {size}, Color: {color}, Followers: {followerno}, User Score: {user_score}"
    )


with gr.Blocks(theme="argilla/argilla-theme", title="Grailed Price Predictor") as demo:
    global size_text_box
    gr.HTML(
        """
    <h1 style="text-align: center;">Grailed Price Predictor</h1>
    <p>Welcome to our Grailed Price Prediction Model! This project, developed for the ID2223 course,
     demonstrates how modern machine learning systems can be effectively implemented using feature stores and serverless computing.
     To get started, simply fill out the form and click "Submit."</p>
    """
    )

    main_category = gr.State("")
    sub_category = gr.State("")
    sub_sub_category = gr.State("")

    with gr.Group():
        title_text_box = gr.Textbox(label="Title")
        description_text_area = gr.TextArea(label="Description")

        condition_dropdown = gr.Dropdown(
            ["new", "gently used", "used", "worn"],
            label="Condition",
            info="The condition of your item.",
            interactive=True,
        )

        with gr.Row():
            with gr.Column():

                @gr.render(inputs=[main_category, sub_category, sub_sub_category])
                def show_categories(main, sub, sub_sub):
                    global size_text_box
                    main_dropdown = gr.Dropdown(
                        list(SIZES_PER_CATEGORY.keys()),
                        label="Main Category",
                        value=main,
                        interactive=True,
                    )
                    main_dropdown.change(
                        update_main_categories,
                        main_dropdown,
                        [main_category, sub_category, sub_sub_category, size_text_box],
                    )

                    if not main or isinstance(SIZES_PER_CATEGORY[main], list):
                        return

                    sub_dropdown = gr.Dropdown(
                        list(SIZES_PER_CATEGORY[main].keys()),
                        label="Sub Category",
                        interactive=True,
                        value=sub,
                    )
                    sub_dropdown.change(
                        update_sub_categories,
                        [main_dropdown, sub_dropdown],
                        [main_category, sub_category, sub_sub_category, size_text_box],
                    )

                    if not sub or isinstance(SIZES_PER_CATEGORY[main][sub], list):
                        return

                    sub_sub_dropdown = gr.Dropdown(
                        list(SIZES_PER_CATEGORY[main][sub].keys()),
                        label="Sub Sub Category",
                        interactive=True,
                        value=sub_sub,
                    )
                    sub_dropdown.change(
                        update_sub_sub_categories,
                        [main_dropdown, sub_dropdown, sub_sub_dropdown],
                        [main_category, sub_category, sub_sub_category, size_text_box],
                    )

            size_text_box = gr.Dropdown(
                label="Size",
                interactive=False,
            )

        color_text_box = gr.Textbox(label="Color")

        with gr.Row():
            with gr.Column():
                hashtags_text_box = gr.Textbox(label="Hashtags")

                @gr.render(inputs=hashtags_text_box)
                def show_hashtags(textbox):
                    if not textbox:
                        return
                    for i, hashtag in enumerate(textbox.split(" ")):
                        gr.Textbox(label=f"Hashtag #{i+1}", value=hashtag)

            with gr.Column():
                designer_names_text_box = gr.Textbox(label="Designer Names")

                @gr.render(inputs=designer_names_text_box)
                def show_designer_names(textbox):
                    if not textbox:
                        return
                    for i, designer_name in enumerate(textbox.split(" ")):
                        gr.Textbox(label=f"Designer Name #{i+1}", value=designer_name)

        with gr.Row():
            followernoNumber = gr.Number(label="Number of Followers")
            userScoreNumber = gr.Number(label="Your User Score")

        submitButton = gr.Button("Submit")
        submitButton.click(
            submit_form,
            [
                title_text_box,
                description_text_area,
                condition_dropdown,
                main_category,
                sub_category,
                sub_sub_category,
                size_text_box,
                color_text_box,
                hashtags_text_box,
                followernoNumber,
                userScoreNumber,
            ],
        )


if __name__ == "__main__":
    demo.launch()