app/main_multianimal.py

import gradio as gr
from gradio_modal import Modal

from utils_json import *
from utils_df import save_individual_to_gallery
from maps import get_location
from functools import partial
from dead import show_section_dead
from wounded import show_section_wounded
from circumstances import show_causes
from circumstances_dropdowns import *
from physical_select_animal import show_physical, find_bounding_box
from physical_checkbox import on_select_body_part, hide_physical
from behavior_checkbox import show_behavior, on_select_behavior
from style import *
from theme import theme, css

with gr.Blocks(theme=theme, css=css) as demo:
    create_json_all_individuals()
    with gr.Row(): 
        show_modal = gr.Button("Add an Animal", scale=3)
        submit_button = gr.Button("Submit All Animals", scale=1)
    # df = gr.Dataframe(headers=get_headers(),
    #                   visible=False)
    gallery = gr.Gallery(
        label="Gallery of Records", elem_id="gallery", 
        columns=[1], rows=[1],
        object_fit="contain", height="auto", interactive=False)
    with Modal(visible=False) as modal:
        # ---------------------------------------------------------
        # Intro Text
        with gr.Row():
            with gr.Column(scale=1):
                title = gr.Markdown("# Welcome to Digiwild", label="Title")
                description = gr.Markdown("Please record your wildlife observations here !", label="description")

        # ---------------------------------------------------------
        # Camera
        with gr.Row():
            def save_image(camera):
                add_data_to_individual("image", camera.tolist())

            camera = gr.Image(elem_id="image")
            camera.input(save_image, inputs=[camera])
        # ---------------------------------------------------------
        # Location
        #with gr.Row():
            with gr.Column(scale=1):
                location = gr.Textbox(visible=True, interactive=True, label="Location of Sighting")
                #display location processing
                identified_location= gr.Textbox(visible=False, interactive=False, 
                                                label="Identified GPS Location")
                with gr.Row():
                    #to submit it
                    submit_location = gr.Button("Get GPS Coordinates", 
                                                visible=True, interactive=True, scale=3)
                    submit_location.click(get_location, inputs=[location], outputs=[identified_location])
                    #to clear it
                    clear_location = gr.ClearButton(components=[location, identified_location], 
                                                    visible=True, interactive=True, scale=1
                                                    )
                    clear_location.click()
        
        # ---------------------------------------------------------
         # ---------------------------------------------------------
        # Dead and Wounded Buttons
        gr.Markdown("## The State of the Animal", label="Title")
        gr.Markdown("Please tell us if the animal was wounded or dead.", label="description")
        with gr.Row() as block_form:
            with gr.Column(scale=1):
                butt_wounded = gr.Button("Wounded", elem_id="wounded")
            with gr.Column(scale=1):
                butt_dead = gr.Button("Dead", elem_id="dead")

        # ---------------------------------------------------------
        # Initiate sections
        section_dead, \
            button_collision_dead, button_deliberate_destruction_dead, button_indirect_destruction_dead, button_natural_cause_dead, \
                dropdown_dead, dropdown_level2_dead, openfield_level2_dead, dropdown_extra_level2_dead \
                    = show_section_dead(False)
        section_wounded, radio_cause_wounded, radio_behavior_wounded, radio_physical_wounded, \
            button_collision_wounded, button_deliberate_destruction_wounded, button_indirect_destruction_wounded, button_natural_cause_wounded, \
                dropdown_wounded, dropdown_level2_wounded, openfield_level2_wounded, dropdown_extra_level2_wounded, \
                    behavior_checkbox, behavior_text, \
                        physical_boxes_wounded, \
                            checkbox_beak, text_beak, checkbox_body, text_body, checkbox_feathers, text_feathers, checkbox_head, text_head, checkbox_legs, text_legs \
                            = show_section_wounded(False)
    
        # ---------------------------------------------------------
        # ---------------------------------------------------------
        # ---------------------------------------------------------
        # Dead Button Logic
        partial_show_section_dead = partial(show_section_dead, True)
        partial_hide_section_wounded = partial(show_section_wounded, False)
        butt_dead.click(partial_show_section_dead, 
                        inputs=None, 
                        outputs=[section_dead, 
                                button_collision_dead, button_deliberate_destruction_dead, button_indirect_destruction_dead, button_natural_cause_dead, 
                                dropdown_dead, dropdown_level2_dead, openfield_level2_dead, dropdown_extra_level2_dead
                                ])
        butt_dead.click(partial_hide_section_wounded, 
                        inputs=None, 
                        outputs=[section_wounded, 
                                radio_cause_wounded, radio_behavior_wounded, radio_physical_wounded,
                                button_collision_wounded, button_deliberate_destruction_wounded, button_indirect_destruction_wounded, button_natural_cause_wounded, 
                                dropdown_wounded, dropdown_level2_wounded, openfield_level2_wounded, dropdown_extra_level2_wounded,
                                behavior_checkbox, behavior_text, 
                                physical_boxes_wounded, 
                                checkbox_beak, text_beak, checkbox_body, text_body, checkbox_feathers, text_feathers, checkbox_head, text_head, checkbox_legs, text_legs
                                ])
        # ---------------------------------------------------------
        # Wounded Button Logic
        partial_show_section_wounded = partial(show_section_wounded, True)
        partial_hide_section_dead = partial(show_section_dead, False)

        butt_wounded.click(partial_show_section_wounded, 
                        inputs=None, 
                        outputs=[section_wounded, 
                                    radio_cause_wounded, radio_behavior_wounded, radio_physical_wounded,
                                    button_collision_wounded, button_deliberate_destruction_wounded, button_indirect_destruction_wounded, button_natural_cause_wounded, 
                                    dropdown_wounded, dropdown_level2_wounded, openfield_level2_wounded, dropdown_extra_level2_wounded,
                                    behavior_checkbox, behavior_text, 
                                    physical_boxes_wounded, 
                                    checkbox_beak, text_beak, checkbox_body, text_body, checkbox_feathers, text_feathers, checkbox_head, text_head, checkbox_legs, text_legs
                                    ])
        butt_wounded.click(partial_hide_section_dead, inputs=None, outputs=[section_dead, 
                                                                            button_collision_dead, button_deliberate_destruction_dead, button_indirect_destruction_dead, button_natural_cause_dead, 
                                                                            dropdown_dead, dropdown_level2_dead, openfield_level2_dead, dropdown_extra_level2_dead
                                                                            ])
        # ---------------------------------------------------------
        # Dropdowns Dead
        button_collision_dead.click(dropdown_collision,  
                                    outputs=[dropdown_dead, dropdown_level2_dead, openfield_level2_dead, dropdown_extra_level2_dead])
        button_deliberate_destruction_dead.click(dropdown_deliberate_destruction, outputs=[dropdown_dead, dropdown_level2_dead, openfield_level2_dead, dropdown_extra_level2_dead])
        button_indirect_destruction_dead.click(dropdown_indirect_destruction, outputs=[dropdown_dead, dropdown_level2_dead, openfield_level2_dead, dropdown_extra_level2_dead])
        button_natural_cause_dead.click(dropdown_natural_cause, outputs=[dropdown_dead, dropdown_level2_dead, openfield_level2_dead, dropdown_extra_level2_dead])

        dropdown_dead.select(on_select, None, [dropdown_level2_dead, openfield_level2_dead, dropdown_extra_level2_dead])
        dropdown_level2_dead.select(on_select_dropdown_level2)
        openfield_level2_dead.select(on_select_openfield_level2)
        dropdown_extra_level2_dead.select(on_select_dropdown_extra_level2)
        # ---------------------------------------------------------
        # Radio Cause Wounded
        radio_cause_wounded.change(fn=show_causes,
                                inputs=[radio_cause_wounded],
                                outputs=[button_collision_wounded, button_deliberate_destruction_wounded, button_indirect_destruction_wounded, button_natural_cause_wounded, 
                                            dropdown_wounded, dropdown_level2_wounded, openfield_level2_wounded, dropdown_extra_level2_wounded]
                                )
        
        # Dropdowns Cause Wounded
        button_collision_wounded.click(dropdown_collision,  
                                    outputs=[dropdown_wounded, dropdown_level2_wounded, openfield_level2_wounded, dropdown_extra_level2_wounded])
        button_deliberate_destruction_wounded.click(dropdown_deliberate_destruction, outputs=[dropdown_wounded, dropdown_level2_wounded, openfield_level2_wounded, dropdown_extra_level2_wounded])
        button_indirect_destruction_wounded.click(dropdown_indirect_destruction, outputs=[dropdown_wounded, dropdown_level2_wounded, openfield_level2_wounded, dropdown_extra_level2_wounded])
        button_natural_cause_wounded.click(dropdown_natural_cause, outputs=[dropdown_wounded, dropdown_level2_wounded, openfield_level2_wounded, dropdown_extra_level2_wounded])
        
        dropdown_wounded.select(on_select, None, [dropdown_level2_wounded, openfield_level2_wounded, dropdown_extra_level2_wounded])
        dropdown_level2_wounded.select(on_select_dropdown_level2)
        openfield_level2_wounded.select(on_select_openfield_level2)
        dropdown_extra_level2_wounded.select(on_select_dropdown_extra_level2)
        # ---------------------------------------------------------
        # Radio Behavior Wounded
        radio_behavior_wounded.change(fn=show_behavior,
                                    inputs=[radio_behavior_wounded, gr.Text("wounded", visible=False)],
                                    outputs=[behavior_checkbox, behavior_text])
        behavior_checkbox.select(on_select_behavior, 
                                 inputs=[behavior_checkbox])
        # ---------------------------------------------------------
        # Radio Physical Wounded
        radio_physical_wounded.change(fn=show_physical,
                                    inputs=[radio_physical_wounded, gr.Text("wounded", visible=False)],
                                    outputs=[physical_boxes_wounded])

        # Checkbox Physical Wounded
        physical_boxes_wounded.select(find_bounding_box, 
                        inputs=[physical_boxes_wounded, gr.Textbox(value="wounded", visible=False)], 
                        outputs=[checkbox_beak, text_beak, 
                                 checkbox_body, text_body, 
                                 checkbox_feathers, text_feathers, 
                                 checkbox_head, text_head, 
                                 checkbox_legs, text_legs
                                    ])
        checkbox_beak.select(on_select_body_part, inputs=[checkbox_beak, gr.Text("beak", visible=False)])
        checkbox_body.select(on_select_body_part, inputs=[checkbox_body, gr.Text("body", visible=False)])
        checkbox_feathers.select(on_select_body_part, inputs=[checkbox_feathers, gr.Text("feathers", visible=False)])
        checkbox_head.select(on_select_body_part, inputs=[checkbox_head, gr.Text("head", visible=False)])
        checkbox_legs.select(on_select_body_part, inputs=[checkbox_legs, gr.Text("legs", visible=False)])

        # ---------------------------------------------------------
        # Add One Individual's Data to the Dataframe
        # And Allow clearing of all previous output
        with gr.Row(): 
            button_df = gr.Button("Submit Animal Report", scale = 3)
            button_clear = gr.ClearButton(scale = 1, 
                                          components=[
                camera,
                location, identified_location, 
                button_collision_dead, button_deliberate_destruction_dead, button_indirect_destruction_dead, button_natural_cause_dead, 
                dropdown_dead, dropdown_level2_dead, openfield_level2_dead, dropdown_extra_level2_dead,
                radio_cause_wounded, radio_behavior_wounded, radio_physical_wounded,
                button_collision_wounded, button_deliberate_destruction_wounded, button_indirect_destruction_wounded, button_natural_cause_wounded, 
                dropdown_wounded, dropdown_level2_wounded, openfield_level2_wounded, dropdown_extra_level2_wounded,
                behavior_checkbox, behavior_text, 
                physical_boxes_wounded, 
                checkbox_beak, text_beak, checkbox_body, text_body, checkbox_feathers, text_feathers, checkbox_head, text_head, checkbox_legs, text_legs
                ])
            
        button_clear.click()
        button_clear.click(hide_physical,
                           outputs=[checkbox_beak, text_beak, checkbox_body, text_body, checkbox_feathers, text_feathers, checkbox_head, text_head, checkbox_legs, text_legs])
        # button_df.click(save_individual_to_df, 
        #                 inputs=[df],
        #                 outputs=[df])
        button_df.click(save_individual_to_gallery, 
                        inputs=[gallery],
                        outputs=[gallery])
        button_df.click(lambda: Modal(visible=False), None, modal)
    
    # ---------------------------------------------------------
    # Event Functions of the landing page buttons
    show_modal.click(lambda: Modal(visible=True), None, modal)
    show_modal.click(create_json_one_individual)
    #submit_button.click(save_and_rest_df, inputs=[df], outputs=[df])
    #submit_button.click(save_individual_to_gallery)


     
if __name__ == "__main__":
    demo.launch(server_name="0.0.0.0", server_port=3333)