app.py

from dash import Dash, html, dcc, callback, Output, Input, no_update, ALL
import dash_bootstrap_components as dbc


from mission_utils import get_missions, get_missions_ids, get_mission_time
import pandas as pd
from data_retrieval import get_swift_data
from content_layout import (
    single_layout,
    single_info_card,
    multi_layout,
    multi_info_card,
)
from card_elements import multi_card, single_card
from graphs import multi_graphs, get_single_graphs, SingleGraphs

# Initialize the Dash app
app = Dash(
    __name__,
    external_stylesheets=[dbc.themes.BOOTSTRAP],
    suppress_callback_exceptions=True,
)

app.title = "Dashboard"

server = app.server

# Set the default mission to the first mission in the list.
all_missions = get_missions()
default_mission = all_missions[0]["value"]

# Define the layout of the app
app.layout = html.Div(
    [
        dbc.Row(
            [
                dbc.Col(
                    html.Div(
                        [
                            html.H1(["MicroSWIFT", html.Br(), "Dashboard"]),
                            html.Img(
                                src="assets/SWIFTlogo_r.png", width="150", height="150"
                            ),
                            html.Hr(),
                            dcc.Dropdown(
                                options=all_missions,
                                value=default_mission,
                                id="mission_dropdown",
                            ),
                            html.Br(),
                            dcc.Dropdown(id="buoy_id"),
                            html.Br(),
                            html.Hr(),
                            dbc.Card(id="latest_info", body=True),
                            html.Footer("Version 0.8"),
                        ]
                    ),
                    width=2,
                    style={
                        "background-color": "grey",
                        "padding": "10px",
                        "position": "fixed",
                        "height": "100%",
                        "overflow": "auto",
                    },
                ),
                dbc.Col(
                    html.Div(
                        id="graph_area",
                        style={
                            "margin": "auto",
                            "text-align": "center",
                            "width": "750px",
                            "height": "100%",
                            "background-color": "#f8f9fa",
                        },
                    ),
                    width={"size": 10, "offset": 2},
                ),
            ]
        )
    ]
)


# Update buoy IDs based on selected mission
@callback(
    [Output("buoy_id", "options"), Output("buoy_id", "value")],
    [Input("mission_dropdown", "value")],
)
def update_ids(mission):
    buoy_ids = get_missions_ids(mission)
    options = [{"label": buoy_id, "value": buoy_id} for buoy_id in buoy_ids]
    options.append({"label": "All Buoys", "value": "All"})
    default_id = buoy_ids[0] if buoy_ids else None
    return options, default_id


# Update layout and info based on selected buoy ID
@callback(
    [
        Output("graph_area", "children"),
        Output("latest_info", "children"),
    ],
    [Input("buoy_id", "value")],
)
def update_layout(buoy_id):
    if buoy_id == "All":
        return multi_layout, multi_info_card
    else:
        style = {"display": "block"}
        return single_layout, single_info_card


# Update multi-buoy graphs
@callback(
    [
        Output("position_temperature_multi", "figure"),
        Output("position_salinity_multi", "figure"),
        Output("position_wave_height_multi", "figure"),
        Output("buoy_time", "children"),
    ],
    [Input("buoy_id", "value"), Input("mission_dropdown", "value")],
)
def update_multi(buoy_id, mission):
    if buoy_id == "All":
        start_date, end_date = get_mission_time(mission)
        df = get_swift_data(get_missions_ids(mission), start_date, end_date)
        position_temperature, position_salinity, position_height = multi_graphs(df)
        return (
            position_temperature,
            position_salinity,
            position_height,
            multi_card(df, get_missions_ids(mission)),
        )
    return no_update, no_update, no_update, no_update


# Update single buoy graphs and info
@callback(
    [
        # Output('peak_direction_single', 'figure'),
        # Output('peak_period_single', 'figure'),
        Output("wave_height_single", "figure"),
        Output("position_temperature_single", "figure"),
        Output("position_salinity_single", "figure"),
        Output("position_wave_height_single", "figure"),
        # Output("loglog_frequency_energy", "figure"),
        # Output("linear_frequency_energy", "figure"),
        Output("spectrogram_plot", "figure"),
        Output("current_value", "children"),
        Output("buoy_info", "children"),
    ],
    [
        Input("buoy_id", "value"),
        Input("mission_dropdown", "value"),
    ],
)
def update_single(buoy_id, mission):
    single_graphs = SingleGraphs(
        wave_height=no_update,
        position_temperature=no_update,
        position_salinity=no_update,
        position_height=no_update,
        loglog=no_update,
        linear=no_update,
        spectrogram_fig=no_update,
    )
    if buoy_id != "All":
        start_date, end_date = get_mission_time(mission)
        df = get_swift_data([buoy_id], start_date, end_date)

        if df is not None:
            single_graphs = get_single_graphs(df)
            # Info Card
            single_graphs.current_id = buoy_id
            single_graphs.buoy_info = single_card(df, buoy_id)

        elif df is None:
            single_graphs.current_id = buoy_id
            single_graphs.buoy_info = "No valid data retrieved for buoy ID"
    return [
        # single_graphs.peak_direction,
        # single_graphs.peak_period,
        single_graphs.wave_height,
        single_graphs.position_temperature,
        single_graphs.position_salinity,
        single_graphs.position_height,
        # single_graphs.loglog,
        # single_graphs.linear,
        single_graphs.spectrogram_fig,
        single_graphs.current_id,
        single_graphs.buoy_info,
    ]


if __name__ == "__main__":
    app.run(debug=True)