app_withmenu.py

# Import Libraries
import numpy as np
import pandas as pd
import plotly.graph_objects as go
import datetime

# Read CSV file 

df = pd.read_csv("covid19pt_data.csv")

## set the index as the data column
## all of the remaining columns are now of the same int64 type

df = df.set_index('data')

# assign colors to column  using a dictionary
# colors = dict('casos_novos_t':'coral',
#         'casos_novos_d':'darkviolet')

# Assing the chart to a variable 
fig = go.Figure()

#set up ONE trace
fig.add_trace(go.Bar(x=df.index,
 y=df[df.columns[0]],
						 # marker_color = colors,
						 visible=True))


updatemenu = []
buttons = []

# button with one option for each column 
for col in df.columns:
  buttons.append(dict(method='restyle',
    label=col,
    visible=True,
    args=[{'y':[df[col]],
    'x':[df.index],
    'type':'bar'}, [0]],
    )
  )

# adjustments to the updatemenus

updatemenu = []
main_menu = dict()
updatemenu.append(main_menu)

updatemenu[0]['buttons'] = buttons
updatemenu[0]['direction'] = 'down'
updatemenu[0]['showactive'] = True

# add dropdown menus to the figure
fig.update_layout(showlegend=False, updatemenus=updatemenu)

fig.update_layout(
  template='plotly_dark',
  margin=dict(r=10, t=25, b=40, l=60)
  )


def customLegend(fig, nameSwap):
    for i, dat in enumerate(fig.data):
        for elem in dat:
            if elem == 'name':
                fig.data[i].name = nameSwap[fig.data[i].name]
    return(fig)

fig = customLegend(fig=fig, nameSwap = {'casos_novos_t': 'Novos Casos Total', 'casos_novos_d':'Novos Casos Diários'})

# Show Graphics
fig.show()