How can i add edge color to inner bars?

Hi everyone,
I need to create hetmap with true and false values (1 and 0 is counts), and when I create it I saw something like this,

But as you see edges of the boxes are not clear. So I wanna make it smth like this.

I wonder if you have any suggestion to make this possible?
This is the sample code, you could show it on doce if it s possible.

import plotly.express as px
data=[[1, 0,1,1,1], [1,1,1,0,1], [1,1,1,1,0]]
fig = px.imshow(data,
                labels=dict(x="Day of Week", y="Time of Day", color="Productivity"),
                x=['Monday', 'Tuesday', 'Wednesday', 'Thursday', 'Friday'],
                y=['Morning', 'Afternoon', 'Evening']
               )
fig.update_xaxes(side="top")
fig.show()

Thanks in advance!!