import gradio as gr
from circumstances.circumstances import create_circumstances
from physical.physical_select_animal import create_bird_anatomy
from physical.physical_checkbox import process_body_parts
from behavior.behavior_checkbox import create_behavior_checkbox
from follow_up.followup_events import create_followup_dropdowns, create_followup_open
from validation_submission.utils_individual import add_data_to_individual

from dotenv import load_dotenv
import os

load_dotenv()
PATH = os.getcwd() + "/"
PATH_ASSETS = os.getenv("PATH_ASSETS")
PATH_ICONS = PATH + PATH_ASSETS + "icons/"


def show_section_wounded(visible, mode, individual):
    if visible == True:
        individual = add_data_to_individual("wounded_state", "Yes", individual)
        individual = add_data_to_individual("dead_state", "No", individual)

    with gr.Column(visible=visible, elem_id="wounded") as wounded_section:
        gr.Markdown("## The animal is wounded / sick.")

        gr.Button(
            "Do you know what conditions caused this?",
            icon=PATH_ICONS + "eye.png",
            variant="primary",
        )
        radio_cause = gr.Radio(
            ["Yes", "No"], value=None, show_label=False, interactive=True
        )
        (
            button_collision,
            button_deliberate_destruction,
            button_indirect_destruction,
            button_natural_cause,
            dropdown,
            dropdown_level2,
            openfield_level2,
            dropdown_extra_level2,
        ) = create_circumstances(visible=False)

        gr.Button(
            "Is the animal displaying behavioural changes?",
            icon=PATH_ICONS + "neuron.png",
            variant="primary",
        )
        radio_behaviour = gr.Radio(
            ["Yes", "No"], value=None, show_label=False, interactive=True
        )
        with gr.Row():
            behavior_checkbox, behavior_text = create_behavior_checkbox(
                "wounded", mode, False
            )

        gr.Button(
            "Are there physical changes on the animal?",
            icon=PATH_ICONS + "cardiogram.png",
            variant="primary",
        )
        radio_physical = gr.Radio(
            ["Yes", "No"], value=None, show_label=False, interactive=True
        )
        with gr.Row():
            physical_boxes = create_bird_anatomy(False, "wounded")
            with gr.Column():
                (
                    checkbox_beak,
                    text_beak,
                    checkbox_body,
                    text_body,
                    checkbox_feathers,
                    text_feathers,
                    checkbox_head,
                    text_head,
                    checkbox_legs,
                    text_legs,
                ) = process_body_parts("wounded", mode, "None")

        gr.Button(
            "Follow-Up Events", icon=PATH_ICONS + "schedule.png", variant="primary"
        )
        gr.Markdown("Please tell us what you did with the animal.", label="description")
        with gr.Row():
            (
                fe_collection_dropdown,
                fe_recepient_dropdown,
                fe_radio_dropdown,
                fe_answer_dropdown,
            ) = create_followup_dropdowns(visible, "wounded")
        with gr.Row():
            fe_name_recipient, fe_collection_ref = create_followup_open(
                visible, "wounded"
            )

    # Change variables and names
    return (
        wounded_section,
        individual,
        radio_cause,
        radio_behaviour,
        radio_physical,
        button_collision,
        button_deliberate_destruction,
        button_indirect_destruction,
        button_natural_cause,
        dropdown,
        dropdown_level2,
        openfield_level2,
        dropdown_extra_level2,
        behavior_checkbox,
        behavior_text,
        physical_boxes,
        checkbox_beak,
        text_beak,
        checkbox_body,
        text_body,
        checkbox_feathers,
        text_feathers,
        checkbox_head,
        text_head,
        checkbox_legs,
        text_legs,
        fe_collection_dropdown,
        fe_recepient_dropdown,
        fe_radio_dropdown,
        fe_answer_dropdown,
        fe_name_recipient,
        fe_collection_ref,
    )