from pkg.dash.app_init import app
from dash.dependencies import Input, Output


@app.callback(
    [Output('date-picker-container', 'style'),
     Output('days-display-container', 'style'),
     Output('time-zone-checklist', 'style')],
    [Input('tabs', 'value')]
)
def toggle_controls_visibility(tab):
    if tab == 'heatmap' or tab == 'one_day_heatmap':
        return {'display': 'none'}, {'display': 'block'}, {'display': 'none'}
    return {'display': 'block'}, {'display': 'none'}, {'display': 'block'}