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, 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", 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", "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